add test for rate limit

Author: hfoxy

data.go

@@ -10,6 +10,7 @@ type IRData interface {
 	IsLoggedIn() bool
 	GetLoginExpiration() time.Time
 
+	RateLimit() *RateLimit
 	Car() DataCar
 	CarClass() DataCarClass
 	Constant() DataConstant
@@ -50,6 +51,10 @@ func (d *irdata) GetLoginExpiration() time.Time {
 	return d.expiration
 }
 
+func (d *irdata) RateLimit() *RateLimit {
+	return d.rateLimit
+}
+
 func (d *irdata) get(url string) (resp *http.Response, err error) {
 	if err = d.Reauthenticate(); err != nil {
 		return nil, err

ratelimit_test.go

@@ -0,0 +1,31 @@
+package irdata
+
+import (
+	"github.com/stretchr/testify/assert"
+	"testing"
+	"time"
+)
+
+func TestRateLimitUpdates(t *testing.T) {
+	api := DefaultClient.Season()
+	assert.NotNil(t, api)
+
+	_, err := api.RaceGuide()
+	assert.Nil(t, err)
+
+	rl := DefaultClient.RateLimit()
+	assert.NotNil(t, rl)
+
+	remaining := rl.Remaining()
+	total := rl.TotalLimit()
+	reset := rl.ResetAt()
+
+	assert.True(t, reset.After(time.Now()))
+
+	_, err = api.RaceGuide()
+	assert.Nil(t, err)
+
+	assert.NotEqual(t, remaining, total, "remaining rate limit should not be equal to total")
+	assert.NotEqual(t, rl.Remaining(), remaining, "remaining rate limit should not be equal to remaining on previous request")
+	assert.True(t, reset.After(time.Now()))
+}