95 lines
No EOL
3.7 KiB
Python
95 lines
No EOL
3.7 KiB
Python
import importlib
|
|
import unittest
|
|
import tests.test_data as test_data
|
|
import shopping
|
|
|
|
import ingredients, products
|
|
|
|
from db import connect, create
|
|
|
|
def reload_test_data():
|
|
global test_data
|
|
test_data = importlib.reload(test_data)
|
|
|
|
def first(iterable: list, predicate: callable):
|
|
for item in iterable:
|
|
if predicate(item):
|
|
return item
|
|
return None
|
|
|
|
class TestShopping(unittest.IsolatedAsyncioTestCase):
|
|
async def asyncSetUp(self):
|
|
self.conn = await connect(':memory:')
|
|
await create(self.conn)
|
|
reload_test_data()
|
|
return await super().asyncSetUp()
|
|
|
|
async def asyncTearDown(self) -> None:
|
|
await self.conn.close()
|
|
return await super().asyncTearDown()
|
|
|
|
async def test_current_shopping_list(self):
|
|
shopping_list = await shopping.current_shopping_list(self.conn)
|
|
self.assertEqual(len(shopping_list.requests), 0)
|
|
self.assertEqual(len(shopping_list.items), 0)
|
|
|
|
async def test_sync_persons_requests(self):
|
|
ingredient = test_data.Ingredients.one_apple
|
|
person = test_data.Persons.jacob
|
|
|
|
await products.insert_product(self.conn, ingredient.product, {})
|
|
|
|
shopping_list = await shopping.current_shopping_list(self.conn)
|
|
await shopping.sync_persons_requests(self.conn, shopping_list, person, [ingredient])
|
|
|
|
shopping_list = await shopping.current_shopping_list(self.conn)
|
|
self.assertEqual(len(shopping_list.requests), 1)
|
|
self.assertEqual(len(shopping_list.items), 0)
|
|
|
|
request = shopping_list.requests[0]
|
|
self.assertEqual(request.person_id, person.id)
|
|
self.assertEqual(request.ingredient.line, ingredient.line)
|
|
|
|
async def test_sync_persons_requests_multiple_add_item(self):
|
|
first = test_data.Ingredients.one_apple
|
|
second = test_data.Ingredients.salt
|
|
|
|
person = test_data.Persons.jacob
|
|
|
|
await products.insert_product(self.conn, first.product, {})
|
|
await products.insert_product(self.conn, second.product, {})
|
|
|
|
shopping_list = await shopping.current_shopping_list(self.conn)
|
|
await shopping.sync_persons_requests(self.conn, shopping_list, person, [first])
|
|
await shopping.sync_persons_requests(self.conn, shopping_list, person, [first, second])
|
|
|
|
shopping_list = await shopping.current_shopping_list(self.conn)
|
|
self.assertEqual(len(shopping_list.requests), 2)
|
|
self.assertEqual(len(shopping_list.items), 0)
|
|
|
|
request_by_line = {r.ingredient.line: r for r in shopping_list.requests}
|
|
self.assertEqual(len(request_by_line), 2)
|
|
|
|
for requested_ingredient in [first, second]:
|
|
request = request_by_line[requested_ingredient.line]
|
|
self.assertEqual(request.person_id, person.id)
|
|
self.assertEqual(request.ingredient.line, requested_ingredient.line)
|
|
|
|
async def test_mark_found(self):
|
|
ingredient = test_data.Ingredients.one_apple
|
|
person = test_data.Persons.jacob
|
|
|
|
await products.insert_product(self.conn, ingredient.product, {})
|
|
|
|
shopping_list = await shopping.current_shopping_list(self.conn)
|
|
await shopping.sync_persons_requests(self.conn, shopping_list, person, [ingredient])
|
|
await shopping.mark_found(self.conn, ingredient.product, 2, 'items')
|
|
|
|
shopping_list = await shopping.current_shopping_list(self.conn)
|
|
self.assertEqual(len(shopping_list.requests), 1)
|
|
|
|
self.assertEqual(len(shopping_list.items), 1)
|
|
self.assertEqual(shopping_list.items[0].product_id, ingredient.product.id)
|
|
self.assertEqual(shopping_list.items[0].quantity, 2)
|
|
self.assertEqual(shopping_list.items[0].unit, 'items')
|
|
|