ToOne with json_serializable & freezed - generator integration tests

Author: vaind

generator/integration-tests/part-partof/1.dart

@@ -2,10 +2,10 @@ import 'dart:io';
 
 import 'package:test/test.dart';
 
-import '../test_env.dart';
 import '../common.dart';
-import 'lib/json.dart';
+import '../test_env.dart';
 import 'lib/frozen.dart';
+import 'lib/json.dart';
 import 'lib/objectbox.g.dart';
 
 void main() {
@@ -16,12 +16,23 @@ void main() {
     expect(File('lib/objectbox-model.json').existsSync(), true);
   });
 
-  setupTestsFor(JsonEntity(id: 0, str: 'foo', date: DateTime.now()));
-  setupTestsFor(FrozenEntity(id: 1, str: 'foo', date: DateTime.now()));
+  group('package:JsonSerializable', () {
+    setupTestsFor(JsonEntity(id: 0, str: 'foo', date: DateTime.now()));
+    setupRelTestsFor(JsonBook.fromJson({
+      'author': {'name': 'Charles'}
+    }));
+  });
+
+  group('package:Freezed', () {
+    setupTestsFor(FrozenEntity(id: 1, str: 'foo', date: DateTime.now()));
+    final author = FrozenPerson(id: 1, name: 'Charles');
+    setupRelTestsFor(
+        FrozenBook(id: 1, author: ToOne<FrozenPerson>(target: author)), author);
+  });
 }
 
 void setupTestsFor<EntityT>(EntityT newObject) {
-  group('${EntityT}', () {
+  group(EntityT.toString(), () {
     late TestEnv<EntityT> env;
     setUp(() => env = TestEnv(getObjectBoxModel()));
     tearDown(() => env.close());
@@ -32,3 +43,24 @@ void setupTestsFor<EntityT>(EntityT newObject) {
     });
   });
 }
+
+void setupRelTestsFor<BookEntityT, PersonEntityT>(BookEntityT book,
+    [PersonEntityT? author]) {
+  group(BookEntityT.toString(), () {
+    late TestEnv<BookEntityT> env;
+    setUp(() => env = TestEnv(getObjectBoxModel()));
+    tearDown(() => env.close());
+
+    test('relations', () {
+      if (author != null) {
+        env.store.box<PersonEntityT>().put(author);
+        (book as dynamic).author.target = author;
+      }
+      env.box.put(book);
+
+      final bookRead = env.box.get(1)! as dynamic;
+      expect(bookRead.author.targetId, 1);
+      expect(bookRead.author.target!.name, 'Charles');
+    });
+  });
+}

generator/integration-tests/part-partof/lib/frozen.dart

@@ -1,5 +1,5 @@
-import 'package:objectbox/objectbox.dart';
 import 'package:freezed_annotation/freezed_annotation.dart';
+import 'package:objectbox/objectbox.dart';
 
 part 'frozen.freezed.dart';
 
@@ -11,3 +11,19 @@ class FrozenEntity with _$FrozenEntity {
       required String str,
       required DateTime date}) = _FrozenEntity;
 }
+
+@freezed
+class FrozenPerson with _$FrozenPerson {
+  @Entity(realClass: FrozenPerson)
+  factory FrozenPerson(
+      {@Id(assignable: true) required int id,
+      required String name}) = _FrozenPerson;
+}
+
+@freezed
+class FrozenBook with _$FrozenBook {
+  @Entity(realClass: FrozenBook)
+  factory FrozenBook(
+      {@Id(assignable: true) required int id,
+      required ToOne<FrozenPerson> author}) = _FrozenBook;
+}

generator/integration-tests/part-partof/lib/json.dart

@@ -1,5 +1,5 @@
-import 'package:objectbox/objectbox.dart';
 import 'package:json_annotation/json_annotation.dart';
+import 'package:objectbox/objectbox.dart';
 
 part 'json.g.dart';
 
@@ -17,3 +17,44 @@ class JsonEntity {
 
   Map<String, dynamic> toJson() => _$JsonEntityToJson(this);
 }
+
+@Entity()
+@JsonSerializable()
+class JsonPerson {
+  int? id;
+  String name;
+
+  JsonPerson({required this.name});
+
+  factory JsonPerson.fromJson(Map<String, dynamic> json) =>
+      _$JsonPersonFromJson(json);
+
+  Map<String, dynamic> toJson() => _$JsonPersonToJson(this);
+}
+
+@Entity()
+@JsonSerializable()
+class JsonBook {
+  int? id;
+
+  @_PersonRelToOneConverter()
+  final ToOne<JsonPerson> author;
+
+  JsonBook({required this.author});
+
+  factory JsonBook.fromJson(Map<String, dynamic> json) => _$JsonBookFromJson(json);
+
+  Map<String, dynamic> toJson() => _$JsonBookToJson(this);
+}
+
+class _PersonRelToOneConverter
+    implements JsonConverter<ToOne<JsonPerson>, Map<String, dynamic>?> {
+  const _PersonRelToOneConverter();
+
+  @override
+  ToOne<JsonPerson> fromJson(Map<String, dynamic>? json) => ToOne<JsonPerson>(
+      target: json == null ? null : JsonPerson.fromJson(json));
+
+  @override
+  Map<String, dynamic>? toJson(ToOne<JsonPerson> rel) => rel.target?.toJson();
+}

generator/integration-tests/test_env.dart

@@ -7,13 +7,12 @@ export 'package:objectbox/internal.dart';
 class TestEnv<Entity> {
   static final dir = Directory('testdata');
   late final Store store;
-  late final Box<Entity> box;
+  late final Box<Entity> box = store.box();
 
   TestEnv(ModelDefinition defs) {
     if (dir.existsSync()) dir.deleteSync(recursive: true);
 
     store = Store(defs, directory: dir.path);
-    box = Box<Entity>(store);
   }
 
   void close() {