diff --git a/src/main/java/org/springframework/samples/petclinic/newDataStore/NewOwnerStore.java b/src/main/java/org/springframework/samples/petclinic/newDataStore/NewOwnerStore.java index bd55e94e6..48beb2900 100644 --- a/src/main/java/org/springframework/samples/petclinic/newDataStore/NewOwnerStore.java +++ b/src/main/java/org/springframework/samples/petclinic/newDataStore/NewOwnerStore.java @@ -33,4 +33,8 @@ public class NewOwnerStore { public StaticOwner convertToStaticOwner(Owner ownerEntity) { return new StaticOwner(ownerEntity.getAddress(), ownerEntity.getCity(), ownerEntity.getTelephone()); } + + public Map getNewOwnerStore() { + return this.ownerStore; + } } diff --git a/src/main/java/org/springframework/samples/petclinic/owner/StaticOwner.java b/src/main/java/org/springframework/samples/petclinic/owner/StaticOwner.java index 2513bebde..0f9ab2a4a 100644 --- a/src/main/java/org/springframework/samples/petclinic/owner/StaticOwner.java +++ b/src/main/java/org/springframework/samples/petclinic/owner/StaticOwner.java @@ -17,4 +17,17 @@ public class StaticOwner { this.city = city; this.telephone = telephone; } + + public String getAddress () { + return address; + } + + public String getCity() { + return city; + } + + public String getTelephone () { + return telephone; + } + } diff --git a/src/test/java/org/springframework/samples/petclinic/newDataStore/NewOwnerStoreTest.java b/src/test/java/org/springframework/samples/petclinic/newDataStore/NewOwnerStoreTest.java new file mode 100644 index 000000000..7c7c26ca1 --- /dev/null +++ b/src/test/java/org/springframework/samples/petclinic/newDataStore/NewOwnerStoreTest.java @@ -0,0 +1,44 @@ +/* + * Software property of Acquisio. Copyright 2003-2018. + */ +package org.springframework.samples.petclinic.newDataStore; + +import java.util.Iterator; +import java.util.Map; +import org.junit.Test; +import org.junit.runner.RunWith; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.boot.test.autoconfigure.orm.jpa.DataJpaTest; +import org.springframework.samples.petclinic.owner.OwnerRepository; +import org.springframework.samples.petclinic.owner.StaticOwner; +import org.springframework.test.context.junit4.SpringRunner; + +/** + * @author Gibran + */ +@RunWith(SpringRunner.class) +@DataJpaTest +public class NewOwnerStoreTest { + + @Autowired + OwnerRepository owner; + + NewOwnerStore testOwnerStore; + + @Test + public void testPopulation() { + testOwnerStore = new NewOwnerStore(owner); + testOwnerStore.populateStore(); + Map ownerStore = testOwnerStore.getNewOwnerStore(); + + for (Integer id: ownerStore.keySet()){ + + Integer key = id; + String value = ownerStore.get(id).getAddress(); + System.out.println(key + " " + value); + } + + + } + +}