def setUp(self):
self.instance = Mock()
self.instance.session = Dummy()
self.component = HealthComponent(20)
self.component.instance = self.instance
self.component.initialize()
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)
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()
