class DataBaseBuilderTest(unittest.TestCase):
def setUp(self):
from database.databasebuilder import DataBaseBuilder
self.builder = DataBaseBuilder()
def test_addTxtFile(self):
self.builder.addTxtFile("sample.txt")
def test_generateDataBase_exception(self):
from database import DataBaseBuilderException
with self.assertRaises(DataBaseBuilderException):
self.builder.generateDataBase()
def test_readFileContents_exception(self):
from database import DataBaseBuilderException
with self.assertRaises(DataBaseBuilderException):
contents = self.builder._readFileContents("fff.txt")
def test_readFileContents(self):
contents = self.builder._readFileContents("sample.txt")
self.assertIsInstance(contents,str)
self.assertIn("aaa", contents)
self.assertIn("ddd", contents)
def setUp(self):
from database.databasebuilder import DataBaseBuilder
self.builder = DataBaseBuilder()
