Apache Storm Junit测试用例螺栓
问题描述:
我写过一个螺栓和一个拓扑结构。现在我需要做一个Junit测试用例。我的实际输入是一个Json,并且在我的帮助下,我编写了代码将其存储在MySQL中。现在我需要测试我在方法中编写的逻辑。那么如何在测试用例中为元组创建一个对象,或者是否有其他方法可以做到这一点。我附上了我的方法的代码。Apache Storm Junit测试用例螺栓
public void execute(Tuple input, BasicOutputCollector collector) {
int size = input.getFields().size();
Fields fields = input.getFields();
if (size == 1) {
logger.info("PostbackToRDSBolt execute method starts");
try {
eventJson = (JSONObject) JSONSerializer.toJSON((String) input.getValueByField(fields.get(0)));
address = (String) input.getValueByField(fields.get(0));
Id = (String) eventJson.get("Id");
click = (String) eventJson.get("click");
time = (String) eventJson.get("time");
uuid = UUID.randomUUID().toString();
type = "post";
//Class.forName("com.mysql.jdbc.Driver");
con = DriverManager.getConnection(
dBUrl, username,
password);
stmt = con.createStatement();
stmt.execute("INSERT INTO " + tableName
+ "(id,data,type,txnid,groupname,time)values('" + uuid
+ "','" + address + "'," + "'" + type + "','"
+ Id + "','" + click + "','" + time
+ "')");
logger.info("inserted successfully in mysql");
} catch (SQLException e) {
e.printStackTrace();
} catch (Exception e1) {
logger.info("" + e1);
e1.printStackTrace();
}
}
}
}
答
如果你想要做的JUnit测试,一种方法是使用的Mockito库:
例如:
@Test
public void testExecute() {
Tuple t = mock(Tuple.class);
when(t.getFields()).thenReturn(new Fields("myAttribute"));
when(t.getValueByField("myAttribute").thenReturn("my json string to be tested");
MyBolt b = new MyBolt();
b.execute(t, mock(BasicOutputCollector.class));
// put your assertions here
}
对于这个工作,你应该在更改代码一点点。从
int size = input.getFields().size();
Fields fields = input.getFields();
到
Fields fields = input.getFields();
int size = fields.size();
否则,input.getFields()
被调用两次,但该测试仅用于该方法中指定一个返回值,并且因此,第二呼叫将返回null
。 (作为替代,还可以模拟到getFields()
两个电话。)
如果你不想嘲笑一切,你也可以实例化对象TupleImpl
(Tuple t = new TupleImpl(...)
),并在您的测试传递给。
最佳实践是将您的逻辑封装到服务类中以简化单元测试,而不是直接测试您的螺栓。然后,您可以使用LocalCluster实例创建集成测试。 – fhussonnois