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

来源

2015-05-14 sbgowtham

最佳实践是将您的逻辑封装到服务类中以简化单元测试，而不是直接测试您的螺栓。然后，您可以使用LocalCluster实例创建集成测试。 – fhussonnois

如果你想要做的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(...)），并在您的测试传递给。

来源

2015-09-30 14:59:26

Apache Storm Junit测试用例螺栓

回答

相关问题