Update create history model test to use register() instead of the i…

…nternal API
jazzband · Apr 3, 2015 · ae20ef4 · ae20ef4
1 parent 7e97aec
commit ae20ef4
Showing 1 changed file with 6 additions and 12 deletions.
diff --git a/simple_history/tests/tests/test_models.py b/simple_history/tests/tests/test_models.py
@@ -327,29 +327,23 @@ def test_register_custome_records(self):
 class CreateHistoryModelTests(TestCase):
 
     def test_create_history_model_with_one_to_one_field_to_integer_field(self):
-        records = HistoricalRecords()
-        records.module = AdminProfile.__module__
         try:
-            records.create_history_model(AdminProfile)
-        except:
+            register(AdminProfile)
+        except Exception:
             self.fail("SimpleHistory should handle foreign keys to one to one"
                       "fields to integer fields without throwing an exception")
 
     def test_create_history_model_with_one_to_one_field_to_char_field(self):
-        records = HistoricalRecords()
-        records.module = Bookcase.__module__
         try:
-            records.create_history_model(Bookcase)
-        except:
+            register(Bookcase)
+        except Exception:
             self.fail("SimpleHistory should handle foreign keys to one to one"
                       "fields to char fields without throwing an exception.")
 
     def test_create_history_model_with_multiple_one_to_ones(self):
-        records = HistoricalRecords()
-        records.module = MultiOneToOne.__module__
         try:
-            records.create_history_model(MultiOneToOne)
-        except:
+            register(MultiOneToOne)
+        except Exception:
             self.fail("SimpleHistory should handle foreign keys to one to one"
                       "fields to one to one fields without throwing an "
                       "exception.")