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 <[email protected]>
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):
