public Milestone create(Milestone milestone) throws ClassNotFoundException, SQLException {
    String id = UUID.randomUUID().toString();
    milestone.setId(id);

    Class.forName("org.h2.Driver");
    Connection connection = DriverManager.getConnection("jdbc:h2:~/dao_db", "sa", "");
    PreparedStatement prepareStatement = connection.prepareStatement("INSERT INTO MILESTONE VALUES(?, ?,?)");

    prepareStatement.setString(1, id);
    prepareStatement.setString(2, milestone.getName());
    prepareStatement.setString(3, milestone.getDescription());
    prepareStatement.executeUpdate();

    connection.close();
    return milestone;}

public Milestone update(Milestone milestone) throws ClassNotFoundException, SQLException{

    Class.forName("org.h2.Driver");
    Connection connection = DriverManager.getConnection("jdbc:h2:~/dao_db", "sa", "");
    PreparedStatement prepareStatement = connection.prepareStatement("UPDATE MILESTONE SET NAME=?, DESCRIPTION = ? WHERE ID=?");

    prepareStatement.setString(1, milestone.getName());
    prepareStatement.setString(2, milestone.getDescription());
    prepareStatement.setString(3, milestone.getId());
    prepareStatement.executeUpdate();

    connection.close();
    return milestone;};

import static org.junit.Assert.*;
    import java.sql.SQLException;
    import org.junit.Test;
    import model.Milestone;

    public class MilestoneDAOImplTest {

@Test
public void createTest() throws ClassNotFoundException, SQLException {
    MilestoneDAOImpl ms = new MilestoneDAOImpl();
    Milestone milestone = new Milestone("test","test");
    assertNotNull("milestone created", ms.create(milestone));   }

@Test
public void updateTest() throws ClassNotFoundException, SQLException {
    MilestoneDAOImpl ms = new MilestoneDAOImpl();
    Milestone milestone = new Milestone("test","test");
    assertNotNull("milestone updated", ms.update(milestone));   }

assertNotNull("milestone created", ms.create(milestone));