[IMP] check for existing fields

Author: nseinlet

requirements.txt

@@ -1,3 +1,3 @@
 odoo-client-lib==1.2.2
-locust==2.24.0
+locust==2.29.1
 names==0.3.0

setup.py

@@ -32,7 +32,7 @@
 from setuptools import setup, find_packages
 
 setup(name='OdooLocust',
-      version='1.6.4',
+      version='1.6.5',
       description='Easily load test Odoo using Locust and odoolib.',
       author='Nicolas Seinlet',
       author_email='',
@@ -41,7 +41,7 @@
       package_dir={'': 'src'},
       install_requires=[
           'odoo-client-lib>=1.2.2',
-          'locust>=2.24.0',
+          'locust>=2.29.1',
           'names>=0.3.0',
       ],
       long_description="See the home page for any information: https://github.com/odoo/OdooLocust.",

src/OdooLocust/OdooTaskSet.py

@@ -65,6 +65,10 @@ def _action_load(self, action_id, action_type=None):
             base_action = self.client.get_model('ir.actions.actions').read(action_id, ['type'])
             action_type = base_action[0]['type']
         return self.client.get_model(action_type).read(action_id, [])
+    
+    def _check_fields(self, model, fields_list):
+        all_fields = self.client.get_model(model).fields_get()
+        return [ f for f in fields_list if f in all_fields.keys() ]
 
 
 class OdooGenericTaskSet(OdooTaskSet):

src/OdooLocust/crm/lead.py

@@ -38,6 +38,16 @@ class CrmLead(OdooTaskSet):
     list_fields = ["stage_id", "active", "company_id", "calendar_event_count", "company_id", "user_company_ids", "date_deadline", "create_date", "name", "contact_name", "partner_name", "email_from", "phone", "city", "state_id", "country_id", "partner_id", "user_id", "team_id", "active", "campaign_id", "referred", "medium_id", "probability", "source_id", "message_needaction", "tag_ids", "priority"]
     form_fields = ["stage_id", "active", "company_id", "calendar_event_count", "sale_amount_total", "sale_order_count", "visitor_page_count", "duplicate_lead_count", "name", "company_currency", "expected_revenue", "recurring_revenue", "recurring_plan", "automated_probability", "is_automated_probability", "probability", "is_partner_visible", "partner_id", "partner_name", "street", "street2", "city", "state_id", "zip", "country_id", "website", "lang_active_count", "lang_code", "lang_id", "is_blacklisted", "partner_is_blacklisted", "phone_blacklisted", "mobile_blacklisted", "email_state", "phone_state", "partner_email_update", "partner_phone_update", "email_from", "phone", "lost_reason_id", "date_conversion", "user_company_ids", "contact_name", "title", "function", "mobile", "type", "user_id", "date_deadline", "priority", "tag_ids", "team_id", "lead_properties", "description", "campaign_id", "medium_id", "source_id", "referred", "date_open", "date_closed", "day_open", "day_close", "display_name"]
     random_id = -1
+    _model_id = -1
+
+    @property
+    def model_id(self):
+        if self._model_id < 0:            
+            irm = self.client.get_model('ir.model.data')
+            irm_id = irm.search_read([('model', '=', 'crm'), ('name', '=', 'model_crm_lead')], ['res_id'])
+            if irm_id:
+                self._model_id = irm_id[0]['res_id']
+        return self._model_id
 
     def on_start(self):
         super().on_start()
@@ -57,7 +67,8 @@ def test_searchread(self):
             limit=80,
             context=self.client.get_user_context()
         )
-        self.random_id = res[0]['id']
+        if res:
+            self.random_id = res[0]['id']
 
     @task(5)
     def test_websearchread(self):
@@ -68,7 +79,8 @@ def test_websearchread(self):
             limit=80,
             context=self.client.get_user_context()
         )
-        self.random_id = res[0]['id']
+        if res:
+            self.random_id = res[0]['id']
 
     @task(20)
     def test_read(self):
