def test_insert_product_element(self):
"""
Test insert products element.
:return:
"""
data = get_categories_data_with_mock(self.fake_data_path, 'categories_short')
Category.insert_data(data)
nb_elements_before = Product.objects.count()
with open(os.path.join(self.fake_data_path, '3222472887966.json'), 'r') as file:
fake_data_product = file.read()
with requests_mock.Mocker() as mock:
mock.get(Product.get_element_api_url(
self.product_id), text=fake_data_product)
data = Product.get_api_data_element(self.product_id)
# test create
Product.insert_data(data)
nb_elements_after = Product.objects.count()
self.assertGreater(nb_elements_after, nb_elements_before)
# test update
Product.insert_data(data)
new_nb_elements_after = Product.objects.count()
self.assertEqual(new_nb_elements_after, nb_elements_after)
def get_categories(from_cache: bool = False):
"""
method in charge of getting categories.
:param from_cache: indicates if data should be recovered from cache or not
:type from_cache: bool
"""
data = Category.get_api_data_list(from_cache=from_cache)
Category.insert_data(data)
def setUp(self):
self.fake_data_path = os.path.join(
os.path.dirname(__file__), 'fake_data')
data = get_categories_data_with_mock(self.fake_data_path, 'categories_short')
Category.insert_data(data)
data = get_products_data_with_mock(self.fake_data_path, 'products', nb_files=3)
Product.insert_data(data)
def test_insert_category_list(self):
"""
Test insert category list.
:return:
"""
nb_elements_before = Category.objects.count()
data = get_categories_data_with_mock(self.fake_data_path, 'categories_short')
Category.insert_data(data)
nb_elements_after = Category.objects.count()
self.assertGreater(nb_elements_after, nb_elements_before)
def test_insert_product_list(self):
"""
Test insert product list.
:return:
"""
data = get_categories_data_with_mock(self.fake_data_path, 'categories_short')
Category.insert_data(data)
nb_elements_before = Product.objects.count()
data = get_products_data_with_mock(self.fake_data_path, 'products', nb_files=3)
# test create
Product.insert_data(data)
nb_elements_after = Product.objects.count()
self.assertGreater(nb_elements_after, nb_elements_before)
# test update
Product.insert_data(data)
new_nb_elements_after = Product.objects.count()
self.assertEqual(new_nb_elements_after, nb_elements_after)
def test_insert_product_list_with_strict_mode(self):
"""
Test insert products list with strict mode.
:return:
"""
data = get_categories_data_with_mock(self.fake_data_path, 'categories_short')
Category.insert_data(data)
nb_elements_before = Product.objects.count()
data = get_products_data_with_mock(self.fake_data_path, 'products', nb_files=3)
# test create
Product.insert_data(data, strict_required_field_mode=True)
nb_elements_after = Product.objects.count()
self.assertGreater(nb_elements_after, nb_elements_before)
# test update
Product.insert_data(data)
new_nb_elements_after = Product.objects.count()
self.assertGreater(new_nb_elements_after, nb_elements_after)