def test_generate_exception(self):
from database import AbilitiesDataBaseBuilder
from database import DataBaseBuilderException
ab = AbilitiesDataBaseBuilder()
with self.assertRaises(DataBaseBuilderException):
ab.generateDataBase()
class AbilitiesDataBaseBuilderTest(unittest.TestCase):
def setUp(self):
from database import AbilitiesDataBaseBuilder
import jarvis
self.ab = AbilitiesDataBaseBuilder()
self.ab.jarvis(jarvis.Jarvis())
def test_generate(self):
from database import AbilitiesDataBase
ab_list = self.ab.generateDataBase()
self.assertIsInstance(ab_list,AbilitiesDataBase)
def test_generate_exception(self):
from database import AbilitiesDataBaseBuilder
from database import DataBaseBuilderException
ab = AbilitiesDataBaseBuilder()
with self.assertRaises(DataBaseBuilderException):
ab.generateDataBase()
def setUp(self):
from database import AbilitiesDataBaseBuilder
import jarvis
self.ab = AbilitiesDataBaseBuilder()
self.ab.jarvis(jarvis.Jarvis())
