X-Git-Url: https://git.r.bdr.sh/rbdr/mobius/blobdiff_plain/5c34f8752938764668d829fb284626b9c62c5475..a0241c250de01f32eacba2afee3f7d5cea30b8e5:/hotline/flattened_file_object_test.go?ds=sidebyside

diff --git a/hotline/flattened_file_object_test.go b/hotline/flattened_file_object_test.go
index 63e7175..df05c09 100644
--- a/hotline/flattened_file_object_test.go
+++ b/hotline/flattened_file_object_test.go
@@ -3,6 +3,9 @@ package hotline
 import (
 	"bytes"
 	"encoding/hex"
+	"fmt"
+	"github.com/stretchr/testify/assert"
+	"os"
 	"testing"
 )
 
@@ -11,26 +14,63 @@ func TestReadFlattenedFileObject(t *testing.T) {
 
 	ffo := ReadFlattenedFileObject(testData)
 
-	format := ffo.FlatFileHeader.Format
+	format := ffo.FlatFileHeader.Format[:]
 	want := []byte("FILP")
 	if !bytes.Equal(format, want) {
 		t.Errorf("ReadFlattenedFileObject() = %q, want %q", format, want)
 	}
 }
 
-//
-//func TestNewFlattenedFileObject(t *testing.T) {
-//	ffo := NewFlattenedFileObject("test/config/files", "testfile.txt")
-//
-//	dataSize := ffo.FlatFileDataForkHeader.DataSize
-//	want := []byte{0, 0, 0, 0x17}
-//	if bytes.Compare(dataSize, want) != 0 {
-//		t.Errorf("%q, want %q", dataSize, want)
-//	}
-//
-//	comment := ffo.FlatFileInformationFork.Comment
-//	want = []byte("Test Comment")
-//	if bytes.Compare(ffo.FlatFileInformationFork.Comment, want) != 0 {
-//		t.Errorf("%q, want %q", comment, want)
-//	}
-//}
+func TestNewFlattenedFileObject(t *testing.T) {
+	type args struct {
+		fileRoot string
+		filePath []byte
+		fileName []byte
+	}
+	tests := []struct {
+		name    string
+		args    args
+		want    *flattenedFileObject
+		wantErr assert.ErrorAssertionFunc
+	}{
+		{
+			name: "with valid file",
+			args: args{
+				fileRoot: func() string { path, _ := os.Getwd(); return path + "/test/config/Files" }(),
+				fileName: []byte("testfile.txt"),
+				filePath: []byte{0, 0},
+			},
+			want: &flattenedFileObject{
+				FlatFileHeader:                NewFlatFileHeader(),
+				FlatFileInformationForkHeader: FlatFileInformationForkHeader{},
+				FlatFileInformationFork:       NewFlatFileInformationFork("testfile.txt"),
+				FlatFileDataForkHeader: FlatFileDataForkHeader{
+					ForkType:        []byte("DATA"),
+					CompressionType: []byte{0, 0, 0, 0},
+					RSVD:            []byte{0, 0, 0, 0},
+					DataSize:        []byte{0x00, 0x00, 0x00, 0x17},
+				},
+				FileData: nil,
+			},
+			wantErr: assert.NoError,
+		},
+		{
+			name: "when file path is invalid",
+			args: args{
+				fileRoot: func() string { path, _ := os.Getwd(); return path + "/test/config/Files" }(),
+				fileName: []byte("nope.txt"),
+			},
+			want:    nil,
+			wantErr: assert.Error,
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			got, err := NewFlattenedFileObject(tt.args.fileRoot, tt.args.filePath, tt.args.fileName)
+			if !tt.wantErr(t, err, fmt.Sprintf("NewFlattenedFileObject(%v, %v, %v)", tt.args.fileRoot, tt.args.filePath, tt.args.fileName)) {
+				return
+			}
+			assert.Equalf(t, tt.want, got, "NewFlattenedFileObject(%v, %v, %v)", tt.args.fileRoot, tt.args.filePath, tt.args.fileName)
+		})
+	}
+}