def test_removal(self):
su = ServiceUser(UID=1)
su.save()
su2 = ServiceUser(UID=2)
su2.save()
ServiceUser.connect(1, 2)
try:
ServiceUser.remove('aaa', 1)
except Exception as e:
self.assertIsInstance(e, InvalidUIDError)
try:
ServiceUser.remove(1, 'bbb')
except Exception as e:
self.assertIsInstance(e, InvalidUIDError)
try:
ServiceUser.remove(1, 7)
except Exception as e:
self.assertIsInstance(e, NonExistingUIDError)
try:
ServiceUser.remove(1, 1)
except Exception as e:
self.assertIsInstance(e, EqualUIDError)
ServiceUser.remove(1, 2)
self.assertEquals(len(su.connected_users.all()), 0)
ServiceUser.remove(1, 2)
self.assertEquals(len(su.connected_users.all()), 0)
def test_connection(self):
su = ServiceUser(UID=1)
su.save()
su2 = ServiceUser(UID=2)
su2.save()
try:
ServiceUser.connect('aaa', 1)
except Exception as e:
self.assertIsInstance(e, InvalidUIDError)
try:
ServiceUser.connect(1, 'bbb')
except Exception as e:
self.assertIsInstance(e, InvalidUIDError)
try:
ServiceUser.connect(1, 7)
except Exception as e:
self.assertIsInstance(e, NonExistingUIDError)
try:
ServiceUser.connect(1, 1)
except Exception as e:
self.assertIsInstance(e, EqualUIDError)
ServiceUser.connect(1, 2)
self.assertEquals(len(su.connected_users.all()), 1)
self.assertEquals(su.connected_users.first(), ServiceUser.get_by_uid(2))
ServiceUser.connect(1, 2)
self.assertEquals(len(su.connected_users.all()), 1)
self.assertEquals(su.connected_users.first(), ServiceUser.get_by_uid(2))
def test_getting_by_uid(self):
su = ServiceUser(UID=1)
su.save()
try:
ServiceUser.get_by_uid(2)
except Exception as e:
self.assertIsInstance(e, NonExistingUIDError)
su = ServiceUser.get_by_uid(1)
self.assertIsInstance(su, ServiceUser)
self.assertEquals(su, ServiceUser.objects.get(UID=1))
def test_creation(self):
try:
ServiceUser(UID='aaa')
except Exception as e:
self.assertIsInstance(e, InvalidUIDError)
try:
ServiceUser(UID=-1)
except Exception as e:
self.assertIsInstance(e, InvalidUIDError)
su = ServiceUser(UID=1)
su.save()
self.assertIsInstance(su, ServiceUser)
self.assertEquals(su, ServiceUser.objects.get(UID=1))
def test_getting_connected_users(self):
su = ServiceUser(UID=1)
su.save()
su2 = ServiceUser(UID=2)
su2.save()
ServiceUser.connect(1, 2)
try:
ServiceUser.get_connected_users(3)
except Exception as e:
self.assertIsInstance(e, NonExistingUIDError)
con_list = ServiceUser.get_connected_users(1)
self.assertEquals(list(con_list), list([su2]))
con_list = ServiceUser.get_connected_users(2)
self.assertEquals(list(con_list), list([su]))
ServiceUser.remove(1, 2)
con_list = ServiceUser.get_connected_users(1)
self.assertEquals(list(con_list), list())
con_list = ServiceUser.get_connected_users(2)
self.assertEquals(list(con_list), list())
