def test_invalid_data_provider(self):
self.wrong_data_provider = 'test data provider'
with self.assertRaises(
db_exceptions.UnrecognizedDataProviderException
):
sqlite_manager.get_element(self.wrong_data_provider,
self.element_reference
)
def test_get_element_duplicate_result(self):
self.duplicate_element_reference = {
generic_constants.TYPE: library_constants.TYPE,
generic_constants.ID: 3
}
with self.assertRaises(db_exceptions.DuplicateElementsException):
self.element = sqlite_manager.get_element(
self.data_provider,
self.duplicate_element_reference
)
def test_get_element_no_result(self):
self.missing_element_reference = {
generic_constants.TYPE: library_constants.TYPE,
generic_constants.ID: 2
}
self.element = sqlite_manager.get_element(
self.data_provider,
self.missing_element_reference
)
self.assertIsNone(self.element, 'method should return None')
def test_get_element_result_basic(self):
self.result = {
generic_constants.TYPE: library_constants.TYPE,
generic_constants.ID: 1,
library_constants.NAME: 'test library 1'
}
self.element = sqlite_manager.get_element(
self.data_provider,
self.element_reference
)
self.assertEqual(self.element, self.result, 'wrong element returned')
def test_get_element_result_reference(self):
self.element_reference = {
generic_constants.TYPE: book_constants.TYPE,
generic_constants.ID: 1
}
self.result = {
generic_constants.TYPE: book_constants.TYPE,
generic_constants.ID: 1,
generic_constants.TITLE: 'test book 1',
generic_constants.PARENT: {
generic_constants.TYPE: library_constants.TYPE,
generic_constants.ID: 4
}
}
self.element = sqlite_manager.get_element(
self.data_provider,
self.element_reference
)
self.assertEqual(self.element, self.result, 'wrong element returned')
def test_get_element_result_list(self):
self.element_reference = {
generic_constants.TYPE: library_constants.TYPE,
generic_constants.ID: 4
}
self.result = {
generic_constants.TYPE: library_constants.TYPE,
generic_constants.ID: 4,
library_constants.NAME: 'test library 4',
generic_constants.CHILDREN: {
list_constants.ELEMENTS: [
{generic_constants.TYPE: book_constants.TYPE,
generic_constants.ID: 1
}
]
}
}
self.element = sqlite_manager.get_element(
self.data_provider,
self.element_reference
)
self.assertEqual(self.element, self.result, 'wrong element returned')
def test_invalid_element_reference(self):
self.wrong_element_reference = 'test element reference'
with self.assertRaises(db_exceptions.UnrecognizedElementException):
sqlite_manager.get_element(self.data_provider,
self.wrong_element_reference
)
def test_no_element_reference(self):
with self.assertRaises(generic_exceptions.UndefinedParameterError):
sqlite_manager.get_element(self.data_provider, None)
