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()