From 75646a1874bd27c121e88ecbf4d809670920cc5d Mon Sep 17 00:00:00 2001 From: Trey Hunner Date: Tue, 30 Jul 2013 10:16:43 -0700 Subject: [PATCH] Improve FieldTracker tests for inherited models --- model_utils/tests/tests.py | 17 +++++++++++++---- 1 file changed, 13 insertions(+), 4 deletions(-) diff --git a/model_utils/tests/tests.py b/model_utils/tests/tests.py index 9c427bc..bc6acb9 100644 --- a/model_utils/tests/tests.py +++ b/model_utils/tests/tests.py @@ -1074,11 +1074,16 @@ class FieldTrackerForeignKeyTests(FieldTrackerTestCase): self.assertCurrent(fk=self.instance.fk_id) -# TODO test stuff with name2 class InheritedFieldTrackerTests(FieldTrackerTests): tracked_class = InheritedTracked + def test_child_fields_not_tracked(self): + self.name2 = 'test' + self.assertEqual(self.tracker.previous('name2'), None) + self.assertRaises(FieldError, self.tracker.has_changed, 'name2') + + class ModelTrackerTests(FieldTrackerTests): tracked_class = ModelTracked @@ -1204,7 +1209,11 @@ class ModelTrackerForeignKeyTests(FieldTrackerForeignKeyTests): self.assertCurrent(fk=self.instance.fk) -# TODO test stuff with name2 -class InheritanceModelTrackerTests(ModelTrackerTests): +class InheritedModelTrackerTests(ModelTrackerTests): - tracked_class = InheritedModelTracked \ No newline at end of file + tracked_class = InheritedModelTracked + + def test_child_fields_not_tracked(self): + self.name2 = 'test' + self.assertEqual(self.tracker.previous('name2'), None) + self.assertTrue(self.tracker.has_changed('name2'))