117 lines
4.2 KiB
Python
117 lines
4.2 KiB
Python
import unittest
|
|
|
|
import tests.test_data as test_data
|
|
|
|
import products.db as products_db
|
|
from db import connect, create
|
|
|
|
import importlib
|
|
|
|
|
|
def reload_test_data():
|
|
global test_data
|
|
test_data = importlib.reload(test_data)
|
|
|
|
|
|
class TestProductsDb(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 testCreateAndFind(self) -> None:
|
|
product = test_data.Products.broccoli
|
|
await products_db.insert_product(self.conn, product, {})
|
|
self.assertIsNotNone(product)
|
|
self.assertGreater(product.id, 0)
|
|
|
|
product_by_id = await products_db.find_product_by_id(self.conn, product.id)
|
|
self.assertIsNotNone(product_by_id)
|
|
self.assertEqual(product_by_id.id, product.id)
|
|
self.assertEqual(product_by_id.name, product.name)
|
|
self.assertEqual(product_by_id.link, product.link)
|
|
self.assertEqual(product_by_id.img_large, product.img_large)
|
|
self.assertEqual(product_by_id.img_small, product.img_small)
|
|
|
|
|
|
from . import httpx_mocks
|
|
from products import woolworths
|
|
|
|
|
|
class TestWoolworths(unittest.IsolatedAsyncioTestCase):
|
|
async def asyncSetUp(self) -> None:
|
|
local_path = "./tests/sample_files/woolworths"
|
|
woolworths._get_client = lambda: httpx_mocks.MockAsyncClient(local_path)
|
|
# woolworths._get_client = lambda: httpx_mocks.RecordingAsyncClient(local_path)
|
|
return await super().asyncSetUp()
|
|
|
|
async def test_get_product_id(self) -> None:
|
|
params = [
|
|
("https://www.woolworths.com.au/shop/productdetails/144607/strawberries", "144607"),
|
|
(
|
|
"https://www.woolworths.com.au/shop/productdetails/133211/cavendish-bananas",
|
|
"133211",
|
|
),
|
|
("https://www.coles.com.au/product/coles-strawberries-250g-5191256", None),
|
|
]
|
|
|
|
for url, id in params:
|
|
self.assertEqual(woolworths.get_product_id(url), id)
|
|
|
|
async def test_get_strawberries(self) -> None:
|
|
details, raw_data = await woolworths.scrape("144607")
|
|
expected = {
|
|
"name": "Strawberries",
|
|
"quantity": 250,
|
|
"unit": "g Punnet",
|
|
"img_small": "https://cdn0.woolworths.media/content/wowproductimages/small/144607.jpg",
|
|
"img_large": "https://cdn0.woolworths.media/content/wowproductimages/large/144607.jpg",
|
|
}
|
|
|
|
for key, value in expected.items():
|
|
self.assertEqual(details[key], value, msg=key)
|
|
|
|
|
|
from products import coles
|
|
|
|
|
|
class TestColes(unittest.IsolatedAsyncioTestCase):
|
|
async def asyncSetUp(self) -> None:
|
|
local_path = "./tests/sample_files/coles"
|
|
coles._get_client = lambda: httpx_mocks.MockAsyncClient(local_path)
|
|
# coles._get_client = lambda: httpx_mocks.RecordingAsyncClient(local_path)
|
|
return await super().asyncSetUp()
|
|
|
|
async def test_get_product_id(self) -> None:
|
|
params = [
|
|
(
|
|
"https://www.coles.com.au/product/coles-strawberries-250g-5191256",
|
|
"coles-strawberries-250g-5191256",
|
|
),
|
|
(
|
|
"https://www.coles.com.au/product/coles-blueberries-170g-3571948",
|
|
"coles-blueberries-170g-3571948",
|
|
),
|
|
("https://www.woolworths.com.au/shop/productdetails/144607/strawberries", None),
|
|
]
|
|
|
|
for url, id in params:
|
|
self.assertEqual(coles.get_product_id(url), id)
|
|
|
|
async def test_get_strawberries(self) -> None:
|
|
details, raw_data = await coles.scrape("coles-strawberries-250g-5191256")
|
|
expected = {
|
|
"name": "Strawberries",
|
|
"quantity": 250,
|
|
"unit": "g",
|
|
"img_small": "https://shop.coles.com.au/wcsstore/Coles-CAS/images/5/1/9/5191256-th.jpg",
|
|
"img_large": "https://shop.coles.com.au/wcsstore/Coles-CAS/images/5/1/9/5191256.jpg",
|
|
}
|
|
|
|
for key, value in expected.items():
|
|
self.assertEqual(details[key], value, msg=key)
|