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