def test_add_user(self):
owner = User(name="joe")
room = Room(owner=owner)
user = User(name="james")
room.add_user(user)
self.assertIn(
user._id,
Room(_id=room._id)._object['users']
)
def test_remove_inexistent_user(self):
owner = User(name="joe")
room = Room(owner=owner)
user = User(name="james")
room.add_user(user)
room.remove_user(user)
self.assertNotIn(
user._id,
Room(_id=room._id)._object['users']
)
def test_draft(self):
owner = User(name="joe")
room = Room(owner=owner)
for name in ['john','carl','james']:
user = User(name=name)
room.add_user(user)
self.assertIn(
room.draft(),
Room(_id=room._id)._object['users']
)
def test_add_user_twice(self):
owner = User(name="joe")
room = Room(owner=owner)
user = User(name="james")
room.add_user(user)
room.add_user(user)
self.assertEqual(
Room(_id=room._id)._object['users'].count(user._id),
1
)
