restic/src/restic/backend/location/location_test.go

package location

import (
	"net/url"
	"reflect"
	"testing"

	"restic/backend/b2"
	"restic/backend/local"
	"restic/backend/rest"
	"restic/backend/s3"
	"restic/backend/sftp"
)

func parseURL(s string) *url.URL {
	u, err := url.Parse(s)
	if err != nil {
		panic(err)
	}

	return u
}

var parseTests = []struct {
	s string
	u Location
}{
	{
		"local:/srv/repo",
		Location{Scheme: "local",
			Config: local.Config{
				Path: "/srv/repo",
			},
		},
	},
	{
		"local:dir1/dir2",
		Location{Scheme: "local",
			Config: local.Config{
				Path: "dir1/dir2",
			},
		},
	},
	{
		"local:dir1/dir2",
		Location{Scheme: "local",
			Config: local.Config{
				Path: "dir1/dir2",
			},
		},
	},
	{
		"dir1/dir2",
		Location{Scheme: "local",
			Config: local.Config{
				Path: "dir1/dir2",
			},
		},
	},
	{
		"local:../dir1/dir2",
		Location{Scheme: "local",
			Config: local.Config{
				Path: "../dir1/dir2",
			},
		},
	},
	{
		"/dir1/dir2",
		Location{Scheme: "local",
			Config: local.Config{
				Path: "/dir1/dir2",
			},
		},
	},

	{
		"sftp:user@host:/srv/repo",
		Location{Scheme: "sftp",
			Config: sftp.Config{
				User: "user",
				Host: "host",
				Path: "/srv/repo",
			},
		},
	},
	{
		"sftp:host:/srv/repo",
		Location{Scheme: "sftp",
			Config: sftp.Config{
				User: "",
				Host: "host",
				Path: "/srv/repo",
			},
		},
	},
	{
		"sftp://user@host/srv/repo",
		Location{Scheme: "sftp",
			Config: sftp.Config{
				User: "user",
				Host: "host",
				Path: "srv/repo",
			},
		},
	},
	{
		"sftp://user@host//srv/repo",
		Location{Scheme: "sftp",
			Config: sftp.Config{
				User: "user",
				Host: "host",
				Path: "/srv/repo",
			},
		},
	},

	{
		"s3://eu-central-1/bucketname",
		Location{Scheme: "s3",
			Config: s3.Config{
				Endpoint: "eu-central-1",
				Bucket:   "bucketname",
				Prefix:   "restic",
			},
		},
	},
	{
		"s3://hostname.foo/bucketname",
		Location{Scheme: "s3",
			Config: s3.Config{
				Endpoint: "hostname.foo",
				Bucket:   "bucketname",
				Prefix:   "restic",
			},
		},
	},
	{
		"s3://hostname.foo/bucketname/prefix/directory",
		Location{Scheme: "s3",
			Config: s3.Config{
				Endpoint: "hostname.foo",
				Bucket:   "bucketname",
				Prefix:   "prefix/directory",
			},
		},
	},
	{
		"s3:eu-central-1/repo",
		Location{Scheme: "s3",
			Config: s3.Config{
				Endpoint: "eu-central-1",
				Bucket:   "repo",
				Prefix:   "restic",
			},
		},
	},
	{
		"s3:eu-central-1/repo/prefix/directory",
		Location{Scheme: "s3",
			Config: s3.Config{
				Endpoint: "eu-central-1",
				Bucket:   "repo",
				Prefix:   "prefix/directory",
			},
		},
	},
	{
		"s3:https://hostname.foo/repo",
		Location{Scheme: "s3",
			Config: s3.Config{
				Endpoint: "hostname.foo",
				Bucket:   "repo",
				Prefix:   "restic",
			},
		},
	},
	{
		"s3:https://hostname.foo/repo/prefix/directory",
		Location{Scheme: "s3",
			Config: s3.Config{
				Endpoint: "hostname.foo",
				Bucket:   "repo",
				Prefix:   "prefix/directory",
			},
		},
	},
	{
		"s3:http://hostname.foo/repo",
		Location{Scheme: "s3",
			Config: s3.Config{
				Endpoint: "hostname.foo",
				Bucket:   "repo",
				Prefix:   "restic",
				UseHTTP:  true,
			},
		},
	},
	{
		"rest:http://hostname.foo:1234/",
		Location{Scheme: "rest",
			Config: rest.Config{
				URL: parseURL("http://hostname.foo:1234/"),
			},
		},
	},
	{
		"b2:bucketname:/prefix", Location{Scheme: "b2",
			Config: b2.Config{
				Bucket:      "bucketname",
				Prefix:      "prefix",
				Connections: 5,
			},
		},
	},
	{
		"b2:bucketname", Location{Scheme: "b2",
			Config: b2.Config{
				Bucket:      "bucketname",
				Prefix:      "",
				Connections: 5,
			},
		},
	},
}

func TestParse(t *testing.T) {
	for i, test := range parseTests {
		t.Run(test.s, func(t *testing.T) {
			u, err := Parse(test.s)
			if err != nil {
				t.Fatalf("unexpected error: %v", err)
			}

			if test.u.Scheme != u.Scheme {
				t.Errorf("test %d: scheme does not match, want %q, got %q",
					i, test.u.Scheme, u.Scheme)
			}

			if !reflect.DeepEqual(test.u.Config, u.Config) {
				t.Errorf("test %d: cfg map does not match, want:\n  %#v\ngot: \n  %#v",
					i, test.u.Config, u.Config)
			}
		})
	}
}