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