@@ -89,16 +101,14 @@ def lead_stage_change(self):
 
     @task(5)
     def test_activity(self):
-        model_model = self.client.get_model('ir.model')
-        crm_lead_model_id = model_model.search([('model', '=', 'crm.lead')], limit=1, context=self.client.get_user_context())
         activity_model = self.client.get_model('mail.activity')
         res = activity_model.create({
             "activity_type_id": 4,
             "calendar_event_id": False,
             "date_deadline": (date(2012, 1 ,1)+timedelta(days=random.randint(1,3650))).isoformat(),
             "note": False,
             "res_id": self.random_id,
-            "res_model_id": crm_lead_model_id[0],
+            "res_model_id": self.model_id,
             "summary": "Todo",
             "user_id": self.client.user_id,
         }, context=self.client.get_user_context())

src/OdooLocust/crm/quotation.py

@@ -34,7 +34,7 @@
 
 class SaleOrder(OdooTaskSet):
     list_fields = ["activity_exception_decoration", "activity_exception_icon", "activity_state", "activity_summary", "activity_type_icon", "activity_type_id", "name", "create_date", "commitment_date", "expected_date", "partner_id", "website_id", "user_id", "activity_ids", "team_id", "tag_ids", "company_id", "amount_untaxed", "amount_tax", "amount_total", "state", "validity_date", "invoice_status", "message_needaction", "currency_id"]
