Skip to content

Commit cdd721b

Browse files
Code review comments.
Signed-off-by: naveensrinivasan <[email protected]>
1 parent f70f744 commit cdd721b

File tree

1 file changed

+42
-26
lines changed

1 file changed

+42
-26
lines changed

clients/githubrepo/roundtripper/tokens/accessor_test.go

+42-26
Original file line numberDiff line numberDiff line change
@@ -42,35 +42,17 @@ func TestMakeTokenAccessor(t *testing.T) {
4242
useServer: true,
4343
},
4444
}
45+
t.Setenv("GITHUB_AUTH_TOKEN", "")
46+
t.Setenv("GITHUB_TOKEN", "")
4547
for _, tt := range tests {
4648
t.Run(tt.name, func(t *testing.T) {
4749
switch {
4850
case tt.useGitHubToken:
49-
token := "test"
50-
t.Setenv("GITHUB_TOKEN", token)
51-
got := MakeTokenAccessor()
52-
if got == nil {
53-
t.Errorf("MakeTokenAccessor() = nil, want not nil")
54-
}
55-
raccess, ok := got.(*roundRobinAccessor)
56-
if !ok {
57-
t.Errorf("MakeTokenAccessor() = %v, want *roundRobinAccessor", got)
58-
}
59-
if raccess.accessTokens[0] != token {
60-
t.Errorf("accessTokens[0] = %v, want %v", raccess.accessTokens[0], token)
61-
}
51+
t.Helper()
52+
testToken(t)
6253
case tt.useServer:
63-
t.Setenv("GITHUB_AUTH_SERVER", "localhost:8080")
64-
server := startTestServer()
65-
defer serverShutdown(server)
66-
myRPCService := &MyRPCService{}
67-
rpc.Register(myRPCService) //nolint:errcheck
68-
server.Handler = nil
69-
rpc.HandleHTTP()
70-
got := MakeTokenAccessor()
71-
if got == nil {
72-
t.Errorf("MakeTokenAccessor() = nil, want not nil")
73-
}
54+
t.Helper()
55+
testServer(t)
7456
default:
7557
got := MakeTokenAccessor()
7658
if got != nil {
@@ -81,15 +63,49 @@ func TestMakeTokenAccessor(t *testing.T) {
8163
}
8264
}
8365

66+
func testToken(t *testing.T) {
67+
t.Helper()
68+
token := "test"
69+
t.Setenv("GITHUB_AUTH_TOKEN", token)
70+
got := MakeTokenAccessor()
71+
if got == nil {
72+
t.Errorf("MakeTokenAccessor() = nil, want not nil")
73+
}
74+
raccess, ok := got.(*roundRobinAccessor)
75+
if !ok {
76+
t.Errorf("MakeTokenAccessor() = %v, want *roundRobinAccessor", got)
77+
}
78+
if raccess.accessTokens[0] != token {
79+
t.Errorf("accessTokens[0] = %v, want %v", raccess.accessTokens[0], token)
80+
}
81+
}
82+
83+
func testServer(t *testing.T) {
84+
t.Helper()
85+
t.Setenv("GITHUB_AUTH_SERVER", "localhost:8080")
86+
server := startTestServer()
87+
t.Cleanup(func() {
88+
serverShutdown(server)
89+
})
90+
myRPCService := &MyRPCService{}
91+
rpc.Register(myRPCService) //nolint:errcheck
92+
server.Handler = nil
93+
rpc.HandleHTTP()
94+
got := MakeTokenAccessor()
95+
if got == nil {
96+
t.Errorf("MakeTokenAccessor() = nil, want not nil")
97+
}
98+
}
99+
84100
type MyRPCService struct {
85101
// Define your RPC service methods here
86102
}
87103

88104
func startTestServer() *http.Server {
89105
// Create a new server
90106
server := &http.Server{ //nolint:gosec
91-
Addr: ":8080", // Use any available port
92-
Handler: nil, // Use the default handler
107+
Addr: ":8080",
108+
Handler: nil, // Use the default handler
93109
}
94110

95111
// Start the server in a separate goroutine

0 commit comments

Comments
 (0)