import org.junit.*;
import static org.junit.Assert.*;
import java.util.Arrays;
import org.junit.Rule;

public class ClientTest {

  @Rule
  public DatabaseRule database = new DatabaseRule();

  @Test
  public void all_emptyAtFirst() {
    assertEquals(Client.all().size(), 0);
  }

  @Test
  public void equals_returnsTrueIfNamesAretheSame() {
    Client firstClient = new Client("Gavrilo Princip", 1);
    Client secondClient = new Client("Gavrilo Princip", 1);
    assertTrue(firstClient.equals(secondClient));
  }

  @Test
  public void save_savesIntoDatabase_true() {
    Client myClient = new Client("Gavrilo Princip", 1);
    myClient.save();
    assertTrue(Client.all().get(0).equals(myClient));
  }

  @Test
  public void find_findClientInDatabase_true() {
    Client myClient = new Client("Gavrilo Princip", 1);
    myClient.save();
    Client savedClient = Client.find(myClient.getId());
    assertTrue(myClient.equals(savedClient));
  }

  @Test
  public void update_updatesClientNameInDatabase_newName() {
    Client myClient = new Client("Gavrilo Princip", 1);
    myClient.save();
    myClient.update("The City of Sarajevo", 1);
    assertEquals(myClient.getName(), "The City of Sarajevo", "The City of Sarajevo");
  }

  @Test
  public void delete_deletesClientEntry_true() {
    Client myClient = new Client("Gavrilo Princip", 1);
    myClient.save();
    myClient.delete();
    assertEquals(Client.all().size(), 0);
  }

}

@@ -1431,10 +1431,364 @@
 );%0A  %7D%0A%0A
+  @Test%0A  public void save_savesStylistIdIntoDB_true() %7B%0A    Stylist myStylist = new Stylist(%22Franz Ferdinand%22);%0A    myStylist.save();%0A    Client myClient = new Client(%22Gavrilo Princip%22, myStylist.getId());%0A    myClient.save();%0A    Client savedClient = Client.find(myClient.getId());%0A    assertEquals(savedClient.getStylistId(), myStylist.getId());%0A  %7D%0A%0A
 %7D%0A

Add test to ensure Stylist number is saved properly in Client database entry.