def test_code_with_multiple_company(self): current_ean = TEST_EAN13 company1 = CompanyFactory.create(name='test-company1', description='test-description') company2 = CompanyFactory.create(name='test-company2', description='test-description') product = ProductFactory.create(code=current_ean, company=company1, brand__company=company2) with mock.patch("pola.logic.get_by_code", return_value=product): response = get_result_from_code(current_ean) # TODO: Add support for multiple companies in this response expected_response = ( { 'altText': None, 'card_type': 'type_grey', 'code': '5900084231145', 'description': 'test-description', 'is_friend': False, 'name': company1.official_name, 'plCapital': None, 'plCapital_notes': None, 'plNotGlobEnt': None, 'plNotGlobEnt_notes': None, 'plRegistered': None, 'plRegistered_notes': None, 'plRnD': None, 'plRnD_notes': None, 'plScore': None, 'plWorkers': None, 'plWorkers_notes': None, 'product_id': product.id, 'report_button_text': 'Zgłoś', 'report_button_type': 'type_white', 'report_text': ('Zgłoś jeśli posiadasz bardziej aktualne dane na temat tego produktu'), 'sources': {}, }, {"was_590": True, "was_plScore": False, "was_verified": False}, product, ) self.maxDiff = None self.assertEqual(expected_response[0], response[0]) self.assertEqual(expected_response, response)
def test_should_return_200_when_polish_and_known_product(self): c = CompanyFactory.create( plCapital=100, plWorkers=0, plRnD=100, plRegistered=100, plNotGlobEnt=100, description="KOTEK", sources="KOTEK|BBBB", verified=True, is_friend=True, plCapital_notes="AA", plWorkers_notes="BBB", plRnD_notes="CCC", plRegistered_notes="DDD", plNotGlobEnt_notes="EEEE", ) p = ProductFactory.create(code=5900049011829, company=c, brand=None) response = self.json_request(self.url + "?device_id=TEST-DEVICE-ID&code=" + str(p.code)) self.assertEqual(200, response.status_code)
def setUp(self): self.aem_admin = UserFactory.create(group=AemGroupFactory.create( slug_field=settings.AEM_ADMIN_SLUG_FIELD, linked_group__name=settings.AEM_ADMIN_LINKED_GROUP_NAME, can_add_permission_slugs=settings. AEM_ADMIN_CAN_ADD_USER_PERMISSIONS, client_permissions=settings.AEM_ADMIN_CLIENT_PERMISSIONS, customer_permissions=settings.AEM_ADMIN_CUSTOMER_PERMISSIONS, company_permissions=settings.AEM_ADMIN_COMPANY_PERMISSIONS)) self.aem_employee = UserFactory.create(group=AemGroupFactory.create( slug_field=settings.AEM_EMPLOYEE_SLUG_FIELD, linked_group__name=settings.AEM_EMPLOYEE_LINKED_GROUP_NAME, can_add_permission_slugs=settings. AEM_EMPLOYEE_CAN_ADD_USER_PERMISSIONS, client_permissions=settings.AEM_EMPLOYEE_CLIENT_PERMISSIONS, customer_permissions=settings.AEM_EMPLOYEE_CUSTOMER_PERMISSIONS, company_permissions=settings.AEM_EMPLOYEE_COMPANY_PERMISSIONS)) client_company = CompanyFactory.create() self.aem_customer_super_user = UserFactory.create( company=client_company, group=AemGroupFactory.create( slug_field=settings.AEM_CUSTOMER_SUPER_USER_SLUG_FIELD, linked_group__name=settings. AEM_CUSTOMER_SUPER_USER_LINKED_GROUP_NAME, can_add_permission_slugs=settings. AEM_CUSTOMER_SUPER_USER_CAN_ADD_USER_PERMISSIONS, client_permissions=settings. AEM_CUSTOMER_SUPER_USER_CLIENT_PERMISSIONS, customer_permissions=settings. AEM_CUSTOMER_SUPER_USER_CUSTOMER_PERMISSIONS, company_permissions=settings. AEM_CUSTOMER_SUPER_USER_COMPANY_PERMISSIONS)) self.aem_customer_admin = UserFactory.create( company=client_company, group=AemGroupFactory.create( slug_field=settings.AEM_CUSTOMER_ADMIN_SLUG_FIELD, linked_group__name=settings. AEM_CUSTOMER_ADMIN_LINKED_GROUP_NAME, can_add_permission_slugs=settings. AEM_CUSTOMER_ADMIN_CAN_ADD_USER_PERMISSIONS, client_permissions=settings. AEM_CUSTOMER_ADMIN_CLIENT_PERMISSIONS, customer_permissions=settings. AEM_CUSTOMER_ADMIN_CUSTOMER_PERMISSIONS, company_permissions=settings. AEM_CUSTOMER_ADMIN_COMPANY_PERMISSIONS)) self.aem_customer_user = UserFactory.create( company=client_company, group=AemGroupFactory.create( slug_field=settings.AEM_CUSTOMER_USER_SLUG_FIELD, linked_group__name=settings. AEM_CUSTOMER_USER_LINKED_GROUP_NAME, can_add_permission_slugs=settings. AEM_CUSTOMER_USER_CAN_ADD_USER_PERMISSIONS, client_permissions=settings. AEM_CUSTOMER_USER_CLIENT_PERMISSIONS, customer_permissions=settings. AEM_CUSTOMER_USER_CUSTOMER_PERMISSIONS, company_permissions=settings. AEM_CUSTOMER_USER_COMPANY_PERMISSIONS))