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))
