Commit09c2cd4

committed

add test for agent calling RefreshToken()

1 parentf7a8557 commit09c2cd4Copy full SHA for 09c2cd4

File tree

+18

-5

lines changed

+18

-5

lines changed

Lines changed: 3 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -2952,9 +2952,12 @@ func TestAgent_Reconnect(t *testing.T) {`
`2952`	`2952`	`defercloser.Close()`
`2953`	`2953`
`2954`	`2954`	`call1:=testutil.RequireReceive(ctx,t,fCoordinator.CoordinateCalls)`
	`2955`	`+require.Equal(t,client.GetNumRefreshTokenCalls(),1)`
`2955`	`2956`	`close(call1.Resps)// hang up`
`2956`	`2957`	`// expect reconnect`
`2957`	`2958`	`testutil.RequireReceive(ctx,t,fCoordinator.CoordinateCalls)`
	`2959`	`+// Check that the agent refreshes the token when it reconnects.`
	`2960`	`+require.Equal(t,client.GetNumRefreshTokenCalls(),2)`
`2958`	`2961`	`closer.Close()`
`2959`	`2962`	`}`
`2960`	`2963`

Lines changed: 15 additions & 5 deletions

Original file line number	Diff line number	Diff line change
`@@ -88,10 +88,11 @@ type Client struct {`
`88`	`88`	`fakeAgentAPI*FakeAgentAPI`
`89`	`89`	`LastWorkspaceAgentfunc()`
`90`	`90`
`91`		`-mu sync.Mutex// Protects following.`
`92`		`-logs []agentsdk.Log`
`93`		`-derpMapUpdateschan*tailcfg.DERPMap`
`94`		`-derpMapOnce sync.Once`
	`91`	`+mu sync.Mutex// Protects following.`
	`92`	`+logs []agentsdk.Log`
	`93`	`+derpMapUpdateschan*tailcfg.DERPMap`
	`94`	`+derpMapOnce sync.Once`
	`95`	`+refreshTokenCallsint`
`95`	`96`	`}`
`96`	`97`
`97`	`98`	`func (*Client)AsRequestOption() codersdk.RequestOption {`
`@@ -104,10 +105,19 @@ func (*Client) GetSessionToken() string {`
`104`	`105`	`return"agenttest-token"`
`105`	`106`	`}`
`106`	`107`
`107`		`-func (*Client)RefreshToken(context.Context)error {`
	`108`	`+func (c*Client)RefreshToken(context.Context)error {`
	`109`	`+c.mu.Lock()`
	`110`	`+deferc.mu.Unlock()`
	`111`	`+c.refreshTokenCalls++`
`108`	`112`	`returnnil`
`109`	`113`	`}`
`110`	`114`
	`115`	`+func (c*Client)GetNumRefreshTokenCalls()int {`
	`116`	`+c.mu.Lock()`
	`117`	`+deferc.mu.Unlock()`
	`118`	`+returnc.refreshTokenCalls`
	`119`	`+}`
	`120`	`+`
`111`	`121`	`func (Client)RewriteDERPMap(tailcfg.DERPMap) {}`
`112`	`122`
`113`	`123`	`func (c*Client)Close() {`

Comments

(0)