def bar_default(name):
"""
Returns a bar of size 1, 2
:param name: the name of the bar
:return:
"""
b = Bar(name)
b.add_box(Box(1))
b.add_box(Box(2))
return b
def bar_consequence(name):
"""
Returns a bar of size 2, 4, 6
:param name: the name of the bar
:return:
"""
b = Bar(name)
b.add_box(Box(2))
b.add_box(Box(4))
b.add_box(Box(6))
return b
class CharacterBarTest(unittest.TestCase):
def setUp(self):
self.c = Character("bob")
self.bar = Bar("Test")
self.bar.add_box(Box(1))
self.bar.add_box(Box(2))
self.bar.add_box(Box(3))
self.bar.add_box(Box(4))
self.c.add_bar(self.bar)
def test_add(self):
self.assertEqual(self.bar, self.c.get_bar("Test"),
"Did not properly add bar to character")
def test_get_empty(self):
self.assertEqual(None, self.c.get_bar("Does not exist"),
"None not returned on non-existent bar")
def test_get_non_case_sensitive(self):
self.assertEqual(self.bar, self.c.get_bar("TEST"),
"not non-case-sensitive")
def test_removal(self):
self.assertEqual(self.bar, self.c.remove_bar("Test"),
"Did not return true on successful removal")
self.assertEqual(None, self.c.get_bar("Test"),
"Did not remove bar properly")
def test_removal_does_not_exit(self):
self.assertFalse(self.c.remove_bar("Does not exist"),
"Did not return false on non-existent bar")
def test_spend(self):
self.c.spend_box("Test", 1)
self.assertTrue(self.bar[1].used, "Did not spend the correct box")
self.assertFalse(self.bar[2].used, "Did not spend the correct box")
self.assertFalse(self.bar[3].used, "Did not spend the correct box")
self.assertFalse(self.bar[4].used, "Did not spend the correct box")
self.c.spend_box("Test", 2)
self.assertTrue(self.bar[1].used, "Did not spend the correct box")
self.assertTrue(self.bar[2].used, "Did not spend the correct box")
self.assertFalse(self.bar[3].used, "Did not spend the correct box")
self.assertFalse(self.bar[4].used, "Did not spend the correct box")
def test_refresh(self):
# spend all
self.bar[1].spend()
self.bar[2].spend()
self.bar[3].spend()
self.bar[4].spend()
self.c.refresh_box("Test", 1)
self.assertFalse(self.bar[1].used, "Did not refresh the correct box")
self.assertTrue(self.bar[2].used, "Did not refresh the correct box")
self.assertTrue(self.bar[3].used, "Did not refresh the correct box")
self.assertTrue(self.bar[4].used, "Did not refresh the correct box")
self.c.refresh_box("Test", 3)
self.assertFalse(self.bar[1].used, "Did not refresh the correct box")
self.assertTrue(self.bar[2].used, "Did not refresh the correct box")
self.assertFalse(self.bar[3].used, "Did not refresh the correct box")
self.assertTrue(self.bar[4].used, "Did not refresh the correct box")
def test_refresh_non_case_sensitive(self):
# spend all
self.bar[1].spend()
self.bar[2].spend()
self.bar[3].spend()
self.bar[4].spend()
self.c.refresh_box("test", 1)
self.assertFalse(self.bar[1].used, "Did not refresh the correct box")
self.assertTrue(self.bar[2].used, "Did not refresh the correct box")
self.assertTrue(self.bar[3].used, "Did not refresh the correct box")
self.assertTrue(self.bar[4].used, "Did not refresh the correct box")
def test_spend_non_case_sensitive(self):
self.c.spend_bar("TEST")
self.assertTrue(self.bar[1].used, "Did not spend the correct box")
self.assertTrue(self.bar[2].used, "Did not spend the correct box")
self.assertTrue(self.bar[3].used, "Did not spend the correct box")
self.assertTrue(self.bar[4].used, "Did not spend the correct box")
def test_refresh_bar(self):
self.c.refresh_bar("test")
self.assertFalse(self.bar[1].used, "Did not refresh")
self.assertFalse(self.bar[2].used, "Did not refresh")
self.assertFalse(self.bar[3].used, "Did not refresh")
self.assertFalse(self.bar[4].used, "Did not refresh")
def test_spend_bar(self):
self.c.spend_bar("test")
self.assertTrue(self.bar[1].used, "Did not spend")
self.assertTrue(self.bar[2].used, "Did not spend")
self.assertTrue(self.bar[3].used, "Did not spend")
self.assertTrue(self.bar[4].used, "Did not spend")
