Unit test PollController

Mockito is lackluster, but i missed you, junit <3

pom.xml

@@ -10,13 +10,13 @@
     <parent>
         <groupId>org.springframework.boot</groupId>
         <artifactId>spring-boot-starter-parent</artifactId>
-        <version>1.5.3.RELEASE</version>
+        <version>2.0.0.RELEASE</version>
         <relativePath/> <!-- lookup parent from repository -->
     </parent>
     <properties>
         <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
         <start-class>io.zipcoder.tc_spring_poll_application.QuickPollApplication</start-class>
-        <java.version>1.7</java.version>
+        <java.version>9</java.version>
     </properties>
     <dependencies>
         <dependency>

src/main/java/io/zipcoder/tc_spring_poll_application/controllers/PollController.java

@@ -0,0 +1,34 @@
+package io.zipcoder.tc_spring_poll_application.controllers;
+
+import io.zipcoder.tc_spring_poll_application.domain.Poll;
+import io.zipcoder.tc_spring_poll_application.repositories.PollRepository;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.http.HttpStatus;
+import org.springframework.http.ResponseEntity;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RequestMethod;
+import org.springframework.web.bind.annotation.RestController;
+
+/**
+ * project: spring-demo
+ * package: io.zipcoder.tc_spring_poll_application.controllers
+ * author: https://github.com/vvmk
+ * date: 4/5/18
+ */
+@RestController
+public class PollController {
+
+    private PollRepository pollRepository;
+
+    @Autowired
+    public PollController(PollRepository pollRepository) {
+        this.pollRepository = pollRepository;
+    }
+
+    @RequestMapping(name = "/polls", method = RequestMethod.GET)
+    public ResponseEntity<Iterable<Poll>> getAllPolls() {
+        Iterable<Poll> allPolls = pollRepository.findAll();
+        return new ResponseEntity<Iterable<Poll>>(allPolls, HttpStatus.OK);
+    }
+
+}

src/main/java/io/zipcoder/tc_spring_poll_application/domain/Poll.java

@@ -25,6 +25,10 @@ public class Poll {
     @OrderBy
     private Set<Option> options;
 
+    public Poll() {
+        // -_-
+    }
+
     public Long getId() {
         return Id;
     }

src/test/java/io/zipcoder/tc_spring_poll_application/controllers/PollControllerTest.java

@@ -0,0 +1,62 @@
+package io.zipcoder.tc_spring_poll_application.controllers;
+
+import io.zipcoder.tc_spring_poll_application.domain.Poll;
+import io.zipcoder.tc_spring_poll_application.repositories.PollRepository;
+import org.junit.Before;
+import org.junit.Test;
+import org.mockito.InjectMocks;
+import org.mockito.Mock;
+import org.mockito.Mockito;
+import org.mockito.MockitoAnnotations;
+import org.springframework.util.Assert;
+
+import java.util.ArrayList;
+import java.util.List;
+
+import static org.junit.Assert.assertEquals;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.verify;
+
+/**
+ * project: spring-demo
+ * package: io.zipcoder.tc_spring_poll_application.controllers
+ * author: https://github.com/vvmk
+ * date: 4/5/18
+ */
+public class PollControllerTest {
+
+    @InjectMocks
+    private PollController pollCtrl;
+
+    @Mock
+    private PollRepository pollRepo;
+
+    private List<Poll> polls = new ArrayList<>();
+
+    @Before
+    public void setup() {
+        MockitoAnnotations.initMocks(this);
+
+        Mockito.when(pollRepo.findAll()).thenReturn(polls);
+    }
+
+    @Test
+    public void getAllPollsReturnsIterable() {
+        Class<?> expected = Iterable.class;
+        Class<?> actual = pollCtrl.getAllPolls().getBody().getClass();
+
+        verify(pollRepo).findAll();
+        Assert.isAssignable(expected, actual);
+    }
+
+    @Test
+    public void getAllPollsReturnsPolls() {
+        polls.add(mock(Poll.class));
+
+        long expected = 1;
+        long actual = pollCtrl.getAllPolls().getBody().spliterator().getExactSizeIfKnown();
+
+        verify(pollRepo).findAll();
+        assertEquals(expected, actual);
+    }
+}