@@ -42,35 +42,17 @@ func TestMakeTokenAccessor(t *testing.T) {
42
42
useServer : true ,
43
43
},
44
44
}
45
+ t .Setenv ("GITHUB_AUTH_TOKEN" , "" )
46
+ t .Setenv ("GITHUB_TOKEN" , "" )
45
47
for _ , tt := range tests {
46
48
t .Run (tt .name , func (t * testing.T ) {
47
49
switch {
48
50
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 )
62
53
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 )
74
56
default :
75
57
got := MakeTokenAccessor ()
76
58
if got != nil {
@@ -81,15 +63,49 @@ func TestMakeTokenAccessor(t *testing.T) {
81
63
}
82
64
}
83
65
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
+
84
100
type MyRPCService struct {
85
101
// Define your RPC service methods here
86
102
}
87
103
88
104
func startTestServer () * http.Server {
89
105
// Create a new server
90
106
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
93
109
}
94
110
95
111
// Start the server in a separate goroutine
0 commit comments