Added unit test for bandwidth

oxtoacart · oxtoacart · commit 005174f7a489 · 2016-06-02T07:58:09.000-05:00
diff --git a/src/github.com/getlantern/bandwidth/bandwidth.go b/src/github.com/getlantern/bandwidth/bandwidth.go
@@ -17,7 +17,7 @@ var (
 	mutex sync.RWMutex
 
 	// Updates is a channel on which one can receive updates to the Quota
-	Updates = make(chan *Quota)
+	Updates = make(chan *Quota, 100)
 )
 
 // Quota encapsulates information about the user's bandwidth quota.
diff --git a/src/github.com/getlantern/bandwidth/bandwidth_test.go b/src/github.com/getlantern/bandwidth/bandwidth_test.go
@@ -0,0 +1,46 @@
+package bandwidth
+
+import (
+	"fmt"
+	"net/http"
+	"testing"
+	"time"
+
+	"github.com/getlantern/eventual"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestRoundTrip(t *testing.T) {
+	fromChannel := eventual.NewValue()
+	go func() {
+		for q := range Updates {
+			fromChannel.Set(q)
+		}
+	}()
+
+	Track(build(5, 500, 2))
+	Track(build(6, 500, 3))
+	Track(build(4, 500, 1))
+
+	q := GetQuota()
+	time.Sleep(50 * time.Millisecond)
+	_q2, ok := fromChannel.Get(1 * time.Second)
+	if assert.True(t, ok, "Should have gotten quota from channel") {
+		q2 := _q2.(*Quota)
+		assert.EqualValues(t, 500, q.MiBAllowed)
+		assert.EqualValues(t, 500, q2.MiBAllowed)
+		assert.EqualValues(t, 6, q.MiBUsed)
+		assert.EqualValues(t, 6, q2.MiBUsed)
+		assert.EqualValues(t, time.Unix(0, 3), q.AsOf)
+		assert.EqualValues(t, time.Unix(0, 3), q2.AsOf)
+	}
+
+}
+
+func build(used uint64, allowed uint64, asof int64) *http.Response {
+	resp := &http.Response{
+		Header: make(http.Header),
+	}
+	resp.Header.Set("XBQ", fmt.Sprintf("%d/%d/%d", used, allowed, asof))
+	return resp
+}
diff --git a/testpackages.txt b/testpackages.txt
@@ -1,5 +1,6 @@
 github.com/getlantern/appdir
 github.com/getlantern/balancer
+github.com/getlantern/bandwidth
 github.com/getlantern/buuid
 github.com/getlantern/bytecounting
 github.com/getlantern/byteexec

Original file line number	Diff line number	Diff line change
`@@ -17,7 +17,7 @@ var (`
`17`	`17`	`mutex sync.RWMutex`
`18`	`18`
`19`	`19`	`// Updates is a channel on which one can receive updates to the Quota`
`20`		`- Updates = make(chan *Quota)`
	`20`	`+ Updates = make(chan *Quota, 100)`
`21`	`21`	`)`
`22`	`22`
`23`	`23`	`// Quota encapsulates information about the user's bandwidth quota.`