How can I use NativeDatabase to define SQLite functions with migration testing?

[notcancername]

I am using a SQLite custom function in my database class:

LazyDatabase _openConnection() {
    return LazyDatabase(() async {
     // ...
      return NativeDatabase(..., setup: (raw) {
          raw.createFunction(...);
    });
  });

How can I ensure this function gets defined even when using InitializedSchema.newConnection in a migration test?
I've read the interfaces, searched the web and these Discussions, but I haven't found a solution.

[simolus3]

This is not possible with a released drift version unfortunately, but I've just added the necessary callback in 37f120d.

[notcancername]

Thanks a lot!