TDL-15317: Fix Primary Key for Feature Events

.circleci/config.yml

@@ -27,7 +27,7 @@ jobs:
           name: 'Unit Tests'
           command: |
             source /usr/local/share/virtualenvs/tap-pendo/bin/activate
-            pip install coverage
+            pip install nose coverage
             nosetests --with-coverage --cover-erase --cover-package=tap_pendo --cover-html-dir=htmlcov tests/unittests
             coverage html
       - store_test_results:

README.md

@@ -95,7 +95,7 @@ This tap:
 **[feature_events](https://developers.pendo.io/docs/?bash#get-an-account-by-id)**
 
 - Endpoint: [https://api/v1/aggregation](https://app.pendo.io/api/v1/aggregation)
-- Primary key fields: `visitor_id`, `account_id`, `server`, `remote_ip`
+- Primary key fields: `feature_id`, `visitor_id`, `account_id`, `server`, `remote_ip`, `user_agent`, `day` or `hour`
 - Replication strategy: INCREMENTAL (query filtered)
   - Bookmark: `day` or `hour`
 - Transformations: Camel to snake case.

tap_pendo/streams.py

@@ -551,7 +551,11 @@ def get_body(self):
 class FeatureEvents(EventsBase):
     name = "feature_events"
     replication_method = "INCREMENTAL"
-    key_properties = ['visitor_id', 'account_id', 'server', 'remote_ip']
+    key_properties = ['feature_id', 'visitor_id', 'account_id', 'server', 'remote_ip', 'user_agent']
+
+    def __init__(self, config):
+        super().__init__(config=config)
+        self.key_properties.append("day" if self.period == 'dayRange' else "hour")
 
     def get_body(self, key_id, period, first):
         return {

tests/tap_tester/base.py

@@ -22,6 +22,7 @@ class TestPendoBase(unittest.TestCase):
     START_DATE_FORMAT = "%Y-%m-%dT00:00:00Z"
     BOOKMARK_COMPARISON_FORMAT = "%Y-%m-%dT%H:%M%S%z"
     start_date = ""
+    is_day_range = True
 
     @staticmethod
     def name():
@@ -39,6 +40,7 @@ def get_type():
 
     def expected_metadata(self):
         """The expected streams and metadata about the streams"""
+        event_replication_key = 'day' if self.is_day_range else 'hour'
         return {
             "accounts": {
                 self.PRIMARY_KEYS: {'account_id'},
@@ -79,19 +81,19 @@ def expected_metadata(self):
                 self.REPLICATION_KEYS: {'last_updated_at'}
             },
             "feature_events":{
-                self.PRIMARY_KEYS:  {"visitor_id", "account_id", "server", "remote_ip"},
+                self.PRIMARY_KEYS: {"feature_id", "visitor_id", "account_id", "server", "remote_ip", "user_agent", event_replication_key},
                 self.REPLICATION_METHOD: self.INCREMENTAL,
-                self.REPLICATION_KEYS: {'day'}
+                self.REPLICATION_KEYS: {event_replication_key}
             },
             "events": {
                 self.PRIMARY_KEYS: {"visitor_id", "account_id", "server", "remote_ip"},
                 self.REPLICATION_METHOD: self.INCREMENTAL,
-                self.REPLICATION_KEYS: {'day'}
+                self.REPLICATION_KEYS: {event_replication_key}
             },
             "page_events": {
                 self.PRIMARY_KEYS: {"visitor_id", "account_id", "server", "remote_ip"},
                 self.REPLICATION_METHOD: self.INCREMENTAL,
-                self.REPLICATION_KEYS: {'day'}
+                self.REPLICATION_KEYS: {event_replication_key}
             },
             "guide_events": {
                 self.PRIMARY_KEYS: {"visitor_id", "account_id", "server_name", "remote_ip"},
@@ -106,7 +108,7 @@ def expected_metadata(self):
             "track_events": {
                 self.PRIMARY_KEYS: {"visitor_id", "account_id", "server", "remote_ip"},
                 self.REPLICATION_METHOD: self.INCREMENTAL,
-                self.REPLICATION_KEYS: {'day'}
+                self.REPLICATION_KEYS: {event_replication_key}
             },
             "metadata_accounts": {
                 self.REPLICATION_METHOD: self.FULL_TABLE,
@@ -136,7 +138,7 @@ def get_properties(self, original: bool = True):
         return_value = {
             "start_date": "2020-09-10T00:00:00Z",
             "lookback_window": "1",
-            "period": "dayRange",
+            "period": "dayRange" if self.is_day_range else "hourRange",
         }
         if original:
             return return_value

tests/tap_tester/test_discovery.py

@@ -23,7 +23,7 @@ class PendoDiscoverTest(TestPendoBase):
     def name(self):
         return "pendo_discover_test"
 
-    def test_run(self):
+    def discovery_test_run(self):
         streams_to_test = self.expected_streams()
 
         conn_id = connections.ensure_connection(self, payload_hook=None)
@@ -123,3 +123,13 @@ def test_run(self):
                          and item.get("breadcrumb", ["properties", None])[1]
                          not in actual_automatic_fields}),
                     msg="Not all non key properties are set to available in metadata")
+
+    def test_run(self):
+
+        #Discovery test for hourRange period
+        self.is_day_range = False
+        self.discovery_test_run()
+
+        #Discovery test for dayRange period
+        self.is_day_range = True
+        self.discovery_test_run()

tests/unittests/test_feature_events_primary_keys.py

@@ -0,0 +1,30 @@
+import unittest
+from tap_pendo.streams import FeatureEvents
+
+class TestFeatureEventsPrimaryKeys(unittest.TestCase):
+
+    def test_feature_event_primary_key_with_hourRange(self):
+        '''
+            Verify that primary keys should have expected fields with 'hour' field when period is hourRange 
+        '''
+        # Reset key properties to default value
+        FeatureEvents.key_properties = ['feature_id', 'visitor_id', 'account_id', 'server', 'remote_ip', 'user_agent']
+        config = {"period": "hourRange"} # set hourRange as a period
+        expected_primary_keys = ["feature_id", "visitor_id", "account_id", "server", "remote_ip", "user_agent", "hour"]
+
+        feature_event_stream1 = FeatureEvents(config) # Initialize FeatuereEvents object which sets primary keys
+
+        self.assertEqual(feature_event_stream1.key_properties, expected_primary_keys)
+
+    def test_feature_event_primary_key_with_dayRange(self):
+        '''
+            Verify that primary keys should have expected fields with 'day' field when period is dayRange 
+        '''
+        # Reset key properties to default value
+        FeatureEvents.key_properties = ['feature_id', 'visitor_id', 'account_id', 'server', 'remote_ip', 'user_agent']
+        config = {"period": "dayRange"} # set dayRange as a period
+        expected_primary_keys = ["feature_id", "visitor_id", "account_id", "server", "remote_ip", "user_agent", "day"]  
+
+        feature_event_stream2 = FeatureEvents(config) # Initialize Events object which sets primary keys
+
+        self.assertEqual(feature_event_stream2.key_properties, expected_primary_keys)