harness-drone/handler/web/varz_test.go
2019-02-19 15:56:41 -08:00

66 lines
1.3 KiB
Go

// Copyright 2019 Drone.IO Inc. All rights reserved.
// Use of this source code is governed by the Drone Non-Commercial License
// that can be found in the LICENSE file.
package web
import (
"encoding/json"
"net/http/httptest"
"net/url"
"testing"
"github.com/drone/drone/core"
"github.com/drone/go-scm/scm"
"github.com/google/go-cmp/cmp"
)
func TestHandleVarz(t *testing.T) {
w := httptest.NewRecorder()
r := httptest.NewRequest("GET", "/", nil)
client := new(scm.Client)
client.BaseURL, _ = url.Parse("https://github.com")
client.SetRate(scm.Rate{
Limit: 5000,
Remaining: 875,
Reset: 1523640878,
})
license := &core.License{
Kind: core.LicenseStandard,
Repos: 50,
Users: 100,
}
HandleVarz(client, license).ServeHTTP(w, r)
if got, want := w.Code, 200; want != got {
t.Errorf("Want response code %d, got %d", want, got)
}
got, want := &varz{}, mockVarz
json.NewDecoder(w.Body).Decode(got)
if diff := cmp.Diff(got, want); diff != "" {
t.Errorf(diff)
}
}
var mockVarz = &varz{
SCM: &scmInfo{
URL: "https://github.com",
Rate: &rateInfo{
Limit: 5000,
Remaining: 875,
Reset: 1523640878,
},
},
License: &licenseInfo{
Kind: "standard",
Seats: 100,
SeatsUsed: 0,
SeatsAvail: 0,
Repos: 50,
ReposUsed: 0,
ReposAvail: 0,
},
}