BAEL-3069 Rename the folder name to spring-persistence-simple-2

2020-03-21 11:57:30 -06:00
parent c986147850
commit 9f3c94929a
7 changed files with 2 additions and 2 deletions
@@ -0,0 +1,44 @@
+package com.baeldung.jdbc;
+
+public class Employee {
+    private int id;
+
+    private String firstName;
+
+    private String lastName;
+
+    private String address;
+
+    public int getId() {
+        return id;
+    }
+
+    public void setId(final int id) {
+        this.id = id;
+    }
+
+    public String getFirstName() {
+        return firstName;
+    }
+
+    public void setFirstName(final String firstName) {
+        this.firstName = firstName;
+    }
+
+    public String getLastName() {
+        return lastName;
+    }
+
+    public void setLastName(final String lastName) {
+        this.lastName = lastName;
+    }
+
+    public String getAddress() {
+        return address;
+    }
+
+    public void setAddress(final String address) {
+        this.address = address;
+    }
+
+}
@@ -0,0 +1,21 @@
+package com.baeldung.jdbc;
+
+import javax.sql.DataSource;
+
+import org.springframework.jdbc.core.JdbcTemplate;
+import org.springframework.stereotype.Repository;
+
+@Repository
+public class EmployeeDAO {
+    private JdbcTemplate jdbcTemplate;
+
+    public void setDataSource(final DataSource dataSource) {
+        jdbcTemplate = new JdbcTemplate(dataSource);
+    }
+
+    public int getCountOfEmployees() {
+        return jdbcTemplate.queryForObject("SELECT COUNT(*) FROM EMPLOYEE", Integer.class);
+    }
+
+   
+}
@@ -0,0 +1,7 @@
+CREATE TABLE EMPLOYEE
+(
+    ID int NOT NULL PRIMARY KEY,
+    FIRST_NAME varchar(255),
+    LAST_NAME varchar(255),
+    ADDRESS varchar(255)
+);
@@ -0,0 +1,4 @@
+INSERT INTO EMPLOYEE VALUES (1, 'James', 'Gosling', 'Canada');
+INSERT INTO EMPLOYEE VALUES (2, 'Donald', 'Knuth', 'USA');
+INSERT INTO EMPLOYEE VALUES (3, 'Linus', 'Torvalds', 'Finland');
+INSERT INTO EMPLOYEE VALUES (4, 'Dennis', 'Ritchie', 'USA');
@@ -0,0 +1,47 @@
+package com.baeldung.jdbc;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+
+import javax.sql.DataSource;
+
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.mockito.Mock;
+import org.mockito.Mockito;
+import org.mockito.junit.MockitoJUnitRunner;
+import org.springframework.jdbc.core.JdbcTemplate;
+import org.springframework.jdbc.datasource.embedded.EmbeddedDatabaseBuilder;
+import org.springframework.jdbc.datasource.embedded.EmbeddedDatabaseType;
+import org.springframework.test.util.ReflectionTestUtils;
+
+
+@RunWith(MockitoJUnitRunner.class)
+
+public class EmployeeDAOUnitTest {
+    @Mock
+    JdbcTemplate jdbcTemplate;
+
+    @Test
+    public void whenMockJdbcTemplate_thenReturnCorrectEmployeeCount() {
+        EmployeeDAO employeeDAO = new EmployeeDAO();
+        ReflectionTestUtils.setField(employeeDAO, "jdbcTemplate", jdbcTemplate);
+        Mockito.when(jdbcTemplate.queryForObject("SELECT COUNT(*) FROM EMPLOYEE", Integer.class))
+          .thenReturn(4);
+        assertEquals(4, employeeDAO.getCountOfEmployees());
+
+        Mockito.when(jdbcTemplate.queryForObject(Mockito.anyString(), Mockito.eq(Integer.class)))
+          .thenReturn(3);
+        assertEquals(3, employeeDAO.getCountOfEmployees());
+    }
+
+    @Test
+    public void whenInjectInMemoryDataSource_thenReturnCorrectEmployeeCount() {
+        DataSource dataSource = new EmbeddedDatabaseBuilder().setType(EmbeddedDatabaseType.H2)
+          .addScript("classpath:jdbc/schema.sql")
+          .addScript("classpath:jdbc/test-data.sql")
+          .build();
+        EmployeeDAO employeeDAO = new EmployeeDAO();
+        employeeDAO.setDataSource(dataSource);
+        assertEquals(4, employeeDAO.getCountOfEmployees());
+    }
+}