class TestModels(unittest.TestCase): def setUp(self): User.objects.delete({'email':'*****@*****.**'}) self.user = User(email='*****@*****.**', password='******') self.user.save() GPSDevice.objects.delete({'imei':'test'}) self.device = GPSDevice(imei='test') self.device.save() self.user.devices.append(self.device) self.user.save() def tearDown(self): self.user.delete() self.device.delete() # User def test_user(self): assert(self.user.check_password('test')) # GPSDevice def test_device_user(self): assert(self.user==self.device.user) def test_unique_imei(self): try: d = GPSDevice(imei='test') d.save() except mongoengine.NotUniqueError, e: pass assert(mongoengine.NotUniqueError == type(e))
def test_unique_imei(self): try: d = GPSDevice(imei='test') d.save() except mongoengine.NotUniqueError, e: pass