updated unit_tests

airbytehq · Jun 15, 2022 · c86f971 · c86f971
1 parent 7d0d0ea
commit c86f971
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/airbyte-integrations/connectors/source-zendesk-support/unit_tests/unit_test.py b/airbyte-integrations/connectors/source-zendesk-support/unit_tests/unit_test.py
@@ -15,7 +15,7 @@
 import requests
 from airbyte_cdk import AirbyteLogger
 from source_zendesk_support.source import BasicApiTokenAuthenticator, SourceZendeskSupport
-from source_zendesk_support.streams import (  # streams
+from source_zendesk_support.streams import (
     DATETIME_FORMAT,
     END_OF_STREAM_KEY,
     LAST_END_TIME_KEY,
@@ -406,14 +406,12 @@ def test_get_updated_state(self, stream_cls, current_state, last_record, expecte
             (Macros, None),
             (Organizations, None),
             (Groups, None),
-            (SatisfactionRatings, None),
             (TicketFields, None),
         ],
         ids=[
             "Macros",
             "Organizations",
             "Groups",
-            "SatisfactionRatings",
             "TicketFields",
         ],
     )
@@ -428,17 +426,13 @@ def test_next_page_token(self, stream_cls, expected):
             (Macros, {"start_time": 1622505600}),
             (Organizations, {"start_time": 1622505600}),
             (Groups, {"start_time": 1622505600}),
-            (SatisfactionRatings, {"start_time": 1622505600, "sort_by": "asc"}),
             (TicketFields, {"start_time": 1622505600}),
-            (TicketMetrics, {"start_time": 1622505600}),
         ],
         ids=[
             "Macros",
             "Organizations",
             "Groups",
-            "SatisfactionRatings",
             "TicketFields",
-            "TicketMetrics",
         ],
     )
     def test_request_params(self, stream_cls, expected):
@@ -554,13 +548,15 @@ def test_get_updated_state(self, stream_cls, current_state, last_record, expecte
             (TicketMetricEvents),
             (TicketAudits),
             (TicketMetrics),
+            (SatisfactionRatings),
         ],
         ids=[
             "GroupMemberships",
             "TicketForms",
             "TicketMetricEvents",
             "TicketAudits",
             "TicketMetrics",
+            "SatisfactionRatings",
         ],
     )
     def test_next_page_token(self, requests_mock, stream_cls):
@@ -598,12 +594,16 @@ def test_check_stream_state(self, stream_cls, expected):
             (TicketForms, {"start_time": 1622505600}),
             (TicketMetricEvents, {"start_time": 1622505600}),
             (TicketAudits, {"sort_by": "created_at", "sort_order": "desc", "limit": 1000}),
+            (SatisfactionRatings, {'page': 1, 'per_page': 100, 'sort_by': 'asc', 'start_time': 1622505600}),
+            (TicketMetrics, {'page': 1, 'per_page': 100, 'start_time': 1622505600}),
         ],
         ids=[
             "GroupMemberships",
             "TicketForms",
             "TicketMetricEvents",
             "TicketAudits",
+            "SatisfactionRatings",
+            "TicketMetrics",
         ],
     )
     def test_request_params(self, stream_cls, expected):