@@ -1234,10 +1234,8 @@ endpoint_params:
1234
1234
1235
1235
func TestOAuth2UserAgent (t * testing.T ) {
1236
1236
ts := httptest .NewServer (http .HandlerFunc (func (w http.ResponseWriter , r * http.Request ) {
1237
- if r .URL .Path != "/" {
1238
- if r .Header .Get ("User-Agent" ) != "myuseragent" {
1239
- t .Fatalf ("Expected User-Agent header in oauth request to be 'myuseragent', got '%s'" , r .Header .Get ("User-Agent" ))
1240
- }
1237
+ if r .Header .Get ("User-Agent" ) != "myuseragent" {
1238
+ t .Fatalf ("Expected User-Agent header in oauth request to be 'myuseragent', got '%s'" , r .Header .Get ("User-Agent" ))
1241
1239
}
1242
1240
1243
1241
res , _ := json .Marshal (oauth2TestServerResponse {
@@ -1249,18 +1247,19 @@ func TestOAuth2UserAgent(t *testing.T) {
1249
1247
}))
1250
1248
defer ts .Close ()
1251
1249
1252
- config := & OAuth2 {
1250
+ config := DefaultHTTPClientConfig
1251
+ config .OAuth2 = & OAuth2 {
1253
1252
ClientID : "1" ,
1254
1253
ClientSecret : "2" ,
1255
1254
Scopes : []string {"A" , "B" },
1256
1255
EndpointParams : map [string ]string {"hi" : "hello" },
1257
1256
TokenURL : fmt .Sprintf ("%s/token" , ts .URL ),
1258
1257
}
1259
1258
1260
- opts := defaultHTTPClientOptions
1261
- WithUserAgent ( "myuseragent" )( & opts )
1262
-
1263
- rt := NewOAuth2RoundTripper ( config , http . DefaultTransport , & opts )
1259
+ rt , err := NewRoundTripperFromConfig ( config , "test_oauth2" , WithUserAgent ( "myuseragent" ))
1260
+ if err != nil {
1261
+ t . Fatal ( err )
1262
+ }
1264
1263
1265
1264
client := http.Client {
1266
1265
Transport : rt ,
0 commit comments