class testWBRoom(unittest.TestCase): def setUp(self): self.room = WBRoom() #def tearDown(self): def testCreate(self): self.assertEqual(self.room.statusGetKeys(), []) self.assertEqual(self.room.clientsGetIds(), []) def testClientsAdd(self): self.room.clientsAdd(3) self.room.clientsAdd(4) self.assert_(3 in self.room.clientsGetIds()) self.assert_(4 in self.room.clientsGetIds()) self.assert_(2 not in self.room.clientsGetIds()) def testClientsRemove(self): self.room.clientsAdd(3) self.room.clientsAdd(4) self.room.clientsRemove(3) self.assert_(3 not in self.room.clientsGetIds()) self.assert_(4 in self.room.clientsGetIds()) def testAddTextBox(self): self.room.statusAddTextBox(boxId=1, x=3, y=4, owner=4) self.room.statusSetField(1, "text", "waldo") self.assertEqual(self.room.statusGetField(boxId=1, field="text"), "waldo") self.assertEqual(self.room.statusGetField(boxId=1, field="x"), 3) self.assertEqual(self.room.statusGetField(boxId=1, field="y"), 4) self.assertEqual(self.room.statusGetField(boxId=1, field="owner"), 4) def testStatusGetKeys(self): self.room.statusAddTextBox(boxId=1, x=3, y=4, owner=4) self.assert_(1 in self.room.statusGetKeys()) self.assert_(2 not in self.room.statusGetKeys())