def test_create_dashboard(self): # Setup Expected Response name = "name3373707" display_name = "displayName1615086568" etag = "etag3123477" expected_response = { "name": name, "display_name": display_name, "etag": etag } expected_response = dashboard_pb2.Dashboard(**expected_response) # Mock the API response channel = ChannelStub(responses=[expected_response]) patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = v1.DashboardsServiceClient() # Setup Request parent = "parent-995424086" dashboard = {} response = client.create_dashboard(parent, dashboard) assert expected_response == response assert len(channel.requests) == 1 expected_request = dashboards_service_pb2.CreateDashboardRequest( parent=parent, dashboard=dashboard) actual_request = channel.requests[0][1] assert expected_request == actual_request
def test_get_dashboard(self): # Setup Expected Response name_2 = "name2-1052831874" display_name = "displayName1615086568" etag = "etag3123477" expected_response = { "name": name_2, "display_name": display_name, "etag": etag } expected_response = dashboard_pb2.Dashboard(**expected_response) # Mock the API response channel = ChannelStub(responses=[expected_response]) patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = v1.DashboardsServiceClient() # Setup Request name = "name3373707" response = client.get_dashboard(name) assert expected_response == response assert len(channel.requests) == 1 expected_request = dashboards_service_pb2.GetDashboardRequest( name=name) actual_request = channel.requests[0][1] assert expected_request == actual_request
def test_list_dashboards(self): # Setup Expected Response next_page_token = "" dashboards_element = {} dashboards = [dashboards_element] expected_response = { "next_page_token": next_page_token, "dashboards": dashboards, } expected_response = dashboards_service_pb2.ListDashboardsResponse( **expected_response) # Mock the API response channel = ChannelStub(responses=[expected_response]) patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = v1.DashboardsServiceClient() # Setup Request parent = "parent-995424086" paged_list_response = client.list_dashboards(parent) resources = list(paged_list_response) assert len(resources) == 1 assert expected_response.dashboards[0] == resources[0] assert len(channel.requests) == 1 expected_request = dashboards_service_pb2.ListDashboardsRequest( parent=parent) actual_request = channel.requests[0][1] assert expected_request == actual_request
def checkForDashboard(self, dashboard_display_name): client = v1.DashboardsServiceClient() dashboards = client.list_dashboards(project_name) for dashboard in dashboards: if dashboard.display_name == dashboard_display_name: return True return False
def testUserExpDashboard(self): """ Test that the User Experience Dashboard gets created. """ client = v1.DashboardsServiceClient() dashboards = client.list_dashboards(project_name) found_userexp_dashboard = False for dashboard in dashboards: if dashboard.display_name == 'User Experience Dashboard': found_userexp_dashboard = True self.assertTrue(found_userexp_dashboard)
def cleanupDashboards(project_name): """ Deletes all dashboards. """ client = v1.DashboardsServiceClient() dashboards = True while dashboards: dashboards = list(client.list_dashboards(project_name)) for dashboard in dashboards: try: client.delete_dashboard(dashboard.name) except: print('Could not delete dashboard: ' + dashboard.name)
def test_update_dashboard_exception(self): # Mock the API response channel = ChannelStub(responses=[CustomException()]) patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = v1.DashboardsServiceClient() # Setup request dashboard = {} with pytest.raises(CustomException): client.update_dashboard(dashboard)
def test_list_dashboards_exception(self): channel = ChannelStub(responses=[CustomException()]) patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = v1.DashboardsServiceClient() # Setup request parent = "parent-995424086" paged_list_response = client.list_dashboards(parent) with pytest.raises(CustomException): list(paged_list_response)
def test_delete_dashboard(self): channel = ChannelStub() patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = v1.DashboardsServiceClient() # Setup Request name = "name3373707" client.delete_dashboard(name) assert len(channel.requests) == 1 expected_request = dashboards_service_pb2.DeleteDashboardRequest( name=name) actual_request = channel.requests[0][1] assert expected_request == actual_request