From e5534a1d21a3263c3bd0a9f6774dfcc161f9c773 Mon Sep 17 00:00:00 2001
From: Dawid Rusnak <dawid@drcode.pl>
Date: Fri, 18 Oct 2024 17:27:18 +0200
Subject: [PATCH] chore: pass context to ProContext factory

---
 cmd/api-server/main.go | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/cmd/api-server/main.go b/cmd/api-server/main.go
index 77d11b8a36..881b3bce86 100644
--- a/cmd/api-server/main.go
+++ b/cmd/api-server/main.go
@@ -203,7 +203,7 @@ func main() {
 		grpcClient = cloud.NewTestKubeCloudAPIClient(grpcConn)
 	}
 
-	proContext := newProContext(cfg, grpcClient)
+	proContext := newProContext(ctx, cfg, grpcClient)
 
 	// Check Pro/Enterprise subscription
 	subscriptionChecker, err := checktcl.NewSubscriptionChecker(ctx, proContext, grpcClient, grpcConn)
@@ -710,7 +710,7 @@ func newSlackLoader(cfg *config.Config, envs map[string]string) (*slack.SlackLoa
 		testkube.AllEventTypes, envs), nil
 }
 
-func newProContext(cfg *config.Config, grpcClient cloud.TestKubeCloudAPIClient) config.ProContext {
+func newProContext(ctx context.Context, cfg *config.Config, grpcClient cloud.TestKubeCloudAPIClient) config.ProContext {
 	proContext := config.ProContext{
 		APIKey:                           cfg.TestkubeProAPIKey,
 		URL:                              cfg.TestkubeProURL,
@@ -730,7 +730,7 @@ func newProContext(cfg *config.Config, grpcClient cloud.TestKubeCloudAPIClient)
 		return proContext
 	}
 
-	ctx, cancel := context.WithTimeout(context.Background(), time.Second*3)
+	ctx, cancel := context.WithTimeout(ctx, time.Second*3)
 	md := metadata.Pairs("api-key", cfg.TestkubeProAPIKey)
 	ctx = metadata.NewOutgoingContext(ctx, md)
 	defer cancel()