git.r.bdr.sh - rbdr/mobius/blob - hotline/flattened_file_object_test.go

   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 }