diff options
Diffstat (limited to 'addons/product/tests/test_product_attribute_value_config.py')
| -rw-r--r-- | addons/product/tests/test_product_attribute_value_config.py | 166 |
1 files changed, 166 insertions, 0 deletions
diff --git a/addons/product/tests/test_product_attribute_value_config.py b/addons/product/tests/test_product_attribute_value_config.py new file mode 100644 index 00000000000..8a290aa9c02 --- /dev/null +++ b/addons/product/tests/test_product_attribute_value_config.py @@ -0,0 +1,166 @@ +# -*- coding: utf-8 -*- +# Part of Odoo. See LICENSE file for full copyright and licensing details. + +from odoo.tests.common import TransactionCase + + +class TestProductAttributeValueConfig(TransactionCase): + def setUp(self): + super(TestProductAttributeValueConfig, self).setUp() + + self.computer = self.env['product.template'].create({ + 'name': 'Super Computer' + }) + + self._add_ssd_attribute() + self._add_ram_attribute() + self._add_hdd_attribute() + + self.computer.create_variant_ids() + + self.computer_case = self.env['product.template'].create({ + 'name': 'Super Computer Case' + }) + + self._add_size_attribute() + + self.computer_case.create_variant_ids() + + def test_product_product_attribute_values_creation(self): + self.assertEqual(len(self.computer_ssd_attribute_lines.product_value_ids), 2, 'Product attribute values (ssd) were not automatically created') + self.assertEqual(len(self.computer_ram_attribute_lines.product_value_ids), 3, 'Product attribute values (ram) were not automatically created') + self.assertEqual(len(self.computer_hdd_attribute_lines.product_value_ids), 3, 'Product attribute values (hdd) were not automatically created') + self.assertEqual(len(self.computer_case_size_attribute_lines.product_value_ids), 3, 'Product attribute values (size) were not automatically created') + + def test_product_filtered_exclude_for(self): + """ + Super Computer has 18 variants total (2 ssd * 3 ram * 3 hdd) + RAM 16 excudes HDD 1, that matches 2 variants: + - SSD 256 RAM 16 HDD 1 + - SSD 512 RAM 16 HDD 1 + + => There has to be 16 variants left when filtered + """ + self._add_ram_exclude_for() + self.assertEqual(len(self.computer.get_filtered_variants()), 16) + self.assertFalse(self._get_variant_for_attribute_values(self.computer, [self.ssd_256, self.ram_16, self.hdd_1])) + self.assertFalse(self._get_variant_for_attribute_values(self.computer, [self.ssd_512, self.ram_16, self.hdd_1])) + + def test_children_product_filtered_exclude_for(self): + """ + Super Computer Case has 3 variants total (3 size) + Reference product Computer with HDD 4 excludes Size M + The following variant will be excluded: + - Size M + + => There has to be 2 variants left when filtered + """ + + self._add_hdd_excludes_computer_case() + hdd_4_variant = self.computer.product_variant_ids.filtered( + lambda variant: self.hdd_4 in variant.product_attribute_value_ids.mapped('product_attribute_value_id'))[0] + self.assertEqual(len(self.computer_case.get_filtered_variants(hdd_4_variant)), 2) + self.assertFalse(self._get_variant_for_attribute_values(self.computer_case, [self.size_m], hdd_4_variant)) + + def _add_ssd_attribute(self): + self.ssd_attribute = self.env['product.attribute'].create({'name': 'Memory'}) + self.ssd_256 = self.env['product.attribute.value'].create({ + 'name': '256 GB', + 'attribute_id': self.ssd_attribute.id + }) + self.ssd_512 = self.env['product.attribute.value'].create({ + 'name': '512 GB', + 'attribute_id': self.ssd_attribute.id + }) + self.computer_ssd_attribute_lines = self.env['product.attribute.line'].create({ + 'product_tmpl_id': self.computer.id, + 'attribute_id': self.ssd_attribute.id, + 'value_ids': [(6, 0, [self.ssd_256.id, self.ssd_512.id])] + }) + + def _add_ram_attribute(self): + self.ram_attribute = self.env['product.attribute'].create({'name': 'RAM'}) + self.ram_8 = self.env['product.attribute.value'].create({ + 'name': '8 GB', + 'attribute_id': self.ram_attribute.id + }) + self.ram_16 = self.env['product.attribute.value'].create({ + 'name': '16 GB', + 'attribute_id': self.ram_attribute.id + }) + self.ram_32 = self.env['product.attribute.value'].create({ + 'name': '32 GB', + 'attribute_id': self.ram_attribute.id + }) + self.computer_ram_attribute_lines = self.env['product.attribute.line'].create({ + 'product_tmpl_id': self.computer.id, + 'attribute_id': self.ram_attribute.id, + 'value_ids': [(6, 0, [self.ram_8.id, self.ram_16.id, self.ram_32.id])] + }) + + def _add_hdd_attribute(self): + self.hdd_attribute = self.env['product.attribute'].create({'name': 'HDD'}) + self.hdd_1 = self.env['product.attribute.value'].create({ + 'name': '1 To', + 'attribute_id': self.hdd_attribute.id + }) + self.hdd_2 = self.env['product.attribute.value'].create({ + 'name': '2 To', + 'attribute_id': self.hdd_attribute.id + }) + self.hdd_4 = self.env['product.attribute.value'].create({ + 'name': '4 To', + 'attribute_id': self.hdd_attribute.id + }) + self.computer_hdd_attribute_lines = self.env['product.attribute.line'].create({ + 'product_tmpl_id': self.computer.id, + 'attribute_id': self.hdd_attribute.id, + 'value_ids': [(6, 0, [self.hdd_1.id, self.hdd_2.id, self.hdd_4.id])] + }) + + def _add_ram_exclude_for(self): + self.ram_16_excludes_hdd_1 = self.env['product.attribute.filter.line'].create({ + 'product_tmpl_id': self.computer.id, + 'value_ids': [(6, 0, [self._get_product_value_id(self.computer_hdd_attribute_lines, self.hdd_1).id])] + }) + self._get_product_value_id(self.computer_ram_attribute_lines, self.ram_16).update({ + 'exclude_for': [(6, 0, [self.ram_16_excludes_hdd_1.id])] + }) + + def _add_size_attribute(self): + self.size_attribute = self.env['product.attribute'].create({'name': 'Size'}) + self.size_m = self.env['product.attribute.value'].create({ + 'name': 'M', + 'attribute_id': self.size_attribute.id + }) + self.size_l = self.env['product.attribute.value'].create({ + 'name': 'L', + 'attribute_id': self.size_attribute.id + }) + self.size_xl = self.env['product.attribute.value'].create({ + 'name': 'XL', + 'attribute_id': self.size_attribute.id + }) + self.computer_case_size_attribute_lines = self.env['product.attribute.line'].create({ + 'product_tmpl_id': self.computer_case.id, + 'attribute_id': self.size_attribute.id, + 'value_ids': [(6, 0, [self.size_m.id, self.size_l.id, self.size_xl.id])] + }) + + def _add_hdd_excludes_computer_case(self): + self.hdd_4_excludes_computer_case_m = self.env['product.attribute.filter.line'].create({ + 'product_tmpl_id': self.computer_case.id, + 'value_ids': [(6, 0, [self._get_product_value_id(self.computer_case_size_attribute_lines, self.size_m).id])] + }) + self._get_product_value_id(self.computer_hdd_attribute_lines, self.hdd_4).update({ + 'exclude_for': [(6, 0, [self.hdd_4_excludes_computer_case_m.id])] + }) + + def _get_product_value_id(self, product_attribute_lines, product_attribute_value): + return product_attribute_lines.product_value_ids.filtered( + lambda product_value_id: product_value_id.product_attribute_value_id == product_attribute_value)[0] + + def _get_variant_for_attribute_values(self, product, attribute_values, reference_product=None): + return product.get_filtered_variants(reference_product).filtered( + lambda variant: + all(attribute_value in variant.attribute_value_ids for attribute_value in attribute_values)) |