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
