Question

我写了一个螺栓和一个拓扑。现在我需要做一个Junit测试用例。我的实际输入是一个Json，在我的螺栓中，我编写了代码来存储它在MySQL中。现在我需要测试我用execute()方法编写的逻辑。那么如何在测试用例中为元组创建一个对象，或者还有其他方法可以做到这一点。我附上了execute()方法的代码。

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();
            }
        }
    }
}

Answer 1

如果你想进行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
}

要使其正常工作，您应该稍微更改execute()中的代码。从

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(...)），并在测试中将它们传递给execute()。

用于螺栓的Apache Storm Junit测试用例

1 个答案: