fixed static styling and connection

bdsoha · bdsoha · commit 48fbd896c3ba · 2022-08-07T14:37:25.000+03:00
diff --git a/airflow/providers/apache/livy/hooks/livy.py b/airflow/providers/apache/livy/hooks/livy.py
@@ -76,7 +76,7 @@ def __init__(
         livy_conn_id: str = default_conn_name,
         extra_options: Optional[Dict[str, Any]] = None,
         extra_headers: Optional[Dict[str, Any]] = None,
-        auth_type: Optional[Any] = None
+        auth_type: Optional[Any] = None,
     ) -> None:
         super().__init__(http_conn_id=livy_conn_id)
         self.extra_headers = extra_headers or {}
diff --git a/airflow/providers/apache/livy/operators/livy.py b/airflow/providers/apache/livy/operators/livy.py
@@ -131,7 +131,7 @@ def get_hook(self) -> LivyHook:
                 livy_conn_id=self._livy_conn_id,
                 extra_headers=self._extra_headers,
                 extra_options=self._extra_options,
-                auth_type=self._livy_conn_auth_type
+                auth_type=self._livy_conn_auth_type,
             )
         return self._livy_hook
 
diff --git a/airflow/providers/apache/livy/sensors/livy.py b/airflow/providers/apache/livy/sensors/livy.py
@@ -63,7 +63,7 @@ def get_hook(self) -> LivyHook:
             self._livy_hook = LivyHook(
                 livy_conn_id=self._livy_conn_id,
                 extra_options=self._extra_options,
-                auth_type=self._livy_conn_auth_type
+                auth_type=self._livy_conn_auth_type,
             )
         return self._livy_hook
 
diff --git a/tests/providers/apache/livy/hooks/test_livy.py b/tests/providers/apache/livy/hooks/test_livy.py
@@ -17,7 +17,7 @@
 
 import json
 import unittest
-from unittest.mock import patch, MagicMock
+from unittest.mock import MagicMock, patch
 
 import pytest
 import requests_mock
@@ -47,7 +47,7 @@ def setUpClass(cls):
         )
         db.merge_conn(Connection(conn_id='missing_host', conn_type='http', port=1234))
         db.merge_conn(Connection(conn_id='invalid_uri', uri='http://invalid_uri:4321'))
-        db.merge_conn(Connection(conn_id='with_credentials', login='login', password='secret', host='host'))
+        db.merge_conn(Connection(conn_id='with_credentials', login='login', password='secret', conn_type='http', host='host'))
 
     def test_build_get_hook(self):
 

Original file line number	Diff line number	Diff line change
`@@ -131,7 +131,7 @@ def get_hook(self) -> LivyHook:`
`131`	`131`	`livy_conn_id=self._livy_conn_id,`
`132`	`132`	`extra_headers=self._extra_headers,`
`133`	`133`	`extra_options=self._extra_options,`
`134`		`- auth_type=self._livy_conn_auth_type`
	`134`	`+ auth_type=self._livy_conn_auth_type,`
`135`	`135`	`)`
`136`	`136`	`return self._livy_hook`
`137`	`137`
Original file line number	Diff line number	Diff line change
`@@ -63,7 +63,7 @@ def get_hook(self) -> LivyHook:`
`63`	`63`	`self._livy_hook = LivyHook(`
`64`	`64`	`livy_conn_id=self._livy_conn_id,`
`65`	`65`	`extra_options=self._extra_options,`
`66`		`- auth_type=self._livy_conn_auth_type`
	`66`	`+ auth_type=self._livy_conn_auth_type,`
`67`	`67`	`)`
`68`	`68`	`return self._livy_hook`
`69`	`69`