Question

我在我的系统中使用Google Pub Pub / Sub的push pub / sub，我想构建我的CI测试代码，但我不知道该怎么做。例如，有些代码是这样的：

final Pubsub pubsub = PubsubUtils.getClient();
final PubsubMessage pubsubMessage = new PubsubMessage();

pubsubMessage.encodeData(message.getBytes(StandardCharsets.UTF_8));
Map<String, String> attrs = new HashMap<String, String>();
attrs.put("key", "value");
pubsubMessage.setAttributes(attrs);

final List<PubsubMessage> messages = ImmutableList.of(pubsubMessage);
final PublishRequest publishRequest = new PublishRequest().setMessages(messages);
        final PublishResponse publishResponse = pubsub.projects().topics().publish(topic, publishRequest).execute();
        final List<String> messageIds = publishResponse.getMessageIds();

和此：

final ServletInputStream reader = request.getInputStream();
        try {
            // Parse the JSON message to the POJO model class.
            final JsonParser parser = JacksonFactory.getDefaultInstance().createJsonParser(reader);

            parser.skipToKey("message");

            final PubsubMessage message = parser.parseAndClose(PubsubMessage.class);

            Map<String, String> attrs = message.getAttributes();
            String value = attrs.get("key");

            // Base64-decode the data and work with it.
            final String data = new String(message.decodeData(), StandardCharsets.UTF_8);
            if (data != null || StringUtils.isNotEmpty(data)) {
                Logger.getLogger("logger").info(data);
            }
            // Work with your message
            // Respond with a 20X to acknowledge receipt of the message.
            response.setStatus(HttpServletResponse.SC_OK);
        } finally {
            reader.close();
        }

如何为代码的两个部分编写一个正常的Junit测试用例？而且，我正在使用PowerMockito来模拟对象。

希望有人可以帮助我。

Answer 1

对于第一个示例的单元测试，您可以使用API调用的所需返回值来模拟Pub / Sub客户端对象，如：

import com.google.api.services.pubsub.Pubsub;
import com.google.api.services.pubsub.Pubsub.Projects;
import com.google.api.services.pubsub.Pubsub.Projects.Topics;
import com.google.api.services.pubsub.Pubsub.Projects.Topics.Create;
import com.google.api.services.pubsub.Pubsub.Projects.Topics.Publish;
// ...
@Mock private Pubsub mockPubsub;
@Mock private Projects mockProjects;
@Mock private Topics mockTopics;
@Mock private Create mockCreate;
@Mock private Publish mockPublish;
// ...
String topicName = "projects/myproject/topics/mytopic";
String messageId = "messageId";
List<String> messageIds = ImmutableList.of(messageId);
PublishResponse publishResponse = new PublishResponse()
        .setMessageIds(messageIds);

when(mockPubsub.projects()).thenReturn(mockProjects);
when(mockProjects.topics()).thenReturn(mockTopics);
when(mockTopics.publish(eq(topicName), isA(PublishRequest.class)))
        .thenReturn(mockPublish);
when(mockPublish.execute()).thenReturn(publishResponse);

对于第二个示例，您可以模拟HttpServletRequest和HttpServletResponse，然后使用这些模拟对象调用servlet。

如何使用Google Cloud Pub / Sub进行Junit测试

1 个答案: