Fix the tests

dbt-labs · Jun 1, 2020 · 7a64565 · 7a64565
1 parent 836b247
commit 7a64565
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/test/unit/test_adapter.py b/test/unit/test_adapter.py
@@ -35,7 +35,7 @@ def _get_target_http(self, project):
                     'host': 'myorg.sparkhost.com',
                     'port': 443,
                     'token': 'abc123',
-                    'organization': '0123456789',
+                    'organization': 123456789,
                     'cluster': '01234-23423-coffeetime',
                 }
             },
@@ -65,7 +65,7 @@ def hive_http_connect(thrift_transport):
             self.assertEqual(thrift_transport.scheme, 'https')
             self.assertEqual(thrift_transport.port, 443)
             self.assertEqual(thrift_transport.host, 'myorg.sparkhost.com')
-            self.assertEqual(thrift_transport.path, '/sql/protocolv1/o/0123456789/01234-23423-coffeetime')
+            self.assertEqual(thrift_transport.path, '/sql/protocolv1/o/123456789/01234-23423-coffeetime')
 
         # with mock.patch.object(hive, 'connect', new=hive_http_connect):
         with mock.patch('dbt.adapters.spark.connections.hive.connect', new=hive_http_connect):
@@ -259,7 +259,7 @@ def test_profile_with_database(self):
                     'host': 'myorg.sparkhost.com',
                     'port': 443,
                     'token': 'abc123',
-                    'organization': '0123456789',
+                    'organization': 123456789,
                     'cluster': '01234-23423-coffeetime',
                 }
             },