add method to judge if account is admin (#78)

* add method to judge if account is admin

Author: Wanghb1

rbac/account.go

@@ -80,3 +80,16 @@ type AuthUser struct {
 	Username string `json:"name,omitempty"`
 	Password string `json:"password,omitempty"`
 }
+
+func (a *Account) HasAdminRole() bool {
+	if a.Role == RoleAdmin {
+		return true
+	}
+
+	for _, r := range a.Roles {
+		if r == RoleAdmin {
+			return true
+		}
+	}
+	return false
+}

rbac/account_test.go

@@ -17,7 +17,9 @@
 
 package rbac
 
-import "testing"
+import (
+	"testing"
+)
 
 func TestAccount_Check(t *testing.T) {
 	type fields struct {
@@ -73,3 +75,34 @@ func TestAccount_Check(t *testing.T) {
 		}
 	}
 }
+
+func TestAccount_HasAdminRole(t *testing.T) {
+	tests := []struct {
+		name string
+		a    *Account
+		want bool
+	}{
+		{
+			name: "compatible with old",
+			a:    &Account{Role: RoleAdmin},
+			want: true,
+		},
+		{
+			name: "has admin",
+			a:    &Account{Roles: []string{RoleAdmin, RoleDeveloper}},
+			want: true,
+		},
+		{
+			name: "not admin",
+			a:    &Account{Roles: []string{RoleDeveloper}},
+			want: false,
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			if got := tt.a.HasAdminRole(); got != tt.want {
+				t.Errorf("Account.HasAdminRole() = %v, want %v", got, tt.want)
+			}
+		})
+	}
+}