Merge pull request #169 from andygrunwald/set-unit-tests

Set: Add unit test for New with entries

Author: tylertreat-wf

set/dict_test.go

@@ -76,6 +76,22 @@ func TestExists(t *testing.T) {
 	}
 }
 
+func TestExists_WithNewItems(t *testing.T) {
+	set := New(`test`, `test1`)
+
+	if !set.Exists(`test`) {
+		t.Errorf(`Correct existence not determined`)
+	}
+
+	if !set.Exists(`test1`) {
+		t.Errorf(`Correct existence not determined`)
+	}
+
+	if set.Exists(`test2`) {
+		t.Errorf(`Correct nonexistence not determined.`)
+	}
+}
+
 func TestLen(t *testing.T) {
 	set := New()
 	set.Add(`test`)
@@ -102,6 +118,19 @@ func TestFlattenCaches(t *testing.T) {
 	}
 }
 
+func TestFlattenCaches_CacheReturn(t *testing.T) {
+	set := New()
+	item := `test`
+	set.Add(item)
+
+	flatten1 := set.Flatten()
+	flatten2 := set.Flatten()
+
+	if !reflect.DeepEqual(flatten1, flatten2) {
+		t.Errorf(`Flatten cache is not the same as original result. Got %+v, expected %+v`, flatten2, flatten1)
+	}
+}
+
 func TestAddClearsCache(t *testing.T) {
 	set := New()
 	item := `test`