Skip to content

Commit 08f2ed3

Browse files
committed
add test assert
1 parent 4a04315 commit 08f2ed3

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

springboot-starter-data-authorization/src/test/java/com/codingapi/springboot/authorization/DataAuthorizationContextTest.java

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,6 @@
66
import com.codingapi.springboot.authorization.entity.User;
77
import com.codingapi.springboot.authorization.filter.DefaultDataAuthorizationFilter;
88
import com.codingapi.springboot.authorization.handler.Condition;
9-
import com.codingapi.springboot.authorization.interceptor.SQLRunningContext;
109
import com.codingapi.springboot.authorization.mask.ColumnMaskContext;
1110
import com.codingapi.springboot.authorization.mask.impl.BankCardMask;
1211
import com.codingapi.springboot.authorization.mask.impl.IDCardMask;
@@ -76,8 +75,9 @@ public Condition rowAuthorization(String tableName, String tableAlias) {
7675
String conditionTemplate = "%s.id = " + departId;
7776

7877
// 在条件处理的过程中,执行的查询都将不会被拦截
79-
List<Depart> departs = departRepository.findAll();
78+
List<Depart> departs = departRepository.findAll();
8079
log.info("departs:{}", departs);
80+
assertEquals(2, departs.size());
8181

8282
return Condition.formatCondition(conditionTemplate, tableAlias);
8383
}
@@ -187,7 +187,7 @@ public boolean supportColumnAuthorization(String tableName, String columnName, O
187187

188188
PageRequest request = PageRequest.of(0, 100);
189189
Page<User> users = userRepository.findAll(request);
190-
assertTrue( users.getTotalElements()>=3);
190+
assertTrue(users.getTotalElements() >= 3);
191191

192192
for (User user : users) {
193193
assertEquals("138****5678", user.getPhone());
@@ -196,7 +196,7 @@ public boolean supportColumnAuthorization(String tableName, String columnName, O
196196
CurrentUser.getInstance().setUser(lorne);
197197

198198
users = userRepository.findAll(request);
199-
assertTrue( users.getTotalElements()>=3);
199+
assertTrue(users.getTotalElements() >= 3);
200200

201201
for (User user : users) {
202202
assertEquals("13812345678", user.getPhone());

0 commit comments

Comments
 (0)