@@ -78,7 +78,7 @@ func TestStartCPUProfiler(t *testing.T) {
7878 for _ , test := range SuccessResponseTests {
7979 t .Run (test .name , func (t * testing.T ) {
8080 mockClient := Client {Requester : NewMockClient (& api.EmptyReply {}, test .expectedErr )}
81- err := mockClient .StartCPUProfiler (context . Background ())
81+ err := mockClient .StartCPUProfiler (t . Context ())
8282 require .ErrorIs (t , err , test .expectedErr )
8383 })
8484 }
@@ -88,7 +88,7 @@ func TestStopCPUProfiler(t *testing.T) {
8888 for _ , test := range SuccessResponseTests {
8989 t .Run (test .name , func (t * testing.T ) {
9090 mockClient := Client {Requester : NewMockClient (& api.EmptyReply {}, test .expectedErr )}
91- err := mockClient .StopCPUProfiler (context . Background ())
91+ err := mockClient .StopCPUProfiler (t . Context ())
9292 require .ErrorIs (t , err , test .expectedErr )
9393 })
9494 }
@@ -98,7 +98,7 @@ func TestMemoryProfile(t *testing.T) {
9898 for _ , test := range SuccessResponseTests {
9999 t .Run (test .name , func (t * testing.T ) {
100100 mockClient := Client {Requester : NewMockClient (& api.EmptyReply {}, test .expectedErr )}
101- err := mockClient .MemoryProfile (context . Background ())
101+ err := mockClient .MemoryProfile (t . Context ())
102102 require .ErrorIs (t , err , test .expectedErr )
103103 })
104104 }
@@ -108,7 +108,7 @@ func TestLockProfile(t *testing.T) {
108108 for _ , test := range SuccessResponseTests {
109109 t .Run (test .name , func (t * testing.T ) {
110110 mockClient := Client {Requester : NewMockClient (& api.EmptyReply {}, test .expectedErr )}
111- err := mockClient .LockProfile (context . Background ())
111+ err := mockClient .LockProfile (t . Context ())
112112 require .ErrorIs (t , err , test .expectedErr )
113113 })
114114 }
@@ -118,7 +118,7 @@ func TestAlias(t *testing.T) {
118118 for _ , test := range SuccessResponseTests {
119119 t .Run (test .name , func (t * testing.T ) {
120120 mockClient := Client {Requester : NewMockClient (& api.EmptyReply {}, test .expectedErr )}
121- err := mockClient .Alias (context . Background (), "alias" , "alias2" )
121+ err := mockClient .Alias (t . Context (), "alias" , "alias2" )
122122 require .ErrorIs (t , err , test .expectedErr )
123123 })
124124 }
@@ -128,7 +128,7 @@ func TestAliasChain(t *testing.T) {
128128 for _ , test := range SuccessResponseTests {
129129 t .Run (test .name , func (t * testing.T ) {
130130 mockClient := Client {Requester : NewMockClient (& api.EmptyReply {}, test .expectedErr )}
131- err := mockClient .AliasChain (context . Background (), "chain" , "chain-alias" )
131+ err := mockClient .AliasChain (t . Context (), "chain" , "chain-alias" )
132132 require .ErrorIs (t , err , test .expectedErr )
133133 })
134134 }
@@ -143,14 +143,14 @@ func TestGetChainAliases(t *testing.T) {
143143 Aliases : expectedReply ,
144144 }, nil )}
145145
146- reply , err := mockClient .GetChainAliases (context . Background (), "chain" )
146+ reply , err := mockClient .GetChainAliases (t . Context (), "chain" )
147147 require .NoError (err )
148148 require .Equal (expectedReply , reply )
149149 })
150150
151151 t .Run ("failure" , func (t * testing.T ) {
152152 mockClient := Client {Requester : NewMockClient (& GetChainAliasesReply {}, errTest )}
153- _ , err := mockClient .GetChainAliases (context . Background (), "chain" )
153+ _ , err := mockClient .GetChainAliases (t . Context (), "chain" )
154154 require .ErrorIs (t , err , errTest )
155155 })
156156}
@@ -159,7 +159,7 @@ func TestStacktrace(t *testing.T) {
159159 for _ , test := range SuccessResponseTests {
160160 t .Run (test .name , func (t * testing.T ) {
161161 mockClient := Client {Requester : NewMockClient (& api.EmptyReply {}, test .expectedErr )}
162- err := mockClient .Stacktrace (context . Background ())
162+ err := mockClient .Stacktrace (t . Context ())
163163 require .ErrorIs (t , err , test .expectedErr )
164164 })
165165 }
@@ -182,15 +182,15 @@ func TestReloadInstalledVMs(t *testing.T) {
182182 FailedVMs : expectedFailedVMs ,
183183 }, nil )}
184184
185- loadedVMs , failedVMs , err := mockClient .LoadVMs (context . Background ())
185+ loadedVMs , failedVMs , err := mockClient .LoadVMs (t . Context ())
186186 require .NoError (err )
187187 require .Equal (expectedNewVMs , loadedVMs )
188188 require .Equal (expectedFailedVMs , failedVMs )
189189 })
190190
191191 t .Run ("failure" , func (t * testing.T ) {
192192 mockClient := Client {Requester : NewMockClient (& LoadVMsReply {}, errTest )}
193- _ , _ , err := mockClient .LoadVMs (context . Background ())
193+ _ , _ , err := mockClient .LoadVMs (t . Context ())
194194 require .ErrorIs (t , err , errTest )
195195 })
196196}
@@ -253,7 +253,7 @@ func TestSetLoggerLevel(t *testing.T) {
253253 ),
254254 }
255255 res , err := c .SetLoggerLevel (
256- context . Background (),
256+ t . Context (),
257257 "" ,
258258 tt .logLevel ,
259259 tt .displayLevel ,
@@ -306,7 +306,7 @@ func TestGetLoggerLevel(t *testing.T) {
306306 ),
307307 }
308308 res , err := c .GetLoggerLevel (
309- context . Background (),
309+ t . Context (),
310310 tt .loggerName ,
311311 )
312312 require .ErrorIs (err , tt .clientErr )
@@ -347,7 +347,7 @@ func TestGetConfig(t *testing.T) {
347347 c := Client {
348348 Requester : NewMockClient (tt .expectedResponse , tt .serviceErr ),
349349 }
350- res , err := c .GetConfig (context . Background ())
350+ res , err := c .GetConfig (t . Context ())
351351 require .ErrorIs (err , tt .clientErr )
352352 if tt .clientErr != nil {
353353 return
0 commit comments