Add BlockNoPlainOff() and BlockNoCipherOff() + test

Also, fix key, it is now []byte, not [16]byte

1 files changed, 24 insertions, 2 deletions

diff --git a/cryptfs/content_test.go b/cryptfs/content_test.go
index b53eeeb..8fd33c6 100644
--- a/cryptfs/content_test.go
+++ b/cryptfs/content_test.go
@@ -19,7 +19,7 @@ func TestSplitRange(t *testing.T) {
 		testRange{65444, 54},
 		testRange{6654, 8945})
 
-	var key [16]byte
+	key := make([]byte, 16)
 	f := NewCryptFS(key, true)
 
 	for _, r := range(ranges) {
@@ -42,7 +42,7 @@ func TestCiphertextRange(t *testing.T) {
 		testRange{65444, 54},
 		testRange{6654, 8945})
 
-	var key [16]byte
+	key := make([]byte, 16)
 	f := NewCryptFS(key, true)
 
 	for _, r := range(ranges) {
@@ -58,3 +58,25 @@ func TestCiphertextRange(t *testing.T) {
 		}
 	}
 }
+
+func TestBlockNo(t *testing.T) {
+	key := make([]byte, 16)
+	f := NewCryptFS(key, true)
+
+	b := f.BlockNoCipherOff(788)
+	if b != 0 {
+		t.Errorf("actual: %d", b)
+	}
+	b = f.BlockNoCipherOff(f.CipherBS())
+	if b != 1 {
+		t.Errorf("actual: %d", b)
+	}
+	b = f.BlockNoPlainOff(788)
+	if b != 0 {
+		t.Errorf("actual: %d", b)
+	}
+	b = f.BlockNoPlainOff(f.PlainBS())
+	if b != 1 {
+		t.Errorf("actual: %d", b)
+	}
+}