def test_listAllStatefulSetsWithLabels_custom(self, client_mock): service = KubernetesService() client_mock.reset_mock() labels = {"operated-by": "me", "heritage": "mongo", "name": "name"} result = service.listAllStatefulSetsWithLabels(labels) expected_calls = [ call.AppsV1beta1Api().list_stateful_set_for_all_namespaces( label_selector="operated-by=me,heritage=mongo,name=name") ] self.assertEqual(expected_calls, client_mock.mock_calls) self.assertEqual( client_mock.AppsV1beta1Api().list_stateful_set_for_all_namespaces. return_value, result)
def test_listAllStatefulSetsWithLabels_default(self, client_mock): service = KubernetesService() client_mock.reset_mock() result = service.listAllStatefulSetsWithLabels() expected_calls = [ call.AppsV1beta1Api().list_stateful_set_for_all_namespaces( label_selector= "operated-by=operators.ultimaker.com,heritage=mongos") ] self.assertEqual(expected_calls, client_mock.mock_calls) self.assertEqual( client_mock.AppsV1beta1Api().list_stateful_set_for_all_namespaces. return_value, result)