def test_no_element(self):
self.columns = '[title]'
with self.assertRaises(generic_exceptions.UndefinedParameterError):
sqlite_manager.get_query(db_constants.SELECT,
None,
columns=self.columns
)
def test_unrecognized_element(self):
self.columns = '[title]'
with self.assertRaises(db_exceptions.UnrecognizedElementException):
sqlite_manager.get_query(db_constants.SELECT,
'test_element',
columns=self.columns
)
def test_unrecognized_property(self):
self.columns = '[test_column]'
with self.assertRaises(db_exceptions.UnrecognizedPropertiesException):
sqlite_manager.get_query(db_constants.SELECT,
self.element,
columns=self.columns
)
def test_insert_mismatched_columns_values(self):
self.columns = ['title']
self.values = ['test_value1', 'test_value2']
with self.assertRaises(db_exceptions.MismatchedParametersException):
sqlite_manager.get_query(db_constants.INSERT,
self.element,
columns=self.columns,
values=self.values
)
def test_delete(self):
self.result = ('delete from book', None)
self.query = sqlite_manager.get_query(
db_constants.DELETE,
self.element
)
self.assertEqual(self.query, self.result, 'wrong query returned')
def test_select_two_columns(self):
self.columns = ['title', 'author']
self.result = ('select title,author from book', None)
self.query = sqlite_manager.get_query(db_constants.SELECT,
self.element,
columns=self.columns
)
self.assertEqual(self.query, self.result, 'wrong query returned')
def test_update(self):
self.column = 'title'
self.value = 'test_value'
self.result = ('update book set title=?', (self.value,))
self.query = sqlite_manager.get_query(db_constants.UPDATE,
self.element,
column=self.column,
value=self.value
)
self.assertEqual(self.query, self.result, 'wrong query returned')
def test_insert_one_value(self):
self.columns = ['title']
self.values = ['test_value']
self.result = ('insert into book (title) values (?)', ('test_value',))
self.query = sqlite_manager.get_query(
db_constants.INSERT,
self.element,
columns=self.columns,
values=self.values
)
self.assertEqual(self.query, self.result, 'wrong query returned')
def test_delete_where(self):
self.search_column = 'library_id'
self.search_value = 1
self.result = ('delete from book where library_id=?', (self.search_value,))
self.query = sqlite_manager.get_query(
db_constants.DELETE,
self.element,
search_column=self.search_column,
search_value=self.search_value
)
self.assertEqual(self.query, self.result, 'wrong query returned')
def test_select_where_numeric(self):
self.columns = ['title']
self.search_column = 'library_id'
self.search_value = 1
self.result = ('select title from book where library_id=?', (1,))
self.query = sqlite_manager.get_query(
db_constants.SELECT,
self.element,
columns=self.columns,
search_column=self.search_column,
search_value=self.search_value
)
self.assertEqual(self.query, self.result, 'wrong query returned')
def test_insert_two_values(self):
self.columns = ['title', 'synopsis']
self.values = ['test_value1', 'test_value2']
self.result = ('insert into book (title,synopsis) values (?,?)',
('test_value1', 'test_value2')
)
self.query = sqlite_manager.get_query(
db_constants.INSERT,
self.element,
columns=self.columns,
values=self.values
)
self.assertEqual(self.query, self.result, 'wrong query returned')
def test_select_where_text(self):
self.columns = ['title']
self.search_column = 'title'
self.search_value = 'test_value'
self.result = ('select title from book where title=?',
('test_value',)
)
self.query = sqlite_manager.get_query(
db_constants.SELECT,
self.element,
columns=self.columns,
search_column=self.search_column,
search_value=self.search_value
)
self.assertEqual(self.query, self.result, 'wrong query returned')
def test_unrecognized_query(self):
with self.assertRaises(db_exceptions.UnrecognizedQueryTypeException):
sqlite_manager.get_query('test_query', self.element)
def test_no_query(self):
with self.assertRaises(generic_exceptions.UndefinedParameterError):
sqlite_manager.get_query(None, self.element)
