Create custom contact properties and verify replication of those properties

tests/client.py

@@ -7,6 +7,7 @@
 import requests
 from base import HubspotBaseTest
 from tap_tester import LOGGER
+from copy import deepcopy
 
 DEBUG = False
 BASE_URL = "https://api.hubapi.com"
@@ -37,6 +38,7 @@ def giveup(exc):
                           jitter=None,
                           giveup=giveup,
                           interval=10)
+
     def get(self, url, params=dict()):
         """Perform a GET using the standard requests method and logs the action"""
         response = requests.get(url, params=params, headers=self.HEADERS)
@@ -833,6 +835,107 @@ def create(self, stream, company_ids=[], subscriptions=[], times=1):
         else:
             raise NotImplementedError(f"There is no create_{stream} method in this dipatch!")
 
+    def create_custom_contact_properties(self):
+        """Create custom contact properties of all the types"""
+
+        url = f"{BASE_URL}/properties/v1/contacts/properties"
+        data = []
+        property = {
+            "name": "custom_string",
+            "label": "A New String Custom Property",
+            "description": "A new string property for you",
+            "groupName": "contactinformation",
+            "type": "string",
+            "fieldType": "text",
+            "formField": True,
+            "displayOrder": 6,
+            "options": [
+            ]
+        }
+        data.append(deepcopy(property))
+
+        property = {
+            "name": "custom_number",
+            "label": "A New Number Custom Property",
+            "description": "A new number property for you",
+            "groupName": "contactinformation",
+            "type": "number",
+            "fieldType": "text",
+            "formField": True,
+            "displayOrder": 7,
+            "options": [
+            ]
+        }
+        data.append(deepcopy(property))
+
+        property = {
+            "name": "custom_date",
+            "label": "A New Date Custom Property",
+            "description": "A new date property for you",
+            "groupName": "contactinformation",
+            "type": "date",
+            "fieldType": "text",
+            "formField": True,
+            "displayOrder": 9,
+            "options": [
+            ]
+        }
+        data.append(deepcopy(property))
+
+        property = {
+            "name": "custom_datetime",
+            "label": "A New Datetime Custom Property",
+            "description": "A new datetime property for you",
+            "groupName": "contactinformation",
+            "type": "datetime",
+            "fieldType": "text",
+            "formField": True,
+            "displayOrder": 10,
+            "options": [
+            ]
+        }
+        data.append(deepcopy(property))
+
+        property = {
+            "name": "multi_pick",
+            "label": "multi pick",
+            "description": "multi select picklist test",
+            "groupName": "contactinformation",
+            "type": "enumeration",
+            "fieldType": "checkbox",
+            "hidden": False,
+            "options": [
+              {
+                "label": "Option A",
+                "value": "option_a"
+              },
+              {
+                "label": "Option B",
+                "value": "option_b"
+              },
+              {
+                "label": "Option C",
+                "value": "option_c"
+              }
+            ],
+            "formField": True
+        }
+        data.append(deepcopy(property))
+        # generate a contacts record
+
+        for current_data in data:
+            try:
+                response = self.post(url, current_data)
+                LOGGER.info("response is %s", response)
+            # Setting up the property is a one time task, If exception occurs because, it already exists, ignore
+            except Exception as DataAlreadyExistsExcp:
+                LOGGER.info("Data already exists for %s", current_data)
+                if '409' in str(DataAlreadyExistsExcp):
+                    pass
+                else:
+                    response.raise_for_status()
+
+
     def create_contacts(self):
         """
         Generate a single contacts record.
@@ -843,6 +946,14 @@ def create_contacts(self):
         url = f"{BASE_URL}/contacts/v1/contact"
         data = {
             "properties": [
+                {
+                    "property": "custom_string",
+                    "value": "custom_string_value"
+                },
+                {
+                    "property": "custom_number",
+                    "value": 1567
+                },
                 {
                     "property": "email",
                     "value": f"{record_uuid}@stitchdata.com"

tests/test_hubspot_all_fields.py

@@ -292,7 +292,7 @@ def test_run(self):
                         #     to our test data. We have determined that the filtering of these fields is an expected behavior.
 
                         # deals workaround for 'property_hs_date_entered_<property>' fields
-                        bad_key_prefixes = {'property_hs_date_entered_', 'property_hs_date_exited_',
+                        bad_key_prefixes = {'property_hs_date_entered_', 'property_hs_v2_date_entered', 'property_hs_date_exited_',
                                             'property_hs_time_in'}
                         bad_keys = set()
                         for key in expected_keys_adjusted:

tests/test_hubspot_bookmarks.py

@@ -45,6 +45,8 @@ def setUp(self):
         self.maxDiff = None  # see all output in failure
 
         self.test_client = TestClient(self.get_properties()['start_date'])
+        # Create custom properties for contacts
+        self.test_client.create_custom_contact_properties()
 
     def create_test_data(self, expected_streams):
         """