diff --git a/src/test/java/org/springframework/samples/petclinic/vet/VetTests.java b/src/test/java/org/springframework/samples/petclinic/vet/VetTests.java index de3a7b9bb..e3a1f6d59 100644 --- a/src/test/java/org/springframework/samples/petclinic/vet/VetTests.java +++ b/src/test/java/org/springframework/samples/petclinic/vet/VetTests.java @@ -17,9 +17,14 @@ package org.springframework.samples.petclinic.vet; import org.junit.Test; +import org.springframework.beans.support.MutableSortDefinition; +import org.springframework.beans.support.PropertyComparator; import org.springframework.util.SerializationUtils; +import java.util.List; + import static org.assertj.core.api.Assertions.assertThat; +import static org.junit.Assert.assertTrue; /** * @author Dave Syer @@ -40,4 +45,37 @@ public class VetTests { assertThat(other.getId()).isEqualTo(vet.getId()); } + @Test + public void testGetSpecialties(){ + Vet vet = new Vet(); + vet.setFirstName("Zaphod"); + vet.setLastName("Beeblebrox"); + vet.setId(123); + Specialty radiology = new Specialty(); + Specialty urinalysis = new Specialty(); + Specialty kinesiology = new Specialty(); + vet.addSpecialty(radiology); + vet.addSpecialty(urinalysis); + //vet.addSpecialty(kinesiology); + + List testList = vet.getSpecialties(); + assertThat(testList).isEqualTo(vet.getSpecialties()); + testList.getClass().getSimpleName().equals("UnmodifiableCollection"); + } + + @Test + public void testAddGetNrSpecialty(){ + Vet vet = new Vet(); + vet.setFirstName("Zaphod"); + vet.setLastName("Beeblebrox"); + vet.setId(123); + + int preAddedNumber = vet.getNrOfSpecialties(); + Specialty radiology = new Specialty(); + //radiology.setName("radiology"); + vet.addSpecialty(radiology); + assertThat(vet.getNrOfSpecialties()).isEqualTo(++preAddedNumber); + assertThat((vet.getSpecialtiesInternal()).contains(radiology)); + } + }