예제 #1
0
 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())