def test_solo_manager(self): manager = SoloManager(model_class=SampleModel) self.assertEqual(manager.query.count(), 0) manager.create(**{}) self.assertEqual(manager.query.count(), 1) manager.create(**{}) self.assertEqual(manager.query.count(), 1) self.assertNotEqual(manager.get_solo(), None)
def test_get_solo_nothing_exist(self): manager = SoloManager(model_class=SampleModel) self.assertEqual(manager.query.count(), 0) self.assertNotEqual(manager.get_solo(), None) self.assertEqual(manager.query.count(), 1)