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_raises_ValueError_for_solo_manager(self):
manager = SoloManager(model_class=SampleModel)
self.assertRaises(ValueError, lambda: manager.get(id=1))
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)
