feat: add banned feature and api IsLoginByToken()

This commit is contained in:
weloe
2023-05-24 22:10:07 +08:00
parent 82680c3ce4
commit 8bf577c030
7 changed files with 139 additions and 11 deletions

View File

@@ -419,3 +419,55 @@ func TestEnforcer_JsonAdapter(t *testing.T) {
}
}
func TestEnforcer_Banned(t *testing.T) {
err, enforcer, _ := NewTestEnforcer(t)
if err != nil {
t.Fatalf("NewTestEnforcer() failed: %v", err)
}
err = enforcer.Banned("1", "comment", 1, 100)
if err != nil {
t.Fatalf("Banned() failed: %v", err)
}
isBanned := enforcer.IsBanned("1", "comment")
if !isBanned {
t.Errorf("unexpected isBanned is false")
}
level, err := enforcer.GetBannedLevel("1", "comment")
if err != nil {
t.Errorf("GetBannedLevel() failed: %v", err)
}
if level != 1 {
t.Errorf("unexpected banned level = %v", level)
}
err = enforcer.UnBanned("1", "comment")
if err != nil {
t.Fatalf("UnBanned() failed: %v", err)
}
isBanned = enforcer.IsBanned("1", "comment")
if isBanned {
t.Errorf("unexpected isBanned is false")
}
}
func TestEnforcer_GetBannedTime(t *testing.T) {
err, enforcer, _ := NewTestEnforcer(t)
if err != nil {
t.Fatalf("NewTestEnforcer() failed: %v", err)
}
err = enforcer.Banned("1", "comment", 1, 100)
if err != nil {
t.Fatalf("Banned() failed: %v", err)
}
t.Logf("banned time = %v", enforcer.GetBannedTime("1", "comment"))
err = enforcer.Banned("1", "comment", 1, -1)
if err != nil {
t.Fatalf("Banned() failed: %v", err)
}
t.Logf("banned time = %v", enforcer.GetBannedTime("1", "comment"))
}