From 34f18aacd6fbfefe3758769d7769029395832895 Mon Sep 17 00:00:00 2001
From: Artur Wierzbicki <wierzbicki.artur.94@gmail.com>
Date: Thu, 29 Sep 2022 15:11:53 +0400
Subject: [PATCH] Chore: add more input parameters to
 `serviceaccounts/test/common.go` (#55994)

---
 pkg/services/serviceaccounts/tests/common.go | 20 ++++++++++++--------
 1 file changed, 12 insertions(+), 8 deletions(-)

diff --git a/pkg/services/serviceaccounts/tests/common.go b/pkg/services/serviceaccounts/tests/common.go
index 788d6675629..a82dbcb0016 100644
--- a/pkg/services/serviceaccounts/tests/common.go
+++ b/pkg/services/serviceaccounts/tests/common.go
@@ -20,14 +20,16 @@ type TestUser struct {
 	Role             string
 	Login            string
 	IsServiceAccount bool
+	OrgID            int64
 }
 
 type TestApiKey struct {
-	Name      string
-	Role      org.RoleType
-	OrgId     int64
-	Key       string
-	IsExpired bool
+	Name             string
+	Role             org.RoleType
+	OrgId            int64
+	Key              string
+	IsExpired        bool
+	ServiceAccountID *int64
 }
 
 func SetupUserServiceAccount(t *testing.T, sqlStore *sqlstore.SQLStore, testUser TestUser) *user.User {
@@ -41,6 +43,7 @@ func SetupUserServiceAccount(t *testing.T, sqlStore *sqlstore.SQLStore, testUser
 		IsServiceAccount: testUser.IsServiceAccount,
 		DefaultOrgRole:   role,
 		Name:             testUser.Name,
+		OrgID:            testUser.OrgID,
 	})
 	require.NoError(t, err)
 	return u1
@@ -53,9 +56,10 @@ func SetupApiKey(t *testing.T, sqlStore *sqlstore.SQLStore, testKey TestApiKey)
 	}
 
 	addKeyCmd := &apikey.AddCommand{
-		Name:  testKey.Name,
-		Role:  role,
-		OrgId: testKey.OrgId,
+		Name:             testKey.Name,
+		Role:             role,
+		OrgId:            testKey.OrgId,
+		ServiceAccountID: testKey.ServiceAccountID,
 	}
 
 	if testKey.Key != "" {