@Test
public void testUpdateTreatment() {

    client
        .createExperiment("exp")
        .apply();
    client.addTreatment("exp", "control", "the control");
    client
        .updateTreatment("exp", "control")
        .setDescription("new description")
        .apply();
    final TreatmentDto treatment = client.getTreatment("exp", "control");
    assertEquals("control", treatment.getName());
    assertEquals("new description", treatment.getDescription());
}