def setUp(self):
from body import Body
from database import AbilitiesDataBase
from abilities import a_test
ab_db = AbilitiesDataBase()
ab_db.addAbility(0,a_test.Test())
self.b = Body()
self.b.abilitiesDataBase(ab_db)
def setUp(self):
from database import AbilitiesDataBase
self.a_db = AbilitiesDataBase()
class AbilitiesDataBaseTest(unittest.TestCase):
def setUp(self):
from database import AbilitiesDataBase
self.a_db = AbilitiesDataBase()
def test_addAbility_exception(self):
from database import DataBaseException
from abilities import Ability
with self.assertRaises(DataBaseException):
self.a_db.addAbility(0,None)
ab1 = Ability()
ab2 = Ability()
id1 = id2 = 1
self.a_db.addAbility(id1,ab1)
with self.assertRaises(DataBaseException):
self.a_db.addAbility(id1,ab2)
def test_addAbility(self):
from abilities import Ability
ab = Ability()
id = 0
self.a_db.addAbility(id,ab)
self.assertEqual(ab,self.a_db.getAbility(id))
def test_getAbility_exception(self):
from database import DataBaseException
with self.assertRaises(DataBaseException):
self.a_db.getAbility(0)
def generateDataBase(self):
"""
Generates abilities list
Input:
Nothing
Returns:
Abilities
"""
from jarvis import Jarvis
from database import AbilitiesDataBase
from abilities import a_test
from abilities import a_time
from abilities import a_hello
from abilities import a_exit
from abilities import a_mediacenter
from abilities import a_weather
from abilities import a_didyouknow
if not isinstance(self._jarvis, Jarvis):
raise DataBaseBuilderException
abilities_list = AbilitiesDataBase()
abilities_list.addAbility(0, a_test.Test())
abilities_list.addAbility(1, a_time.Time())
abilities_list.addAbility(2, a_hello.Hello())
exit_ability = a_exit.Exit()
exit_ability.jarvis(self._jarvis)
abilities_list.addAbility(3, exit_ability)
mc = a_mediacenter.MediaCenter()
abilities_list.addAbility(4, mc.factoryBBCRadioStart())
abilities_list.addAbility(5, mc.factoryStop())
abilities_list.addAbility(6, a_weather.Weather())
abilities_list.addAbility(7, a_didyouknow.DidYouKnow())
return abilities_list
