From c81bafec16bdb548cd72fe9754a29c8ce368c8a9 Mon Sep 17 00:00:00 2001 From: Armon Dadgar Date: Sat, 19 Aug 2017 16:43:49 -0700 Subject: [PATCH] acl: Adding IsManagement check --- acl/acl.go | 5 +++++ acl/acl_test.go | 4 ++++ 2 files changed, 9 insertions(+) diff --git a/acl/acl.go b/acl/acl.go index 37d573540..12100966a 100644 --- a/acl/acl.go +++ b/acl/acl.go @@ -204,3 +204,8 @@ func (a *ACL) AllowOperatorWrite() bool { return false } } + +// IsManagement checks if this represents a management token +func (a *ACL) IsManagement() bool { + return a.management +} diff --git a/acl/acl_test.go b/acl/acl_test.go index 326804769..86b2c20e2 100644 --- a/acl/acl_test.go +++ b/acl/acl_test.go @@ -72,6 +72,7 @@ func TestACLManagement(t *testing.T) { assert.Equal(t, true, acl.AllowNamespaceOperation("foo", NamespaceCapabilityListJobs)) // Check the other simpler operations + assert.Equal(t, true, acl.IsManagement()) assert.Equal(t, true, acl.AllowAgentRead()) assert.Equal(t, true, acl.AllowAgentWrite()) assert.Equal(t, true, acl.AllowNodeRead()) @@ -97,6 +98,7 @@ func TestACLMerge(t *testing.T) { assert.Equal(t, false, acl.AllowNamespaceOperation("foo", NamespaceCapabilityListJobs)) // Check the other simpler operations + assert.Equal(t, false, acl.IsManagement()) assert.Equal(t, true, acl.AllowAgentRead()) assert.Equal(t, true, acl.AllowAgentWrite()) assert.Equal(t, true, acl.AllowNodeRead()) @@ -118,6 +120,7 @@ func TestACLMerge(t *testing.T) { assert.Equal(t, false, acl.AllowNamespaceOperation("foo", NamespaceCapabilityListJobs)) // Check the other simpler operations + assert.Equal(t, false, acl.IsManagement()) assert.Equal(t, true, acl.AllowAgentRead()) assert.Equal(t, false, acl.AllowAgentWrite()) assert.Equal(t, true, acl.AllowNodeRead()) @@ -139,6 +142,7 @@ func TestACLMerge(t *testing.T) { assert.Equal(t, false, acl.AllowNamespaceOperation("foo", NamespaceCapabilityListJobs)) // Check the other simpler operations + assert.Equal(t, false, acl.IsManagement()) assert.Equal(t, false, acl.AllowAgentRead()) assert.Equal(t, false, acl.AllowAgentWrite()) assert.Equal(t, false, acl.AllowNodeRead())