From be4b499d174b5f0e9329037ff7ebab3f297f7900 Mon Sep 17 00:00:00 2001 From: Bhuvana Venkataraman Date: Wed, 25 Oct 2023 18:09:07 +0000 Subject: [PATCH] Moved the changes to client.py --- tests/test_hubspot_start_date.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/tests/test_hubspot_start_date.py b/tests/test_hubspot_start_date.py index 16aa907e..5a6f5d3f 100644 --- a/tests/test_hubspot_start_date.py +++ b/tests/test_hubspot_start_date.py @@ -28,16 +28,13 @@ def setUp(self): streams_under_test = self.expected_streams() - {'email_events'} # we get this for free with subscription_changes self.my_start_date = self.get_properties()['start_date'] self.test_client = TestClient(self.my_start_date) - - self.page_size = dict() for stream in streams_under_test: - #self.page_size[stream] = self.expected_metadata().get(stream).get('expected_page_size') if stream == 'contacts_by_company': - companies_records = self.test_client.read('companies', self.expected_page_limits().get(stream), since=self.my_start_date) + companies_records = self.test_client.read('companies', since=self.my_start_date) company_ids = [company['companyId'] for company in companies_records] self.test_client.create(stream, company_ids) else: - self.test_client.create(stream ) + self.test_client.create(stream) def expected_streams(self): """