File tree Expand file tree Collapse file tree 2 files changed +4
-0
lines changed Expand file tree Collapse file tree 2 files changed +4
-0
lines changed Original file line number Diff line number Diff line change @@ -2,6 +2,8 @@ package testutils
22
33import "os"
44
5+ // MustExtractAPITokenFromEnv is a helper function that looks up PAPERSWITHCODE_API_TOKEN.
6+ // It panics if not found.
57func MustExtractAPITokenFromEnv () string {
68 apiToken , ok := os .LookupEnv ("PAPERSWITHCODE_API_TOKEN" )
79
Original file line number Diff line number Diff line change @@ -9,11 +9,13 @@ type transportWithAuthHeader struct {
99
1010var _ http.RoundTripper = transportWithAuthHeader {}
1111
12+ // RoundTrip attaches "Authorization: Token <token>" for every request.
1213func (t transportWithAuthHeader ) RoundTrip (request * http.Request ) (* http.Response , error ) {
1314 request .Header .Set ("Authorization" , "Token " + t .token )
1415 return t .r .RoundTrip (request )
1516}
1617
18+ // NewTransportWithAuthHeader returns middleware that attaches "Authorization: Token <token>" header to every request.
1719func NewTransportWithAuthHeader (token string ) http.RoundTripper {
1820 return transportWithAuthHeader {
1921 r : http .DefaultTransport ,
You can’t perform that action at this time.
0 commit comments