Skip to content

Commit

Permalink
✅ update test compile fail
Browse files Browse the repository at this point in the history
  • Loading branch information
hellokaton committed Aug 22, 2018
1 parent 8fee70b commit c4f0ddc
Showing 1 changed file with 16 additions and 8 deletions.
24 changes: 16 additions & 8 deletions src/test/java/com/blade/security/BasicAuthMiddlewareTest.java
Original file line number Diff line number Diff line change
Expand Up @@ -9,6 +9,7 @@
import com.blade.mvc.http.Request;
import com.blade.mvc.http.Response;
import com.blade.mvc.route.Route;
import com.blade.security.web.auth.AuthOption;
import com.blade.security.web.auth.BasicAuthMiddleware;
import lombok.extern.slf4j.Slf4j;
import org.junit.Test;
Expand All @@ -17,6 +18,8 @@
import java.util.Map;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertTrue;
import static org.mockito.Mockito.when;

/**
Expand All @@ -39,13 +42,12 @@ public void handle(RouteContext context) {
@Test
public void testAuthSuccess() throws Exception {

Request mockRequest = mockHttpRequest("GET");
Request mockRequest = mockHttpRequest("GET");

WebContext.init(Blade.of(), "/");


Map<String, String> headers = new HashMap<>();
headers.put("Authorization", "Basic YmxhZGU6YmxhZGU=");
headers.put("Authorization", "Basic YWRtaW46MTIzNDU2");

when(mockRequest.parameters()).thenReturn(new HashMap<>());
when(mockRequest.headers()).thenReturn(headers);
Expand All @@ -61,14 +63,17 @@ public void testAuthSuccess() throws Exception {

WebContext.set(new WebContext(request, response));

BasicAuthMiddleware basicAuthMiddleware = new BasicAuthMiddleware();
AuthOption authOption = AuthOption.builder().build();
authOption.addUser("admin", "123456");

BasicAuthMiddleware basicAuthMiddleware = new BasicAuthMiddleware(authOption);
boolean flag = basicAuthMiddleware.before(context);
assertEquals(false, flag);
assertTrue(flag);
}

@Test
public void testAuthFail() throws Exception {
Request mockRequest = mockHttpRequest("GET");
Request mockRequest = mockHttpRequest("GET");

WebContext.init(Blade.of(), "/");

Expand All @@ -90,9 +95,12 @@ public void testAuthFail() throws Exception {

WebContext.set(new WebContext(request, response));

BasicAuthMiddleware basicAuthMiddleware = new BasicAuthMiddleware();
AuthOption authOption = AuthOption.builder().build();
authOption.addUser("admin", "123456");

BasicAuthMiddleware basicAuthMiddleware = new BasicAuthMiddleware(authOption);
boolean flag = basicAuthMiddleware.before(context);
assertEquals(false, flag);
assertFalse(flag);
}

}

0 comments on commit c4f0ddc

Please sign in to comment.