adding in a test for builder.Close()

Tonkpils · Aug 6, 2015 · 73b63e1 · 73b63e1
1 parent 06561f8
commit 73b63e1
Showing 1 changed file with 13 additions and 2 deletions.
diff --git a/builder_test.go b/builder_test.go
@@ -1,13 +1,24 @@
 package main
 
 import (
-	"github.com/stretchr/testify/assert"
-
 	"testing"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
 )
 
 func TestNewBuilder(t *testing.T) {
 	b, err := NewBuilder(config{})
 	assert.NoError(t, err)
 	assert.NotNil(t, b)
 }
+
+func TestClose(t *testing.T) {
+	b, err := NewBuilder(config{})
+	require.NoError(t, err)
+
+	b.Close()
+
+	_, ok := <-b.done
+	assert.False(t, ok, "channel 'done' was not closed")
+}