Merge pull request #1611 from micro/auth-has-role

Auth account.HasRole

auth/auth.go

@@ -72,6 +72,21 @@ type Account struct {
 	Secret string `json:"secret"`
 }
 
+// HasRole returns a boolean indicating if the account has the given role
+func (a *Account) HasRole(role string) bool {
+	if a.Roles == nil {
+		return false
+	}
+
+	for _, r := range a.Roles {
+		if r == role {
+			return true
+		}
+	}
+
+	return false
+}
+
 // Token can be short or long lived
 type Token struct {
 	// The token to be used for accessing resources

auth/auth_test.go

@@ -0,0 +1,17 @@
+package auth
+
+import "testing"
+
+func TestHasRole(t *testing.T) {
+	if new(Account).HasRole("foo") {
+		t.Errorf("Expected the blank account to not have a role")
+	}
+
+	acc := Account{Roles: []string{"foo"}}
+	if !acc.HasRole("foo") {
+		t.Errorf("Expected the account to have the foo role")
+	}
+	if acc.HasRole("bar") {
+		t.Errorf("Expected the account to not have the bar role")
+	}
+}