Fix: MACHETE test_get_products lock_table

modules/sale_don_confiao/tests/test_api_donconfiao.py

@@ -1,17 +1,15 @@
-from trytond.tests.test_routes import RoutesTestCase
-from trytond.tests.test_tryton import Client
-from trytond.protocols.wrappers import Response
+from trytond.tests.test_tryton import RouteTestCase
 from trytond.pool import Pool
-from trytond.wsgi import app
-
 from http import HTTPStatus
-
 import uuid
+import json
+from trytond.tests.test_tryton import DB_NAME
+from trytond.transaction import Transaction
 
 
 class DonConfiaoApiRouteTestCase(
-        RoutesTestCase):
-    "Speco API Routes"
+        RouteTestCase):
+    "Don Confiao API Routes"
 
     module = 'sale_don_confiao'
 
@@ -21,23 +19,72 @@ class DonConfiaoApiRouteTestCase(
     def setUpDatabase(cls):
         pool = Pool()
         Application = pool.get('res.user.application')
+        Company = pool.get('company.company')
+        Party = pool.get('party.party')
+
+        # Product = pool.get('product.product')
+
         Application(
             key=cls.key,
             user=1,
             application='sale_don_confiao',
             state='validated').save()
 
+        customer = Party()
+        customer.name = 'Dunkan'
+        customer.save()
+
+        company = Company(
+            party=1,
+            currency=1
+        )
+        company.save()
+
     def test_get_parties(self):
-        client = Client(app, Response)
+        client = self.client()
         response = client.get(
-            '/parties',
+            f'/{self.db_name}/sale_don_confiao/parties',
             headers={
                 'Authorization': f'bearer {self.key}',
             })
+
+        parties = json.loads(
+            response.get_data().decode())
+
         self.assertEqual(response.status_code, HTTPStatus.OK)
-        # response = client.get(
-        #     '/parties',
-        #     headers={
-        #         'Authorization': f'bearer {self.key}',
-        #     })
-        # self.assertEqual(response.status_code, HTTPStatus.OK)
+        self.assertEqual(len(parties), 1)
+
+    def test_get_products(self):
+        pool = Pool(DB_NAME)
+        transaction = Transaction().start(DB_NAME, 0, _lock_tables=[
+            'product_product'])
+        with transaction:
+            # MACHETE: Se bloquea la tabla manualmente
+            # transaction._locked_tables = ['product_product']
+            User = pool.get('res.user')
+            admin, = User.search([('login', '=', 'admin')])
+            admin.password = 'password'
+            admin.save()
+
+            ProductTemplate = pool.get('product.template')
+            Product = pool.get('product.product')
+            Uom = pool.get('product.uom')
+            template, = ProductTemplate.create([
+                {'name': 'Product',
+                 'default_uom': Uom.search([('name', '=', 'Unit')])[0].id
+                 }
+            ])
+
+            Product.create([{'template': template.id}])
+
+        client = self.client()
+        response = client.get(
+            f'/{self.db_name}/sale_don_confiao/products',
+            headers={
+                'Authorization': f'bearer {self.key}',
+            })
+
+        products = json.loads(
+            response.get_data().decode())
+        self.assertEqual(response.status_code, HTTPStatus.OK)
+        self.assertEqual(len(products), 1)