test for nulled & default fields

Author: smn

Diff for: elasticgit/commands/tests/test_avro.py

@@ -391,7 +391,7 @@ class Foo(models.Model):
 
 class TestAvroDataWriter(ToolBaseTest):
 
-    def assertBinaryRoundTrip(self, model_class, data):
+    def binary_roundtrip(self, model_class, data):
         model = model_class(data)
         schema_dumper = self.mk_schema_dumper()
         schema = avro.schema.parse(schema_dumper.dump_schema(model_class))
@@ -404,7 +404,11 @@ def assertBinaryRoundTrip(self, model_class, data):
                 open(file_name, 'rb'),
                 DatumReader(readers_schema=schema)) as reader:
             [row] = reader
-            self.assertEqual(model, model_class(row))
+            return row
+
+    def assertBinaryRoundTrip(self, model_class, data):
+        round_tripped = self.binary_roundtrip(model_class, data)
+        self.assertEqual(model_class(data), model_class(round_tripped))
 
     def test_empty_model(self):
 
@@ -431,3 +435,13 @@ class Foo(models.Model):
 
         self.assertBinaryRoundTrip(
             Foo, {'dict_': {'int': 1, 'text': 'texts'}})
+
+    def test_null_fields(self):
+        class Foo(models.Model):
+            name = models.TextField('name')
+            age = models.IntegerField('age', default=5)
+
+        data = self.binary_roundtrip(Foo, {})
+        model = Foo(data)
+        self.assertEqual(model.age, 5)
+        self.assertEqual(model.name, None)