| // Copyright 2017 The LUCI Authors. |
| // |
| // Licensed under the Apache License, Version 2.0 (the "License"); |
| // you may not use this file except in compliance with the License. |
| // You may obtain a copy of the License at |
| // |
| // http://www.apache.org/licenses/LICENSE-2.0 |
| // |
| // Unless required by applicable law or agreed to in writing, software |
| // distributed under the License is distributed on an "AS IS" BASIS, |
| // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| // See the License for the specific language governing permissions and |
| // limitations under the License. |
| |
| package firebase |
| |
| import ( |
| "context" |
| "encoding/json" |
| "net/http" |
| "net/url" |
| "strings" |
| "testing" |
| "time" |
| |
| "golang.org/x/oauth2" |
| |
| "go.chromium.org/luci/common/clock" |
| "go.chromium.org/luci/common/clock/testclock" |
| |
| . "github.com/smartystreets/goconvey/convey" |
| ) |
| |
| func TestProtocol(t *testing.T) { |
| t.Parallel() |
| |
| ctx := context.Background() |
| ctx, _ = testclock.UseTime(ctx, testclock.TestRecentTimeUTC) |
| token := "tok1" |
| |
| Convey("With server", t, func(c C) { |
| s := Server{ |
| Source: oauth2.StaticTokenSource(&oauth2.Token{ |
| AccessToken: token, |
| Expiry: clock.Now(ctx).Add(30 * time.Minute), |
| }), |
| } |
| addr, err := s.Start(ctx) |
| So(err, ShouldBeNil) |
| defer s.Stop(ctx) |
| |
| call := func(refreshTok string) (*http.Response, error) { |
| form := url.Values{} |
| form.Add("grant_type", "refresh_token") |
| form.Add("refresh_token", refreshTok) |
| form.Add("client_id", "fake_client_id") |
| form.Add("client_secret", "fake_client_secret") |
| return http.Post(addr+"/oauth2/v3/token", "application/x-www-form-urlencoded", strings.NewReader(form.Encode())) |
| } |
| |
| Convey("Happy path", func() { |
| resp, err := call(token) |
| So(err, ShouldBeNil) |
| defer resp.Body.Close() |
| tok := map[string]interface{}{} |
| So(resp.StatusCode, ShouldEqual, 200) |
| So(resp.Header.Get("Content-Type"), ShouldEqual, "application/json") |
| So(json.NewDecoder(resp.Body).Decode(&tok), ShouldBeNil) |
| So(tok, ShouldResemble, map[string]interface{}{ |
| "access_token": "tok1", |
| "expires_in": 1800.0, |
| "token_type": "Bearer", |
| }) |
| }) |
| }) |
| } |