package file import ( "fmt" "io" "net/url" "path/filepath" "testing" "github.com/pkg/errors" ) const dummyFilePath = "testdata/dummy.txt" var loaderHandlerTestCases []loaderHandlerTestCase func init() { dummyFileAbsolutePath, err := filepath.Abs(dummyFilePath) if err != nil { panic(errors.WithStack(err)) } loaderHandlerTestCases = []loaderHandlerTestCase{ { URL: SchemeFile + "://" + dummyFilePath, ExpectMatch: true, ExpectOpenError: false, ExpectOpenContent: "dummy", }, { URL: dummyFilePath, ExpectMatch: true, ExpectOpenError: false, ExpectOpenContent: "dummy", }, { URL: dummyFileAbsolutePath, ExpectMatch: true, ExpectOpenError: false, ExpectOpenContent: "dummy", }, { URL: SchemeFile + "://" + dummyFileAbsolutePath, ExpectMatch: true, ExpectOpenError: false, ExpectOpenContent: "dummy", }, } } type loaderHandlerTestCase struct { URL string ExpectMatch bool ExpectOpenError bool ExpectOpenContent string } func TestLoaderHandler(t *testing.T) { t.Parallel() handler := NewLoaderHandler() for _, tc := range loaderHandlerTestCases { func(tc loaderHandlerTestCase) { t.Run(fmt.Sprintf("Load '%s'", tc.URL), func(t *testing.T) { t.Parallel() url, err := url.Parse(tc.URL) if err != nil { t.Fatal(errors.Wrapf(err, "could not parse url '%s'", tc.URL)) } if e, g := tc.ExpectMatch, handler.Match(url); e != g { t.Errorf("URL '%s': expected matching result '%v', got '%v'", tc.URL, e, g) } if !tc.ExpectMatch { return } reader, err := handler.Open(url) if err != nil && !tc.ExpectOpenError { t.Fatal(errors.Wrapf(err, "could not open url '%s'", url.String())) } defer func() { if err := reader.Close(); err != nil { t.Error(errors.WithStack(err)) } }() if tc.ExpectOpenError { t.Fatal(errors.Errorf("no error was returned as expected when opening url '%s'", url.String())) } data, err := io.ReadAll(reader) if err != nil { t.Fatal(errors.WithStack(err)) } if e, g := tc.ExpectOpenContent, string(data); e != g { t.Errorf("URL '%s': expected content'%v', got '%v'", tc.URL, e, g) } }) }(tc) } }