class TestHealthComponent(TestCase): def setUp(self): self.instance = Mock() self.instance.session = Dummy() self.component = HealthComponent(20) self.component.instance = self.instance self.component.initialize() def test_trivial(self): self.assertEqual(self.component.health, 20) self.assertEqual(self.component.max_health, 20) def test_maxhealth_required(self): self.assertRaises(AssertionError, HealthComponent, None) def test_damage(self): self.component.deal_damage(1, 19) self.assertEqual(self.component.health, 1) self.assertEqual(self.component.max_health, 20) self.assertFalse(self.instance.remove.called) def test_damage_zero_health(self): self.component.deal_damage(1, 20) self.assertEqual(self.component.health, 0) self.assertEqual(self.component.max_health, 20) self.assertTrue(self.instance.remove.called) def test_huge_damage(self): self.component.deal_damage(1, 300) self.assertEqual(self.component.health, 0) self.assertEqual(self.component.max_health, 20) self.assertTrue(self.instance.remove.called)
def setUp(self): self.instance = Mock() self.instance.session = Dummy() self.component = HealthComponent(20) self.component.instance = self.instance self.component.initialize()