add test for shuffling

jraedisch · Dec 31, 2017 · ff9cf9d · ff9cf9d
1 parent 667b961
commit ff9cf9d
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/main_test.go b/main_test.go
@@ -2,6 +2,7 @@ package main
 
 import (
 	"reflect"
+	"sort"
 	"testing"
 )
 
@@ -13,3 +14,12 @@ func TestDistributeVictims(t *testing.T) {
 		t.Errorf("expected\n%v\ngot\n%v", expected, actual)
 	}
 }
+
+func TestShuffleNames(t *testing.T) {
+	sorted := []string{"A", "B", "C", "D"}
+	shuffled := shuffleNames(sorted)
+	sort.Strings(shuffled)
+	if !reflect.DeepEqual(sorted, shuffled) {
+		t.Errorf("shuffled does not contain the right names: %v", shuffled)
+	}
+}