snapshot of https://github.com/golang-migrate/migrate.git (master, 2026-09-15; blobs > 131072B omitted)

source/httpfs/partial_driver_test.go

@@ -0,0 +1,107 @@
+package httpfs_test
+
+import (
+	"errors"
+	"net/http"
+	"strings"
+	"testing"
+
+	"github.com/golang-migrate/migrate/v4/source"
+	"github.com/golang-migrate/migrate/v4/source/httpfs"
+	st "github.com/golang-migrate/migrate/v4/source/testing"
+)
+
+type driver struct{ httpfs.PartialDriver }
+
+func (d *driver) Open(url string) (source.Driver, error) { return nil, errors.New("X") }
+
+type driverExample struct {
+	httpfs.PartialDriver
+}
+
+func (d *driverExample) Open(url string) (source.Driver, error) {
+	parts := strings.Split(url, ":")
+	dir := parts[0]
+	path := ""
+	if len(parts) >= 2 {
+		path = parts[1]
+	}
+
+	var de driverExample
+	return &de, de.Init(http.Dir(dir), path)
+}
+
+func TestDriverExample(t *testing.T) {
+	d, err := (*driverExample)(nil).Open("testdata:sql")
+	if err != nil {
+		t.Errorf("Open() returned error: %s", err)
+	}
+	st.Test(t, d)
+}
+
+func TestPartialDriverInit(t *testing.T) {
+	tests := []struct {
+		name string
+		fs   http.FileSystem
+		path string
+		ok   bool
+	}{
+		{
+			name: "valid dir and empty path",
+			fs:   http.Dir("testdata/sql"),
+			ok:   true,
+		},
+		{
+			name: "valid dir and non-empty path",
+			fs:   http.Dir("testdata"),
+			path: "sql",
+			ok:   true,
+		},
+		{
+			name: "invalid dir",
+			fs:   http.Dir("does-not-exist"),
+		},
+		{
+			name: "file instead of dir",
+			fs:   http.Dir("testdata/sql/1_foobar.up.sql"),
+		},
+		{
+			name: "dir with duplicates",
+			fs:   http.Dir("testdata/duplicates"),
+		},
+	}
+
+	for _, test := range tests {
+		t.Run(test.name, func(t *testing.T) {
+			var d driver
+			err := d.Init(test.fs, test.path)
+			if test.ok {
+				if err != nil {
+					t.Errorf("Init() returned error %s", err)
+				}
+				st.Test(t, &d)
+				if err = d.Close(); err != nil {
+					t.Errorf("Init().Close() returned error %s", err)
+				}
+			} else {
+				if err == nil {
+					t.Errorf("Init() expected error but did not get one")
+				}
+			}
+		})
+	}
+
+}
+
+func TestFirstWithNoMigrations(t *testing.T) {
+	var d driver
+	fs := http.Dir("testdata/no-migrations")
+
+	if err := d.Init(fs, ""); err != nil {
+		t.Errorf("No error on Init() expected, got: %v", err)
+	}
+
+	if _, err := d.First(); err == nil {
+		t.Errorf("Expected error on First(), got: %v", err)
+	}
+}