1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
|
package filter
import (
"bytes"
"code.laria.me/petrific/backup"
"code.laria.me/petrific/storage/memory"
"testing"
)
type XorFilter byte
func (xf XorFilter) Transform(b []byte) ([]byte, error) {
for i := range b {
b[i] ^= byte(xf)
}
return b, nil
}
func genTestData(size int) []byte {
data := make([]byte, size)
for i := range data {
data[i] = byte(i & 0xff)
}
return data
}
func TestXorReverse(t *testing.T) {
orig := genTestData(1 << 20)
filter := XorFilter(42)
filtered := make([]byte, len(orig))
copy(filtered, orig)
filtered, _ = filter.Transform(filtered)
filtered, _ = filter.Transform(filtered)
if !bytes.Equal(filtered, orig) {
t.Errorf("orig != filtered:\n{%x}\n{%x}", orig, filtered)
}
}
func testFilter(t *testing.T, encode, decode Filter) {
base := memory.NewMemoryStorage()
storage := FilterStorage{
Base: base,
Encode: encode,
Decode: decode,
}
size := backup.BlobChunkSize*2 + 10
data := genTestData(size)
id, err := backup.WriteFile(storage, bytes.NewReader(data))
if err != nil {
t.Fatalf("Unexpeced error from WriteFile: %s", err)
}
buf := new(bytes.Buffer)
if err := backup.RestoreFile(storage, id, buf); err != nil {
t.Fatalf("Unexpeced error from RestoreFile: %s", err)
}
if !bytes.Equal(data, buf.Bytes()) {
t.Errorf("data != buf.Bytes()")
}
}
func TestFilterStorage(t *testing.T) {
testFilter(t, XorFilter(42), XorFilter(42))
}
func TestPipeFilter(t *testing.T) {
testFilter(t, PipeFilter([]string{"cat"}), PipeFilter([]string{"cat"}))
}
|