class GKEClusterHookGetTest(unittest.TestCase):
def setUp(self):
self.gke_hook = GKEClusterHook(location=ZONE)
self.gke_hook._client = mock.Mock()
def test_get_cluster(self):
retry_mock, timeout_mock = mock.Mock(), mock.Mock()
client_get = self.gke_hook._client.get_cluster = mock.Mock()
self.gke_hook.get_cluster(name=CLUSTER_NAME, project_id=TEST_PROJECT_ID, retry=retry_mock,
timeout=timeout_mock)
client_get.assert_called_with(project_id=TEST_PROJECT_ID, zone=ZONE,
cluster_id=CLUSTER_NAME,
retry=retry_mock, timeout=timeout_mock)
class GKEClusterHookGetTest(unittest.TestCase):
def setUp(self):
self.gke_hook = GKEClusterHook(location=GKE_ZONE)
self.gke_hook._client = mock.Mock()
def test_get_cluster(self):
retry_mock, timeout_mock = mock.Mock(), mock.Mock()
client_get = self.gke_hook._client.get_cluster = mock.Mock()
self.gke_hook.get_cluster(name=CLUSTER_NAME,
project_id=TEST_GCP_PROJECT_ID,
retry=retry_mock,
timeout=timeout_mock)
client_get.assert_called_with(project_id=TEST_GCP_PROJECT_ID,
zone=GKE_ZONE,
cluster_id=CLUSTER_NAME,
retry=retry_mock, timeout=timeout_mock)
class GKEClusterHookGetTest(unittest.TestCase):
def setUp(self):
with mock.patch.object(GKEClusterHook, "__init__", return_value=None):
self.gke_hook = GKEClusterHook(None, None, None)
self.gke_hook.project_id = TEST_PROJECT_ID
self.gke_hook.location = ZONE
self.gke_hook.client = mock.Mock()
def test_get_cluster(self):
retry_mock, timeout_mock = mock.Mock(), mock.Mock()
client_get = self.gke_hook.client.get_cluster = mock.Mock()
self.gke_hook.get_cluster(name=CLUSTER_NAME, retry=retry_mock,
timeout=timeout_mock)
client_get.assert_called_with(project_id=TEST_PROJECT_ID, zone=ZONE,
cluster_id=CLUSTER_NAME,
retry=retry_mock, timeout=timeout_mock)
class GKEClusterHookGetTest(unittest.TestCase):
def setUp(self):
with mock.patch.object(GKEClusterHook, "__init__", return_value=None):
self.gke_hook = GKEClusterHook(None, None, None)
self.gke_hook.project_id = TEST_PROJECT_ID
self.gke_hook.location = ZONE
self.gke_hook.client = mock.Mock()
def test_get_cluster(self):
retry_mock, timeout_mock = mock.Mock(), mock.Mock()
client_get = self.gke_hook.client.get_cluster = mock.Mock()
self.gke_hook.get_cluster(name=CLUSTER_NAME,
retry=retry_mock,
timeout=timeout_mock)
client_get.assert_called_with(project_id=TEST_PROJECT_ID,
zone=ZONE,
cluster_id=CLUSTER_NAME,
retry=retry_mock,
timeout=timeout_mock)
