def create(cls):
"""Creates an instance of FakeEntryPoint.
Returns:
instance of FakeEntryPoint
"""
return cls('foo', 'bar')
def load(self):
"""Returns FakePlugin instead of resolving module.
Returns:
FakePlugin
"""
return FakePlugin
class DefaultTest(test.TestCase):
@mock.patch.object(pkg_resources, 'iter_entry_points')
def test_get_dynamic_plugin(self, mock_iter_entry_points):
mock_iter_entry_points.return_value = [FakeEntryPoint.create()]
actual_plugins = default.get_dynamic_plugins()
mock_iter_entry_points.assert_called_with('tensorboard_plugins')
self.assertEqual(actual_plugins, [FakePlugin])
if __name__ == "__main__":
test.main()
del request # unused
response = export_service_pb2.StreamExperimentsResponse()
response.experiments.add(experiment_id="789", name="one")
response.experiments.add(experiment_id="012", description="two")
yield response
mock_api_client.StreamExperiments = mock.Mock(wraps=stream_experiments)
gen = exporter_lib.list_experiments(mock_api_client)
mock_api_client.StreamExperiments.assert_not_called()
expected = [
experiment_pb2.Experiment(experiment_id="789", name="one"),
experiment_pb2.Experiment(experiment_id="012", description="two"),
]
self.assertEqual(list(gen), expected)
def _create_mock_api_client():
# Create a stub instance (using a test channel) in order to derive a mock
# from it with autospec enabled. Mocking TensorBoardExporterServiceStub
# itself doesn't work with autospec because grpc constructs stubs via
# metaclassing.
test_channel = grpc_testing.channel(service_descriptors=[],
time=grpc_testing.strict_real_time())
stub = export_service_pb2_grpc.TensorBoardExporterServiceStub(test_channel)
mock_api_client = mock.create_autospec(stub)
return mock_api_client
if __name__ == "__main__":
tb_test.main()
