From 88bc05311ab31ba92a6d16457ace3982db4fa5b1 Mon Sep 17 00:00:00 2001
From: SunXP <roger_lu_dude@hotmail.com>
Date: Sat, 24 Feb 2018 04:51:04 -0500
Subject: [PATCH] adding more unit tests to for the Owner class

---
 .../samples/petclinic/owner/OwnerTest.java    | 33 +++++++++++++++++++
 1 file changed, 33 insertions(+)

diff --git a/src/test/java/org/springframework/samples/petclinic/owner/OwnerTest.java b/src/test/java/org/springframework/samples/petclinic/owner/OwnerTest.java
index d373fe7eb..d09c5c1a0 100644
--- a/src/test/java/org/springframework/samples/petclinic/owner/OwnerTest.java
+++ b/src/test/java/org/springframework/samples/petclinic/owner/OwnerTest.java
@@ -2,6 +2,7 @@ package org.springframework.samples.petclinic.owner;
 
 import static org.junit.Assert.*;
 
+import java.util.ArrayList;
 import java.util.List;
 
 import org.junit.Before;
@@ -63,4 +64,36 @@ public class OwnerTest {
 		assertEquals(pet, onlyPet);
 	}
 
+	@Test
+	public void getPetTest() {	
+		Pet pet = new Pet();
+		pet.setName("Pochi");
+		instance.addPet(pet);
+		
+		assertEquals(pet, instance.getPet("Pochi"));	
+		assertEquals(pet, instance.getPet("Pochi", false)); //tests pet object exists
+		assertEquals(null, instance.getPet("Pochi", true));	//tests pet object doesn't exist
+	}
+	
+	@Test
+	public void getPetsTest() {
+		Pet pet = new Pet();
+		List<Pet> list = new ArrayList<>();
+		list.add(pet);
+		instance.addPet(pet);
+		
+		assertEquals(list, instance.getPets());
+		
+		Pet pet2 = new Pet();
+		list.add(pet2);
+		instance.addPet(pet2);
+		
+		assertEquals(list, instance.getPets());
+	}
+	
+	@Test
+	public void setGetAddress() {
+		instance.setAddress("123 FakeStreet");
+		assertEquals("123 FakeStreet", instance.getAddress());
+	}
 }
\ No newline at end of file