From 23b1d2bb5836c7aa8ea6b4bec9b4a31b08ace6f4 Mon Sep 17 00:00:00 2001
From: Amanda <amawai@github.com>
Date: Sat, 24 Feb 2018 17:47:17 -0500
Subject: [PATCH] Added test for Visit.java and VisitRepository.java

---
 .../petclinic/visit/VisitRepositoryTest.java  | 51 +++++++++++++++++++
 .../samples/petclinic/visit/VisitTest.java    | 31 +++++++++++
 2 files changed, 82 insertions(+)
 create mode 100644 src/test/java/org/springframework/samples/petclinic/visit/VisitRepositoryTest.java
 create mode 100644 src/test/java/org/springframework/samples/petclinic/visit/VisitTest.java

diff --git a/src/test/java/org/springframework/samples/petclinic/visit/VisitRepositoryTest.java b/src/test/java/org/springframework/samples/petclinic/visit/VisitRepositoryTest.java
new file mode 100644
index 000000000..5e096df7e
--- /dev/null
+++ b/src/test/java/org/springframework/samples/petclinic/visit/VisitRepositoryTest.java
@@ -0,0 +1,51 @@
+package org.springframework.samples.petclinic.visit;
+
+import java.util.Date;
+import java.util.List;
+
+import static org.junit.Assert.*;
+
+import org.junit.Before;
+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.boot.test.mock.mockito.MockBean;
+import org.springframework.context.annotation.ComponentScan;
+import org.springframework.stereotype.Service;
+import org.springframework.test.context.junit4.SpringRunner;
+
+@RunWith(SpringRunner.class)
+@DataJpaTest(includeFilters = @ComponentScan.Filter(Service.class))
+public class VisitRepositoryTest {
+	private static final int VISIT_ID = 1;
+	private static final String DESCRIPTION = "Visiting a test case";
+	private static final Date TODAY = new Date();
+	private Visit visit;
+	private List<Visit> visitList;
+	
+	@Autowired
+	private VisitRepository visitRepository;
+	
+	@Before
+	public void setUp(){
+		visit = new Visit();
+		visit.setDescription(DESCRIPTION);
+		visit.setPetId(VISIT_ID);
+		visit.setDate(TODAY);
+	}
+
+	@Test
+	public void shouldFindSavedVisitInVisitRepository() {
+		visitRepository.save(visit);
+		visitList = visitRepository.findByPetId(VISIT_ID);
+		
+		assertEquals(visitList.size(), 1);
+		Visit savedVisit = visitList.get(0);
+		assertEquals((int) savedVisit.getPetId(), VISIT_ID);
+		assertEquals(savedVisit.getDescription(), DESCRIPTION);
+		assertEquals(savedVisit.getDate(), TODAY);
+	}
+	
+
+}
diff --git a/src/test/java/org/springframework/samples/petclinic/visit/VisitTest.java b/src/test/java/org/springframework/samples/petclinic/visit/VisitTest.java
new file mode 100644
index 000000000..97d550d36
--- /dev/null
+++ b/src/test/java/org/springframework/samples/petclinic/visit/VisitTest.java
@@ -0,0 +1,31 @@
+package org.springframework.samples.petclinic.visit;
+
+import java.util.Date;
+import static org.junit.Assert.*;
+import org.junit.Test;
+
+public class VisitTest {
+	private Visit visit;
+
+	@Test
+	public void testGetDate() {
+		Date today = new Date();
+		visit.setDate(today);
+		assertEquals(visit.getDate(), today);
+	}
+	
+	@Test
+	public void testGetDescription() {
+		String description = "The greatest visit of all time";
+		visit.setDescription(description);
+		assertEquals(visit.getDescription(), description);
+	}
+	
+	@Test
+	public void testGetPetId() {
+		int petId = 1;
+		visit.setPetId(petId);
+		assertEquals((int) visit.getPetId(), petId);
+	}
+
+}