Пример #1
0
 def test_get_url_for_product_element(self):
     """
     Test url getters for product element.
     :return:
     """
     url = Product.get_element_api_url(element_id=self.product_id)
     self.assertIsNotNone(url)
Пример #2
0
    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)
Пример #3
0
 def test_get_data_for_product_element(self):
     """
     Test data getters for product element.
     :return:
     """
     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)
         self.assertIsInstance(data, dict)
Пример #4
0
    def test_get_data_for_product_element(self):
        """
        Test data writer for product element.
        :return:
        """
        # file_path = os.path.join(settings.JSON_DIR_PATH, "%s.json" % Product._meta.model_name)

        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)
            file = Product.write_api_data(data)
            self.assertTrue(os.path.exists(file))