修改JwtUtil的部分方法

src/test/java/cn/celess/blog/util/JwtUtilTest.java

@@ -2,51 +2,79 @@ package cn.celess.blog.util;
 
 import cn.celess.blog.BaseTest;
 import cn.celess.blog.entity.User;
+import io.jsonwebtoken.Jwts;
+import io.jsonwebtoken.SignatureAlgorithm;
+import org.junit.FixMethodOrder;
 import org.junit.Test;
+import org.junit.runners.MethodSorters;
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.beans.factory.annotation.Value;
+
+import java.time.Instant;
+import java.util.Date;
 
 import static org.junit.Assert.*;
 
+@FixMethodOrder(MethodSorters.NAME_ASCENDING)
 public class JwtUtilTest extends BaseTest {
 
     @Autowired
     JwtUtil jwtUtil;
-
+    @Value("${jwt.secret}")
+    private String secret;
 
     @Test
-    public void generateToken() {
+    public void testGenerateToken() {
         User user = new User();
         user.setEmail("a@celess.cn");
-        String s = jwtUtil.generateToken(user, true);
-        System.out.println(s);
+        String s = jwtUtil.generateToken(user, false);
         assertNotNull(s);
+        String str = null;
+        try {
+            str = jwtUtil.generateToken(null, false);
+        } catch (Exception e) {
+            // ignore
+        }
+        assertNull(str);
     }
 
     @Test
-    public void validateToken() {
+    public void testIsTokenExpired() throws InterruptedException {
+        String s = Jwts.builder()
+                .setClaims(null)
+                .setExpiration(new Date(Instant.now().toEpochMilli() + 1000))
+                .signWith(SignatureAlgorithm.HS512, secret)
+                .compact();
+        Thread.sleep(1010);
+        assertTrue(jwtUtil.isTokenExpired(s));
+        assertFalse(jwtUtil.isTokenExpired(jwtUtil.generateToken(new User(), false)));
+    }
+
+    @Test
+    public void testGetUsernameFromToken() {
         User user = new User();
         user.setEmail("a@celess.cn");
-        assertTrue(jwtUtil.validateToken(createToken(), user));
+        String s = jwtUtil.generateToken(user, false);
+        assertEquals(user.getEmail(), jwtUtil.getUsernameFromToken(s));
+        user.setEmail("example@celess.cn");
+        assertNotEquals(user.getEmail(), jwtUtil.getUsernameFromToken(s));
     }
 
     @Test
-    public void isTokenExpired() {
-        assertFalse(jwtUtil.isTokenExpired(createToken()));
-    }
-
-    @Test
-    public void getUsernameFromToken() {
-        assertEquals("a@celess.cn", jwtUtil.getUsernameFromToken(createToken()));
-    }
-
-    @Test
-    public void getExpirationDateFromToken() {
-        assertNotNull(jwtUtil.getExpirationDateFromToken(createToken()));
-    }
-
-    private String createToken() {
+    public void testGetExpirationDateFromToken() {
         User user = new User();
         user.setEmail("a@celess.cn");
-        return jwtUtil.generateToken(user, true);
+        String s = jwtUtil.generateToken(user, false);
+        assertNotNull(jwtUtil.getExpirationDateFromToken(s));
+    }
+
+    @Test
+    public void updateTokenDate() {
+        User user = new User();
+        user.setEmail("a@celess.cn");
+        String s = jwtUtil.generateToken(user, false);
+        Date before = jwtUtil.getExpirationDateFromToken(s);
+        String s1 = jwtUtil.updateTokenDate(s);
+        assertTrue(jwtUtil.getExpirationDateFromToken(s1).getTime() - jwtUtil.getExpirationDateFromToken(s).getTime() > 0);
     }
 }