package awesomeProject

import (
    "reflect"
    "testing"
)

func Test_getFiles(t *testing.T) {
    tests := []struct {
        name    string
        dir     string
        want    []string
        wantErr bool
    }{
        {
            name:    "Happy path, empty folder",
            dir:     "testdata/getFiles/folder_empty",
            want:    nil,
            wantErr: false,
        },
        {
            name:    "Happy path, folder with 1 file",
            dir:     "testdata/getFiles/folder_1_file",
            want:    []string{
                "testfile1_1",
            },
            wantErr: false,
        },
        {
            name:    "Happy path, folder with 3 files",
            dir:     "testdata/getFiles/folder_3_files",
            want:    []string{
                "testfile3_1",
                "testfile3_2",
                "testfile3_3",
            },
            wantErr: false,
        },
        {
            name:    "Sad path, folder doesn't exist",
            dir:     "testdata/thistotallydoesn't exist",
            want:    nil,
            wantErr: true,
        },
    }

    for _, tt := range tests {
        t.Run(tt.name, func(t *testing.T) {
            // Run function
            got, err := getFiles(tt.dir)

            // Check error expectation
            if (err != nil) != tt.wantErr {
                t.Errorf("getFiles() error = %v, wantErr %v", err, tt.wantErr)
                return
            }

            // Check output expectation
            if !reflect.DeepEqual(got, tt.want) {
                t.Errorf("getFiles() got = %v, want %v", got, tt.want)
            }
        })
    }
}