package and tests

posthog/temporal/tests/test_external_data_job.py

@@ -8,6 +8,7 @@
 
 from posthog.warehouse.data_load.pipeline import StripeJobInputs, SourceColumnType, SourceSchema
 from posthog.warehouse.data_load.service import ExternalDataJobInputs
+from posthog.warehouse.data_load.stripe import ENDPOINTS
 
 from posthog.temporal.data_imports.external_data_job import (
     create_external_data_job_model,
@@ -119,7 +120,7 @@ async def test_run_stripe_job(activity_environment, team, **kwargs):
                 dataset_name=new_source.draft_folder_path,
             )
         )
-        mock_run_stripe.assert_called_once_with(stripe_secret_key="test-key")
+        mock_run_stripe.assert_called_once_with(stripe_secret_key="test-key", endpoints=ENDPOINTS)
         assert len(schemas) == 1
 
 
@@ -159,7 +160,7 @@ async def test_is_schema_valid_activity(activity_environment, team, **kwargs):
             ),
         )
 
-        assert mock_get_columns.call_count == 1
+        assert mock_get_columns.call_count == 5
 
 
 @pytest.mark.django_db(transaction=True)
@@ -240,10 +241,10 @@ async def test_create_schema_activity(activity_environment, team, **kwargs):
             ),
         )
 
-        assert mock_get_columns.call_count == 4
+        assert mock_get_columns.call_count == 5
         all_tables = DataWarehouseTable.objects.all()
         table_length = await sync_to_async(len)(all_tables)
-        assert table_length == 4
+        assert table_length == 5
 
         # Should still have one after
         await activity_environment.run(
@@ -268,7 +269,7 @@ async def test_create_schema_activity(activity_environment, team, **kwargs):
         all_tables = DataWarehouseTable.objects.all()
         table_length = await sync_to_async(len)(all_tables)
 
-        assert table_length == 4
+        assert table_length == 5
 
 
 @pytest.mark.django_db(transaction=True)

requirements.in

@@ -35,7 +35,7 @@ django-revproxy==0.12.0
 djangorestframework==3.14.0
 djangorestframework-csv==2.1.1
 djangorestframework-dataclasses==1.2.0
-django-fernet-encrypted-fields==0.1.2
+django-fernet-encrypted-fields==0.1.3
 dlt==0.3.24
 dnspython==2.2.1
 drf-exceptions-hog==0.4.0

requirements.txt

@@ -169,7 +169,7 @@ django-deprecate-fields==0.1.1
     # via -r requirements.in
 django-extensions==3.1.2
     # via -r requirements.in
-django-fernet-encrypted-fields==0.1.2
+django-fernet-encrypted-fields==0.1.3
     # via -r requirements.in
 django-filter==2.4.0
     # via -r requirements.in
@@ -243,7 +243,6 @@ giturlparse==0.12.0
     # via dlt
 google-api-core[grpc]==2.11.1
     # via
-    #   google-api-core
     #   google-cloud-bigquery
     #   google-cloud-core
 google-auth==2.22.0