[airflow] branch master updated: Add conn_type to Fix failing Livy Tests (#9258)

[ka...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

kaxilnaik pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/airflow.git


The following commit(s) were added to refs/heads/master by this push:
     new e2a9097  Add conn_type to Fix failing Livy Tests (#9258)
e2a9097 is described below

commit e2a909787c044c6960b58d5f864b9ad8b4fc5885
Author: Kaxil Naik <ka...@gmail.com>
AuthorDate: Fri Jun 12 22:42:04 2020 +0100

    Add conn_type to Fix failing Livy Tests (#9258)
---
 tests/providers/apache/livy/hooks/test_livy.py | 18 ++++++++++--------
 1 file changed, 10 insertions(+), 8 deletions(-)

diff --git a/tests/providers/apache/livy/hooks/test_livy.py b/tests/providers/apache/livy/hooks/test_livy.py
index bd2fbed..65e1d65 100644
--- a/tests/providers/apache/livy/hooks/test_livy.py
+++ b/tests/providers/apache/livy/hooks/test_livy.py
@@ -34,14 +34,16 @@ SAMPLE_GET_RESPONSE = {'id': BATCH_ID, 'state': BatchState.SUCCESS.value}
 class TestLivyHook(unittest.TestCase):
     @classmethod
     def setUpClass(cls):
-        db.merge_conn(Connection(conn_id='livy_default', host='host', schema='http', port='8998'))
-        db.merge_conn(Connection(conn_id='default_port', host='http://host'))
-        db.merge_conn(Connection(conn_id='default_protocol', host='host'))
-        db.merge_conn(Connection(conn_id='port_set', host='host', port=1234))
-        db.merge_conn(Connection(conn_id='schema_set', host='host', schema='zzz'))
-        db.merge_conn(Connection(conn_id='dont_override_schema', host='http://host', schema='zzz'))
-        db.merge_conn(Connection(conn_id='missing_host', port=1234))
-        db.merge_conn(Connection(conn_id='invalid_uri', uri='http://invalid_uri:4321'))
+        db.merge_conn(
+            Connection(conn_id='livy_default', conn_type='http', host='host', schema='http', port='8998'))
+        db.merge_conn(Connection(conn_id='default_port', conn_type='http', host='http://host'))
+        db.merge_conn(Connection(conn_id='default_protocol', conn_type='http', host='host'))
+        db.merge_conn(Connection(conn_id='port_set', host='host', conn_type='http', port=1234))
+        db.merge_conn(Connection(conn_id='schema_set', host='host', conn_type='http', schema='zzz'))
+        db.merge_conn(
+            Connection(conn_id='dont_override_schema', conn_type='http', host='http://host', schema='zzz'))
+        db.merge_conn(Connection(conn_id='missing_host', conn_type='http', port=1234))
+        db.merge_conn(Connection(conn_id='invalid_uri', conn_type='http', uri='http://invalid_uri:4321'))
 
     def test_build_get_hook(self):