]>
Commit | Line | Data |
---|---|---|
1 | package hotline | |
2 | ||
3 | import ( | |
4 | "bytes" | |
5 | "encoding/hex" | |
6 | "fmt" | |
7 | "github.com/stretchr/testify/assert" | |
8 | "os" | |
9 | "testing" | |
10 | ) | |
11 | ||
12 | func TestReadFlattenedFileObject(t *testing.T) { | |
13 | testData, _ := hex.DecodeString("46494c500001000000000000000000000000000000000002494e464f000000000000000000000052414d414354455854747478740000000000000100000000000000000000000000000000000000000000000000000000000000000007700000ba74247307700000ba74247300000008746573742e74787400004441544100000000000000000000000474657374") | |
14 | ||
15 | ffo := ReadFlattenedFileObject(testData) | |
16 | ||
17 | format := ffo.FlatFileHeader.Format[:] | |
18 | want := []byte("FILP") | |
19 | if !bytes.Equal(format, want) { | |
20 | t.Errorf("ReadFlattenedFileObject() = %q, want %q", format, want) | |
21 | } | |
22 | } | |
23 | ||
24 | func TestNewFlattenedFileObject(t *testing.T) { | |
25 | type args struct { | |
26 | fileRoot string | |
27 | filePath []byte | |
28 | fileName []byte | |
29 | } | |
30 | tests := []struct { | |
31 | name string | |
32 | args args | |
33 | want *flattenedFileObject | |
34 | wantErr assert.ErrorAssertionFunc | |
35 | }{ | |
36 | { | |
37 | name: "with valid file", | |
38 | args: args{ | |
39 | fileRoot: func() string { path, _ := os.Getwd(); return path + "/test/config/Files" }(), | |
40 | fileName: []byte("testfile.txt"), | |
41 | filePath: []byte{0, 0}, | |
42 | }, | |
43 | want: &flattenedFileObject{ | |
44 | FlatFileHeader: NewFlatFileHeader(), | |
45 | FlatFileInformationForkHeader: FlatFileInformationForkHeader{}, | |
46 | FlatFileInformationFork: NewFlatFileInformationFork("testfile.txt"), | |
47 | FlatFileDataForkHeader: FlatFileDataForkHeader{ | |
48 | ForkType: []byte("DATA"), | |
49 | CompressionType: []byte{0, 0, 0, 0}, | |
50 | RSVD: []byte{0, 0, 0, 0}, | |
51 | DataSize: []byte{0x00, 0x00, 0x00, 0x17}, | |
52 | }, | |
53 | FileData: nil, | |
54 | }, | |
55 | wantErr: assert.NoError, | |
56 | }, | |
57 | { | |
58 | name: "when file path is invalid", | |
59 | args: args{ | |
60 | fileRoot: func() string { path, _ := os.Getwd(); return path + "/test/config/Files" }(), | |
61 | fileName: []byte("nope.txt"), | |
62 | }, | |
63 | want: nil, | |
64 | wantErr: assert.Error, | |
65 | }, | |
66 | } | |
67 | for _, tt := range tests { | |
68 | t.Run(tt.name, func(t *testing.T) { | |
69 | got, err := NewFlattenedFileObject(tt.args.fileRoot, tt.args.filePath, tt.args.fileName) | |
70 | if !tt.wantErr(t, err, fmt.Sprintf("NewFlattenedFileObject(%v, %v, %v)", tt.args.fileRoot, tt.args.filePath, tt.args.fileName)) { | |
71 | return | |
72 | } | |
73 | assert.Equalf(t, tt.want, got, "NewFlattenedFileObject(%v, %v, %v)", tt.args.fileRoot, tt.args.filePath, tt.args.fileName) | |
74 | }) | |
75 | } | |
76 | } |