def index():
option = ItemSearchOption()
if request.args is not None:
option.category_id = request.args.get('category_id', type=int)
option.q = request.args.get('q', type=str)
mapper = ItemMapper()
items = mapper.find(option)
return ApiResponse(200, data={'items': items})
class TestItemMapper(unittest.TestCase):
def setUp(self):
self.mapper = ItemMapper()
self.db = PgAdapter()
def tearDown(self):
self.db = PgAdapter()
query = """
TRUNCATE TABLE items
RESTART IDENTITY;
"""
self.db.execute(query)
self.db.commit()
def test_add_ok(self):
data = Item(None, 1, 'test', 500)
result = self.mapper.save(data)
self.assertTrue(result)
def test_edit_ok(self):
self.__init_data()
item = Item(1, 1, 'Edit', 500)
result = self.mapper.save(item)
self.assertTrue(result)
def test_delete_ok(self):
self.__init_data()
result = self.mapper.delete(1)
self.assertTrue(result)
def test_find_all_items(self):
self.__init_data()
data = ItemSearchOption(1)
result = self.mapper.find(data)
self.assertEqual(len(result), 3)
def test_find_when_empty_row(self):
data = ItemSearchOption(
category_id=1,
q='No data')
result = self.mapper.find(data)
self.assertEqual(len(result), 0)
def test_find_when_keyword_search(self):
self.__init_data()
data = ItemSearchOption(
category_id=None,
q='パスタ'
)
result = self.mapper.find(data)
self.assertEqual(len(result), 2)
def test_find_when_coplex_search(self):
self.__init_data()
data = ItemSearchOption(
category_id=3,
q='パスタ'
)
result = self.mapper.find(data)
self.assertEqual(len(result), 1)
def test_find_by_category_ids(self):
self.__init_data()
data = ItemSearchOption(
category_ids=[1, 2]
)
result = self.mapper.find_by_category_ids(data)
self.assertNotEqual(0, len(result))
def test_find_by_category_ids_when_empty_row(self):
self.__init_data()
data = ItemSearchOption(
category_ids=[4]
)
result = self.mapper.find_by_category_ids(data)
self.assertEqual(0, len(result))
def test_find_by_category_ids_when_no_ids(self):
self.__init_data()
data = ItemSearchOption(category_ids=[])
result = self.mapper.find_by_category_ids(data)
self.assertEqual(0, len(result))
def test_save_ng_when_invalid_value(self):
with self.assertRaises(ValueError):
self.mapper.save(None)
self.mapper.save(1)
self.mapper.save('test')
def test_delete_ng_when_invalid_value(self):
with self.assertRaises(ValueError):
self.mapper.delete(None)
self.mapper.delete('1')
self.mapper.delete(0)
self.mapper.delete(-1)
def test_invalid_find_option(self):
with self.assertRaises(ValueError):
self.mapper.find(None)
self.mapper.find('')
self.mapper.find(1)
self.mapper.find(True)
self.mapper.find_by_category_ids(None)
self.mapper.find_by_category_ids('')
self.mapper.find_by_category_ids(1)
self.mapper.find_by_category_ids(True)
option = ItemSearchOption()
option.category_ids = None
self.mapper.find_by_category_ids(option)
def __init_data(self):
self.db.execute_proc('create_test_data_items')
self.db.commit()