我在我的系统中使用Google Cloud 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 来模拟对象。
希望可以有人帮帮我。
对于第一个示例的单元测试,您可以使用 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
and HttpServletResponse
然后使用这些模拟对象调用 servlet。
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)