TestBLOBHandler.java 文件源码

java
阅读 33 收藏 0 点赞 0 评论 0

项目:flume-release-1.7.0 作者:
@SuppressWarnings({ "unchecked", "rawtypes" })
@Test
public void testCSVData() throws Exception {
  Map requestParameterMap = new HashMap();
  requestParameterMap.put("param1", new String[] { "value1" });
  requestParameterMap.put("param2", new String[] { "value2" });

  HttpServletRequest req = mock(HttpServletRequest.class);
  final String csvData = "a,b,c";

  ServletInputStream servletInputStream = new DelegatingServletInputStream(
      new ByteArrayInputStream(csvData.getBytes()));

  when(req.getInputStream()).thenReturn(servletInputStream);
  when(req.getParameterMap()).thenReturn(requestParameterMap);

  Context context = mock(Context.class);
  when(
      context.getString(BLOBHandler.MANDATORY_PARAMETERS,
          BLOBHandler.DEFAULT_MANDATORY_PARAMETERS)).thenReturn(
      "param1,param2");

  handler.configure(context);
  List<Event> deserialized = handler.getEvents(req);
  assertEquals(1, deserialized.size());
  Event e = deserialized.get(0);

  assertEquals(new String(e.getBody()), csvData);
  assertEquals(e.getHeaders().get("param1"), "value1");
  assertEquals(e.getHeaders().get("param2"), "value2");
}
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号