forked from mirrors/tar-split
Vincent Batts
032efafc29
When the entry name is not UTF-8, for example ISO-8859-1, then store the raw bytes. To accommodate this, we will have getters and setters for the entry's name now. Since this most heavily affects the json marshalling, we'll double check the sanity of the name before storing it in the JSONPacker.
95 lines
1.8 KiB
Go
95 lines
1.8 KiB
Go
package storage
|
|
|
|
import (
|
|
"encoding/json"
|
|
"sort"
|
|
"testing"
|
|
)
|
|
|
|
func TestEntries(t *testing.T) {
|
|
e := Entries{
|
|
Entry{
|
|
Type: SegmentType,
|
|
Payload: []byte("y'all"),
|
|
Position: 1,
|
|
},
|
|
Entry{
|
|
Type: SegmentType,
|
|
Payload: []byte("doin"),
|
|
Position: 3,
|
|
},
|
|
Entry{
|
|
Type: FileType,
|
|
Name: "./hurr.txt",
|
|
Payload: []byte("deadbeef"),
|
|
Position: 2,
|
|
},
|
|
Entry{
|
|
Type: SegmentType,
|
|
Payload: []byte("how"),
|
|
Position: 0,
|
|
},
|
|
}
|
|
sort.Sort(e)
|
|
if e[0].Position != 0 {
|
|
t.Errorf("expected Position 0, but got %d", e[0].Position)
|
|
}
|
|
}
|
|
|
|
func TestFile(t *testing.T) {
|
|
f := Entry{
|
|
Type: FileType,
|
|
Size: 100,
|
|
Position: 2,
|
|
}
|
|
f.SetName("./hello.txt")
|
|
|
|
buf, err := json.Marshal(f)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
f1 := Entry{}
|
|
if err = json.Unmarshal(buf, &f1); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if f.GetName() != f1.GetName() {
|
|
t.Errorf("expected Name %q, got %q", f.GetName(), f1.GetName())
|
|
}
|
|
if f.Size != f1.Size {
|
|
t.Errorf("expected Size %q, got %q", f.Size, f1.Size)
|
|
}
|
|
if f.Position != f1.Position {
|
|
t.Errorf("expected Position %q, got %q", f.Position, f1.Position)
|
|
}
|
|
}
|
|
|
|
func TestFileRaw(t *testing.T) {
|
|
f := Entry{
|
|
Type: FileType,
|
|
Size: 100,
|
|
Position: 2,
|
|
}
|
|
f.SetNameBytes([]byte{0x2E, 0x2F, 0x68, 0x65, 0x6C, 0x6C, 0x6F, 0xE4, 0x2E, 0x74, 0x78, 0x74})
|
|
|
|
buf, err := json.Marshal(f)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
f1 := Entry{}
|
|
if err = json.Unmarshal(buf, &f1); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if f.GetName() != f1.GetName() {
|
|
t.Errorf("expected Name %q, got %q", f.GetName(), f1.GetName())
|
|
}
|
|
if f.Size != f1.Size {
|
|
t.Errorf("expected Size %q, got %q", f.Size, f1.Size)
|
|
}
|
|
if f.Position != f1.Position {
|
|
t.Errorf("expected Position %q, got %q", f.Position, f1.Position)
|
|
}
|
|
}
|