add mock

descope · Apr 4, 2024 · b104e08 · b104e08
1 parent bb6fff4
commit b104e08
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/descope/tests/mocks/mgmt/managementmock.go b/descope/tests/mocks/mgmt/managementmock.go
@@ -1280,6 +1280,10 @@ type MockAuthz struct {
 	WhatCanTargetAccessResponse []*descope.AuthzRelation
 	WhatCanTargetAccessError    error
 
+	WhatCanTargetAccessWithRelationAssert   func(target string)
+	WhatCanTargetAccessWithRelationResponse []*descope.AuthzRelation
+	WhatCanTargetAccessWithRelationError    error
+
 	GetModifiedAssert   func(since time.Time)
 	GetModifiedResponse *descope.AuthzModified
 	GetModifiedError    error
@@ -1384,6 +1388,13 @@ func (m *MockAuthz) WhatCanTargetAccess(_ context.Context, target string) ([]*de
 	return m.WhatCanTargetAccessResponse, m.WhatCanTargetAccessError
 }
 
+func (m *MockAuthz) WhatCanTargetAccessWithRelation(_ context.Context, target, relationDefinition, namespace string) ([]*descope.AuthzRelation, error) {
+	if m.WhatCanTargetAccessWithRelationAssert != nil {
+		m.WhatCanTargetAccessWithRelationAssert(target)
+	}
+	return m.WhatCanTargetAccessWithRelationResponse, m.WhatCanTargetAccessWithRelationError
+}
+
 func (m *MockAuthz) GetModified(_ context.Context, since time.Time) (*descope.AuthzModified, error) {
 	if m.GetModifiedAssert != nil {
 		m.GetModifiedAssert(since)