Skip to content

Commit

Permalink
test(providers/google): add test cases to GKEPodAsyncHook service_fil…
Browse files Browse the repository at this point in the history
…e support
  • Loading branch information
Lee-W committed Feb 4, 2024
1 parent cdea6ca commit 16a0be4
Showing 1 changed file with 33 additions and 7 deletions.
40 changes: 33 additions & 7 deletions tests/providers/google/cloud/hooks/test_kubernetes_engine.py
Expand Up @@ -317,29 +317,46 @@ def async_hook(self):
)

@pytest.mark.asyncio
@mock.patch(GKE_STRING.format("Token"), mock.MagicMock())
@pytest.mark.parametrize("mock_service_file", ("/tmp/service_file.json", None))
@mock.patch(GKE_STRING.format("Token"))
@mock.patch(GKE_STRING.format("GKEPodAsyncHook.get_conn"))
@mock.patch(GKE_STRING.format("async_client.CoreV1Api.read_namespaced_pod"))
async def test_get_pod(self, read_namespace_pod_mock, get_conn_mock, async_hook):
async def test_get_pod(
self, read_namespace_pod_mock, get_conn_mock, mock_token, async_hook, mock_service_file
):
async_hook.service_file_as_context = mock.MagicMock()
async_hook.service_file_as_context.return_value.__aenter__.return_value = mock_service_file

self.make_mock_awaitable(read_namespace_pod_mock)

await async_hook.get_pod(name=POD_NAME, namespace=POD_NAMESPACE)

mock_token.assert_called_with(
scopes=["https://www.googleapis.com/auth/cloud-platform"], service_file=mock_service_file
)
get_conn_mock.assert_called_once()
read_namespace_pod_mock.assert_called_with(
name=POD_NAME,
namespace=POD_NAMESPACE,
)

@pytest.mark.asyncio
@mock.patch(GKE_STRING.format("Token"), mock.MagicMock())
@pytest.mark.parametrize("mock_service_file", ("/tmp/service_file.json", None))
@mock.patch(GKE_STRING.format("Token"))
@mock.patch(GKE_STRING.format("GKEPodAsyncHook.get_conn"))
@mock.patch(GKE_STRING.format("async_client.CoreV1Api.delete_namespaced_pod"))
async def test_delete_pod(self, delete_namespaced_pod, get_conn_mock, async_hook):
async def test_delete_pod(
self, delete_namespaced_pod, get_conn_mock, mock_token, async_hook, mock_service_file
):
async_hook.service_file_as_context = mock.MagicMock()
async_hook.service_file_as_context.return_value.__aenter__.return_value = mock_service_file

self.make_mock_awaitable(delete_namespaced_pod)

await async_hook.delete_pod(name=POD_NAME, namespace=POD_NAMESPACE)

mock_token.assert_called_with(
scopes=["https://www.googleapis.com/auth/cloud-platform"], service_file=mock_service_file
)
get_conn_mock.assert_called_once()
delete_namespaced_pod.assert_called_with(
name=POD_NAME,
Expand All @@ -348,14 +365,23 @@ async def test_delete_pod(self, delete_namespaced_pod, get_conn_mock, async_hook
)

@pytest.mark.asyncio
@mock.patch(GKE_STRING.format("Token"), mock.MagicMock())
@pytest.mark.parametrize("mock_service_file", ("/tmp/service_file.json", None))
@mock.patch(GKE_STRING.format("Token"))
@mock.patch(GKE_STRING.format("GKEPodAsyncHook.get_conn"))
@mock.patch(GKE_STRING.format("async_client.CoreV1Api.read_namespaced_pod_log"))
async def test_read_logs(self, read_namespaced_pod_log, get_conn_mock, async_hook, caplog):
async def test_read_logs(
self, read_namespaced_pod_log, get_conn_mock, mock_token, async_hook, mock_service_file, caplog
):
async_hook.service_file_as_context = mock.MagicMock()
async_hook.service_file_as_context.return_value.__aenter__.return_value = mock_service_file

self.make_mock_awaitable(read_namespaced_pod_log, result="Test string #1\nTest string #2\n")

await async_hook.read_logs(name=POD_NAME, namespace=POD_NAMESPACE)

mock_token.assert_called_with(
scopes=["https://www.googleapis.com/auth/cloud-platform"], service_file=mock_service_file
)
get_conn_mock.assert_called_once()
read_namespaced_pod_log.assert_called_with(
name=POD_NAME,
Expand Down

0 comments on commit 16a0be4

Please sign in to comment.