def test_create_publisher_topic_not_found(mock_publisher):
client = mock_publisher.return_value
client.get_topic.side_effect = gapi_exceptions.NotFound("foo")
with pytest.raises(SystemExit):
publish._create_publisher("a-topic")
mock_publisher.assert_called_once_with()
client.get_topic.assert_called_once_with("a-topic")
def test_create_publisher_raises(mock_publisher):
client = mock_publisher.return_value
client.get_topic.side_effect = Exception("foo")
with pytest.raises(Exception, match="foo"):
publish._create_publisher("a-topic")
mock_publisher.assert_called_once_with()
client.get_topic.assert_called_once_with("a-topic")
def test_create_publisher(mock_publisher):
client = mock_publisher.return_value
ret_publisher = publish._create_publisher("a-topic")
mock_publisher.assert_called_once_with()
client.get_topic.assert_called_once_with("a-topic")
expected = functools.partial(client.publish, topic="a-topic")
assert expected.func == ret_publisher.func