File tree 2 files changed +4
-0
lines changed
2 files changed +4
-0
lines changed Original file line number Diff line number Diff line change @@ -2,6 +2,8 @@ package testutils
2
2
3
3
import "os"
4
4
5
+ // MustExtractAPITokenFromEnv is a helper function that looks up PAPERSWITHCODE_API_TOKEN.
6
+ // It panics if not found.
5
7
func MustExtractAPITokenFromEnv () string {
6
8
apiToken , ok := os .LookupEnv ("PAPERSWITHCODE_API_TOKEN" )
7
9
Original file line number Diff line number Diff line change @@ -9,11 +9,13 @@ type transportWithAuthHeader struct {
9
9
10
10
var _ http.RoundTripper = transportWithAuthHeader {}
11
11
12
+ // RoundTrip attaches "Authorization: Token <token>" for every request.
12
13
func (t transportWithAuthHeader ) RoundTrip (request * http.Request ) (* http.Response , error ) {
13
14
request .Header .Set ("Authorization" , "Token " + t .token )
14
15
return t .r .RoundTrip (request )
15
16
}
16
17
18
+ // NewTransportWithAuthHeader returns middleware that attaches "Authorization: Token <token>" header to every request.
17
19
func NewTransportWithAuthHeader (token string ) http.RoundTripper {
18
20
return transportWithAuthHeader {
19
21
r : http .DefaultTransport ,
You can’t perform that action at this time.
0 commit comments