From 86b6bae6e6b562203b2033db026f2deaea1d5511 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E7=A6=BE=E5=87=A0=E6=B5=B7?= <zh56462271@qq.com>
Date: Mon, 25 May 2020 00:02:26 +0800
Subject: [PATCH] =?UTF-8?q?dao=E5=B1=82=E4=BF=AE=E6=94=B9=20=20=E5=8D=95?=
 =?UTF-8?q?=E5=85=83=E6=B5=8B=E8=AF=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../java/cn/celess/blog/entity/Visitor.java   |  1 +
 .../cn/celess/blog/mapper/VisitorMapper.java  |  2 +
 .../celess/blog/mapper/VisitorMapperTest.java | 42 +++++++++++++++++++
 3 files changed, 45 insertions(+)
 create mode 100644 src/test/java/cn/celess/blog/mapper/VisitorMapperTest.java

diff --git a/src/main/java/cn/celess/blog/entity/Visitor.java b/src/main/java/cn/celess/blog/entity/Visitor.java
index 2092745..5767cf1 100644
--- a/src/main/java/cn/celess/blog/entity/Visitor.java
+++ b/src/main/java/cn/celess/blog/entity/Visitor.java
@@ -15,6 +15,7 @@ public class Visitor {
     private String ip;
     private Date date;
     private String ua;
+    private boolean delete;
 
     public Visitor(String ip, Date date, String ua) {
         this.ip = ip;
diff --git a/src/main/java/cn/celess/blog/mapper/VisitorMapper.java b/src/main/java/cn/celess/blog/mapper/VisitorMapper.java
index ed84320..e47a661 100644
--- a/src/main/java/cn/celess/blog/mapper/VisitorMapper.java
+++ b/src/main/java/cn/celess/blog/mapper/VisitorMapper.java
@@ -20,5 +20,7 @@ public interface VisitorMapper {
 
     List<Visitor> findAll();
 
+    List<Visitor> findAllNotDeleted();
+
     long count();
 }
diff --git a/src/test/java/cn/celess/blog/mapper/VisitorMapperTest.java b/src/test/java/cn/celess/blog/mapper/VisitorMapperTest.java
new file mode 100644
index 0000000..d782c94
--- /dev/null
+++ b/src/test/java/cn/celess/blog/mapper/VisitorMapperTest.java
@@ -0,0 +1,42 @@
+package cn.celess.blog.mapper;
+
+import cn.celess.blog.BaseTest;
+import cn.celess.blog.entity.Visitor;
+import org.junit.Test;
+import org.springframework.beans.factory.annotation.Autowired;
+
+import java.util.Date;
+
+import static org.junit.Assert.*;
+
+public class VisitorMapperTest extends BaseTest {
+
+    @Autowired
+    VisitorMapper visitorMapper;
+
+    @Test
+    public void insert() {
+        Visitor visitor = new Visitor();
+        visitor.setDate(new Date());
+        visitor.setIp("127.0.0.1");
+        visitor.setUa("ua");
+        assertEquals(1, visitorMapper.insert(visitor));
+    }
+
+    @Test
+    public void delete() {
+        Visitor visitor = new Visitor();
+        visitor.setDate(new Date());
+        visitor.setIp("127.0.0.1");
+        visitor.setUa("ua");
+        visitorMapper.insert(visitor);
+        assertEquals(1, visitorMapper.delete(visitor.getId()));
+    }
+
+    @Test
+    public void count() {
+        assertNotEquals(0, visitorMapper.count());
+    }
+
+
+}
\ No newline at end of file