def test_points(self):
self.assertEqual(0, self.info.points)
cid2 = 2
cid3 = 3
infos = {
cid2: TkInfo(self.mock_plugin, cid2),
cid3: TkInfo(self.mock_plugin, cid3)
}
self.mock_plugin.console.clients.getByCID = Mock(
side_effect=lambda cid: cid)
self.mock_plugin.getClientTkInfo = Mock(
side_effect=lambda cid: infos[cid])
points_2 = 45
self.info.damage(cid2, points_2)
infos[cid2].damaged(self.my_cid, points_2)
self.assertEqual(points_2, self.info.points)
points_3 = 21
self.info.damage(cid3, points_3)
infos[cid3].damaged(self.my_cid, points_3)
self.assertEqual(points_2 + points_3, self.info.points)
def setUp(self):
self.my_cid = 1
self.mock_plugin = Mock(name="plugin", spec=TkPlugin)
self.info = TkInfo(self.mock_plugin, self.my_cid)
def setUp(self):
super().setUp()
self.my_cid = 1
self.mock_plugin = self.p
self.info = TkInfo(self.mock_plugin, self.my_cid)
