kite: refactor TestContext

Author: rjeczalik

client.go

@@ -212,7 +212,7 @@ func (k *Kite) NewClient(remoteURL string) *Client {
 		Concurrent:         true,
 		send:               make(chan *message),
 		interrupt:          make(chan error, 1),
-		ctx:                context.TODO(),
+		ctx:                context.Background(),
 		cancel:             func() {},
 	}
 

kite_test.go

@@ -86,15 +86,10 @@ func TestContext(t *testing.T) {
 	want := []int{1, 2, 3, 4}
 	timeout := time.After(2 * time.Second)
 
-collect:
-	for {
+	for len(got) != len(want) {
 		select {
 		case i := <-ch:
 			got = append(got, i)
-
-			if len(got) == len(want) {
-				break collect
-			}
 		case <-timeout:
 			t.Fatal("timed out collecting checkpoints")
 		}