-    form_fields = ["authorized_transaction_ids", "is_subscription", "enterprise_is_new_plan", "state", "subscription_management", "is_upselling", "stage_category", "end_date", "archived_product_count", "partner_credit_warning", "project_ids", "is_product_milestone", "project_count", "milestone_count", "tasks_count", "timesheet_count", "timesheet_total_duration", "timesheet_encode_uom_id", "expense_count", "delivery_count", "id", "planning_first_sale_line_id", "planning_initial_date", "planning_hours_to_plan", "planning_hours_planned", "invoice_count", "attendee_count", "event_booth_count", "history_count", "recurring_monthly", "mrp_production_count", "database_count", "payment_exception", "to_renew", "renew_state", "name", "partner_id", "commission_plan_frozen", "referrer_id", "commission_plan_id", "delivery_set", "is_all_service", "recompute_delivery_price", "sale_order_template_id", "validity_date", "date_order", "recurrence_id", "invoice_date_diff", "next_invoice_date", "expected_next_invoice_date", "show_update_pricelist", "pricelist_id", "company_id", "currency_id", "tax_country_id", "payment_term_id", "new_sub_id", "old_sub_id", "order_line", "is_avatax", "note", "tax_totals", "commission", "sale_order_option_ids", "user_id", "team_id", "website_id", "is_abandoned_cart", "cart_recovery_email_sent", "require_signature", "require_payment", "website_description", "reference", "client_order_ref", "tag_ids", "start_date", "first_contract_date", "avatax_unique_code", "show_update_fpos", "fiscal_position_id", "partner_invoice_id", "analytic_account_id", "visible_project", "project_id", "invoice_status", "commercial_partner_id", "payment_token_id", "warehouse_id", "picking_policy", "commitment_date", "expected_date", "show_json_popover", "json_popover", "effective_date", "delivery_status", "origin", "campaign_id", "medium_id", "source_id", "internal_note", "enterprise_final_customer_id", "enterprise_security_email", "is_staff_restricted", "subscription_tag_ids", "enterprise_saas", "upsell_order_id", "upsell_date", "upsell_content", "maintenance_subscription_id", "maintenance_main_subscription_id", "maintenance_is_activated", "maintenance_stage_category", "odoo_sh_extra_gb", "odoo_sh_extra_stagings", "odoo_sh_extra_workers", "odoo_sh_extra_dedicated", "exclude_dashboard", "github_user_ids", "is_taxcloud", "is_taxcloud_configured", "display_name"]
+    form_fields = ["authorized_transaction_ids", "enterprise_is_new_plan", "state", "subscription_management", "is_upselling", "stage_category", "end_date", "archived_product_count", "partner_credit_warning", "project_ids", "is_product_milestone", "project_count", "milestone_count", "tasks_count", "timesheet_count", "timesheet_total_duration", "timesheet_encode_uom_id", "expense_count", "delivery_count", "id", "planning_first_sale_line_id", "planning_initial_date", "planning_hours_to_plan", "planning_hours_planned", "invoice_count", "attendee_count", "event_booth_count", "history_count", "recurring_monthly", "mrp_production_count", "database_count", "payment_exception", "to_renew", "renew_state", "name", "partner_id", "commission_plan_frozen", "referrer_id", "commission_plan_id", "delivery_set", "is_all_service", "recompute_delivery_price", "sale_order_template_id", "validity_date", "date_order", "recurrence_id", "invoice_date_diff", "next_invoice_date", "expected_next_invoice_date", "show_update_pricelist", "pricelist_id", "company_id", "currency_id", "tax_country_id", "payment_term_id", "new_sub_id", "old_sub_id", "order_line", "is_avatax", "note", "tax_totals", "commission", "sale_order_option_ids", "user_id", "team_id", "website_id", "is_abandoned_cart", "cart_recovery_email_sent", "require_signature", "require_payment", "website_description", "reference", "client_order_ref", "tag_ids", "start_date", "first_contract_date", "avatax_unique_code", "show_update_fpos", "fiscal_position_id", "partner_invoice_id", "analytic_account_id", "visible_project", "project_id", "invoice_status", "commercial_partner_id", "payment_token_id", "warehouse_id", "picking_policy", "commitment_date", "expected_date", "show_json_popover", "json_popover", "effective_date", "delivery_status", "origin", "campaign_id", "medium_id", "source_id", "internal_note", "enterprise_final_customer_id", "enterprise_security_email", "is_staff_restricted", "subscription_tag_ids", "enterprise_saas", "upsell_order_id", "upsell_date", "upsell_content", "maintenance_subscription_id", "maintenance_main_subscription_id", "maintenance_is_activated", "maintenance_stage_category", "odoo_sh_extra_gb", "odoo_sh_extra_stagings", "odoo_sh_extra_workers", "odoo_sh_extra_dedicated", "exclude_dashboard", "github_user_ids", "is_taxcloud", "is_taxcloud_configured", "display_name"]
     line_fields = ["analytic_precision", "sequence", "display_type", "product_uom_category_id", "product_type", "product_updatable", "product_id", "product_template_id", "product_template_attribute_value_ids", "product_custom_attribute_value_ids", "product_no_variant_attribute_value_ids", "is_configurable_product", "event_id", "event_ticket_id", "is_event_booth", "event_booth_category_id", "event_booth_pending_ids", "name", "temporal_type", "website_description", "analytic_distribution", "product_uom_qty", "qty_delivered", "virtual_available_at_date", "qty_available_today", "free_qty_today", "scheduled_date", "forecast_expected_date", "warehouse_id", "move_ids", "qty_to_deliver", "is_mto", "display_qty_widget", "qty_delivered_method", "qty_invoiced", "qty_to_invoice", "product_uom_readonly", "product_uom", "customer_lead", "recompute_delivery_price", "is_delivery", "price_unit", "tax_id", "price_tax", "discount", "is_downpayment", "price_subtotal", "state", "invoice_status", "currency_id", "company_id", "database_id", "computation"]
     random_id = 2795659
 
@@ -76,7 +76,8 @@ def test_set_to_quotation(self):
     def test_quotation_confirm(self):
         saleorder_model = self.client.get_model('sale.order')
         res = saleorder_model.search([['state', '=', 'draft'], '!', [ 'order_line.product_id.active', '=', False ], [ 'partner_id.country_id', '!=', False], [ 'company_id', '=', 1]], context=self.client.get_user_context())
-        saleorder_model.action_confirm(random.choice(res), context=self.client.get_user_context())
+        if res:
+            saleorder_model.action_confirm(random.choice(res), context=self.client.get_user_context())
 
     @task
     def test_quotation_sendemail(self):