Added tested addJunk func

jontonsoup · Mar 20, 2013 · 097410a · 097410a
1 parent c4c5fc7
commit 097410a
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 0 deletions.
diff --git a/tfh/crypto.go b/tfh/crypto.go
@@ -268,3 +268,12 @@ func trimDecryptedFile(file []byte, numToTrim int) (trimFile []byte) {
 	trimFile = file[:len(file)-numToTrim]
 	return
 }
+
+func addJunk(bs ([][]byte), numBytes int) (newBs [][]byte) {
+	newBs = bs[:]
+	for i := 0; i < numBytes; i++ {
+		junk := makeRandKey(numBytes)
+		newBs = append(newBs, junk)
+	}
+	return
+}
diff --git a/tfh/crypto_test.go b/tfh/crypto_test.go
@@ -102,3 +102,22 @@ func TestPadFile(t *testing.T) {
 		}
 	}
 }
+
+func TestaddJunk(t *testing.T) {
+	bs := make([][]byte, 2)
+	bs[0] = []byte{1, 2}
+	bs[1] = []byte{3}
+	newBs := addJunk(bs, 2)
+	if len(newBs) != 4 {
+		t.Fail()
+		return
+	}
+	for i, b := range bs {
+		for j, b2 := range b {
+			if uint(b2) != uint(newBs[i][j]) {
+				t.Fail()
+				return
+			}
+		}
+	}
+}