junit 5 example (#2050)

2017-06-12 11:20:42 +03:00
parent 080a4234a9
commit b6d47b00a3
12 changed files with 582 additions and 0 deletions
--- a/guest/junit5-example/src/main/java/com/stackify/daos/UserDAO.java
+++ b/guest/junit5-example/src/main/java/com/stackify/daos/UserDAO.java
@@ -0,0 +1,141 @@
+package com.stackify.daos;
+
+import java.sql.Connection;
+import java.sql.PreparedStatement;
+import java.sql.ResultSet;
+import java.sql.SQLException;
+import java.util.ArrayList;
+import java.util.List;
+
+import org.apache.logging.log4j.LogManager;
+import org.apache.logging.log4j.Logger;
+
+import com.stackify.models.User;
+import com.stackify.utils.ConnectionUtil;
+
+public class UserDAO {
+
+    private Logger logger = LogManager.getLogger(UserDAO.class);
+
+    public void createTable() {
+        try (Connection con = ConnectionUtil.getConnection()) {
+            String createQuery = "CREATE TABLE users(email varchar(50) primary key, name varchar(50))";
+            PreparedStatement pstmt = con.prepareStatement(createQuery);
+
+            pstmt.execute();
+        } catch (SQLException exc) {
+            logger.error(exc.getMessage());
+        }
+
+    }
+
+    public void add(User user) {
+        try (Connection con = ConnectionUtil.getConnection()) {
+
+            String insertQuery = "INSERT INTO users(email,name) VALUES(?,?)";
+            PreparedStatement pstmt = con.prepareStatement(insertQuery);
+            pstmt.setString(1, user.getEmail());
+            pstmt.setString(2, user.getName());
+
+            pstmt.executeUpdate();
+        } catch (SQLException exc) {
+            logger.error(exc.getMessage());
+        }
+    }
+
+    public void update(User user) {
+        try (Connection con = ConnectionUtil.getConnection()) {
+
+            String updateQuery = "UPDATE users SET name=? WHERE email=?";
+            PreparedStatement pstmt = con.prepareStatement(updateQuery);
+            pstmt.setString(1, user.getName());
+            pstmt.setString(2, user.getEmail());
+
+            pstmt.executeUpdate();
+        } catch (SQLException exc) {
+            logger.error(exc.getMessage());
+        }
+    }
+
+    public void delete(User user) {
+        try (Connection con = ConnectionUtil.getConnection()) {
+
+            String deleteQuery = "DELETE FROM users WHERE email=?";
+            PreparedStatement pstmt = con.prepareStatement(deleteQuery);
+            pstmt.setString(1, user.getEmail());
+
+            pstmt.executeUpdate();
+        } catch (SQLException exc) {
+            logger.error(exc.getMessage());
+        }
+    }
+
+    public void delete(String email) {
+        try (Connection con = ConnectionUtil.getConnection()) {
+
+            String deleteQuery = "DELETE FROM users WHERE email=?";
+            PreparedStatement pstmt = con.prepareStatement(deleteQuery);
+            pstmt.setString(1, email);
+
+            pstmt.executeUpdate();
+        } catch (SQLException exc) {
+            logger.error(exc.getMessage());
+        }
+    }
+
+    public User findOne(String email) {
+        User user = null;
+
+        try (Connection con = ConnectionUtil.getConnection()) {
+            String query = "SELECT * FROM users WHERE email=?";
+            PreparedStatement pstmt = con.prepareStatement(query);
+            pstmt.setString(1, email);
+
+            ResultSet rs = pstmt.executeQuery();
+            while (rs.next()) {
+                user = new User();
+                user.setEmail(rs.getString("email"));
+                user.setName(rs.getString("name"));
+            }
+
+        } catch (SQLException exc) {
+            logger.error(exc.getMessage());
+        }
+
+        return user;
+    }
+
+    public List<User> findAll() {
+        List<User> users = new ArrayList<>();
+
+        try (Connection con = ConnectionUtil.getConnection()) {
+            String query = "SELECT * FROM users";
+            PreparedStatement pstmt = con.prepareStatement(query);
+
+            ResultSet rs = pstmt.executeQuery();
+            while (rs.next()) {
+                User user = new User();
+                user.setEmail(rs.getString("email"));
+                user.setName(rs.getString("name"));
+                users.add(user);
+            }
+        } catch (SQLException exc) {
+            logger.error(exc.getMessage());
+        }
+
+        return users;
+    }
+
+    public void deleteAll() {
+        try (Connection con = ConnectionUtil.getConnection()) {
+
+            String deleteQuery = "DELETE FROM users";
+            PreparedStatement pstmt = con.prepareStatement(deleteQuery);
+
+            pstmt.executeUpdate();
+        } catch (SQLException exc) {
+            logger.error(exc.getMessage());
+        }
+    }
+
+}