add tests

Author: yash-agarwa-l

dart/packages/fory-test/test/codegen_test/enum_codegen_test.dart

@@ -32,24 +32,34 @@ void main() {
       EnumSpec enumSpec = EnumSpec(EnumFoo, [EnumFoo.A, EnumFoo.B]);
       EnumSpec enumSubTypeSpec =
           EnumSpec(EnumSubClass, [EnumSubClass.A, EnumSubClass.B]);
-      EnumSpec enumWithIdsSpec = EnumSpec(
-        EnumWithIds,
-        [EnumWithIds.A, EnumWithIds.B, EnumWithIds.C],
-        {
-          10: EnumWithIds.A,
-          20: EnumWithIds.B,
-          30: EnumWithIds.C,
-        });
-      EnumSpec enumPartialIdsSpec =
-          EnumSpec(EnumPartialIds,[EnumPartialIds.A, EnumPartialIds.B, EnumPartialIds.C]);
-      EnumSpec enumDuplicateIdsSpec = 
-          EnumSpec(EnumDuplicateIds, [EnumDuplicateIds.A, EnumDuplicateIds.B, EnumDuplicateIds.C]);
 
       check($EnumFoo).equals(enumSpec);
       check($EnumSubClass).equals(enumSubTypeSpec);
+    });
+
+    test('test per-value @ForyEnumId spec generation', () {
+      EnumSpec enumWithIdsSpec = EnumSpec(EnumWithIds,
+          [EnumWithIds.A, EnumWithIds.B, EnumWithIds.C],
+          {10: EnumWithIds.A, 20: EnumWithIds.B, 30: EnumWithIds.C});
+      EnumSpec enumPartialIdsSpec = EnumSpec(
+          EnumPartialIds, [EnumPartialIds.A, EnumPartialIds.B, EnumPartialIds.C]);
+      EnumSpec enumDuplicateIdsSpec = EnumSpec(EnumDuplicateIds,
+          [EnumDuplicateIds.A, EnumDuplicateIds.B, EnumDuplicateIds.C]);
+
       check($EnumWithIds).equals(enumWithIdsSpec);
       check($EnumPartialIds).equals(enumPartialIdsSpec);
       check($EnumDuplicateIds).equals(enumDuplicateIdsSpec);
     });
+
+    test('test field-based @ForyEnumId spec generation', () {
+      EnumSpec enumFieldBasedIdsSpec = EnumSpec(EnumFieldBasedIds,
+          [EnumFieldBasedIds.A, EnumFieldBasedIds.B, EnumFieldBasedIds.C],
+          {10: EnumFieldBasedIds.A, 20: EnumFieldBasedIds.B, 30: EnumFieldBasedIds.C});
+      EnumSpec enumFieldBasedDuplicateIdsSpec = EnumSpec(EnumFieldBasedDuplicateIds,
+          [EnumFieldBasedDuplicateIds.A, EnumFieldBasedDuplicateIds.B, EnumFieldBasedDuplicateIds.C]);
+
+      check($EnumFieldBasedIds).equals(enumFieldBasedIdsSpec);
+      check($EnumFieldBasedDuplicateIds).equals(enumFieldBasedDuplicateIdsSpec);
+    });
   });
 }

dart/packages/fory-test/test/datatype_test/enum_serializer_test.dart

@@ -144,5 +144,48 @@ void main() {
         ),
       ).throws<DeserializationRangeException>();
     });
+
+    test('writes and reads field-based enum ids', () {
+      final EnumSerializer serializer = EnumSerializer(false, [
+        EnumFieldBasedIds.A,
+        EnumFieldBasedIds.B,
+        EnumFieldBasedIds.C,
+      ], {
+        10: EnumFieldBasedIds.A,
+        20: EnumFieldBasedIds.B,
+        30: EnumFieldBasedIds.C,
+      });
+
+      final ByteWriter writer = ByteWriter();
+      serializer.write(writer, EnumFieldBasedIds.B, _newSerializationContext());
+      final ByteReader reader = ByteReader.forBytes(writer.takeBytes());
+      check(reader.readVarUint32Small7()).equals(20);
+
+      final ByteWriter idWriter = ByteWriter();
+      idWriter.writeVarUint32Small7(30);
+      final Enum value = serializer.read(
+        ByteReader.forBytes(idWriter.takeBytes()),
+        0,
+        _newDeserializationContext(),
+      );
+      check(value).equals(EnumFieldBasedIds.C);
+    });
+
+    test('field-based duplicate ids fall back to ordinal', () {
+      final EnumSerializer serializer = EnumSerializer(false, [
+        EnumFieldBasedDuplicateIds.A,
+        EnumFieldBasedDuplicateIds.B,
+        EnumFieldBasedDuplicateIds.C,
+      ]);
+
+      final ByteWriter writer = ByteWriter();
+      serializer.write(
+        writer,
+        EnumFieldBasedDuplicateIds.B,
+        _newSerializationContext(),
+      );
+      final ByteReader reader = ByteReader.forBytes(writer.takeBytes());
+      check(reader.readVarUint32Small7()).equals(1);
+    });
   });
 }