/usr/share/gocode/src/github.com/JamesClonk/vultr/lib/client_test.go is in golang-github-jamesclonk-vultr-dev 1.10-1.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 | package lib
import (
"fmt"
"net/http"
"net/http/httptest"
"sync"
"testing"
"time"
"github.com/juju/ratelimit"
"github.com/stretchr/testify/assert"
)
func getTestServerAndClient(code int, body string) (*httptest.Server, *Client) {
server := getTestServer(code, body)
return server, getTestClient(server.URL)
}
func getTestServer(code int, body string) *httptest.Server {
return httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
w.WriteHeader(code)
w.Header().Set("Content-Type", "application/json")
fmt.Fprint(w, body)
}))
}
func getTestServerThrottled(body string) *httptest.Server {
var rateLimiter *ratelimit.Bucket
// Rate limit: 2 req/s, capacity 2
rateLimiter = ratelimit.NewBucket(500*time.Millisecond, 2)
return httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
code := 200
if tokens := rateLimiter.TakeAvailable(1); tokens == 0 {
code = 503
}
w.WriteHeader(code)
w.Header().Set("Content-Type", "application/json")
fmt.Fprint(w, body)
}))
}
func getTestClient(endpoint string) *Client {
options := Options{
Endpoint: endpoint,
RateLimitation: 1 * time.Millisecond,
}
return NewClient("test-key", &options)
}
func Test_Client_Options(t *testing.T) {
options := Options{
HTTPClient: http.DefaultClient,
UserAgent: "test-agent",
Endpoint: "http://test",
}
client := NewClient("test-key", &options)
if assert.NotNil(t, client) {
assert.Equal(t, "test-key", client.APIKey)
assert.Equal(t, "http://test", client.Endpoint.String())
assert.Equal(t, "test-agent", client.UserAgent)
}
}
func Test_Client_NewClient(t *testing.T) {
client := NewClient("test-key-new", nil)
if assert.NotNil(t, client) {
assert.Equal(t, "test-key-new", client.APIKey)
assert.Equal(t, "https://api.vultr.com/", client.Endpoint.String())
assert.Equal(t, "vultr-go/"+Version, client.UserAgent)
}
}
// Test that API queries are throttled
func Test_Client_Throttling(t *testing.T) {
const ERROR = 100 * time.Millisecond
const EXPECTED_DURATION = 400 * time.Millisecond
server, _ := getTestServerAndClient(http.StatusOK, `{
"balance":-15.97,"pending_charges":"2.34",
"last_payment_date":"2015-01-29 05:06:27","last_payment_amount":"-5.00"}`)
defer server.Close()
options := Options{
Endpoint: server.URL,
RateLimitation: 100 * time.Millisecond,
}
client := NewClient("test-key", &options)
// The first query should not be throttled
info, _ := client.GetAccountInfo()
// The next four queries should be throttled and take 400 milliseconds
before := time.Now()
info, _ = client.GetAccountInfo()
info, _ = client.GetAccountInfo()
info, _ = client.GetAccountInfo()
info, _ = client.GetAccountInfo()
after := time.Now()
lower := EXPECTED_DURATION - ERROR
upper := EXPECTED_DURATION + ERROR
assert.NotNil(t, info)
if diff := after.Sub(before); diff < lower || diff > upper {
t.Errorf("Waited %s seconds, though really should have waited between %s and %s", diff.String(), lower.String(), upper.String())
}
}
// Test that retry logic is working
func Test_Client_Retry(t *testing.T) {
server := getTestServerThrottled(`{
"balance":-15.97,"pending_charges":"2.34",
"last_payment_date":"2015-01-29 05:06:27","last_payment_amount":"-5.00"}`)
defer server.Close()
options := Options{
Endpoint: server.URL,
MaxRetries: 5,
}
client := NewClient("test-key", &options)
var wg sync.WaitGroup
wg.Add(10)
for i := 0; i < 10; i++ {
go func() {
defer wg.Done()
_, err := client.GetAccountInfo()
assert.NoError(t, err)
}()
}
wg.Wait()
}
|