web_huangguo2_server/pub/storage/proxy/utils_test.go

package proxy

import "testing"

func TestImageUploadAddressRepair(t *testing.T) {
	type args struct {
		domain               string
		defaultBucketAddress []string
	}
	tests := []struct {
		name string
		args args
		want string
	}{
		{
			name: "a",
			args: args{
				domain:               "http://aaa",
				defaultBucketAddress: []string{},
			},
			want: "http://aaa/file/upload",
		},
		{
			name: "b",
			args: args{
				domain:               "http://ts.bbaazz.me",
				defaultBucketAddress: []string{},
			},
			want: "http://ts.bbaazz.me/file/upload",
		},
		{
			name: "c",
			args: args{
				domain:               "http://resourcets.bbaazz.me",
				defaultBucketAddress: []string{},
			},
			want: "http://resourcets.bbaazz.me/file/upload",
		},
		{
			name: "d",
			args: args{
				domain:               "",
				defaultBucketAddress: []string{BucketImageAddressV1},
			},
			want: "http://ts.bbaazz.me/file/upload",
		},
		{
			name: "e",
			args: args{
				domain:               "",
				defaultBucketAddress: []string{BucketImageAddressV2},
			},
			want: "http://resourcets.bbaazz.me/file/upload",
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			if got := ImageUploadAddressRepair(tt.args.domain, tt.args.defaultBucketAddress...); got != tt.want {
				t.Errorf("ImageUploadAddressRepair() = %v, want %v", got, tt.want)
			}
		})
	}
}

func TestVideoBodyAddressRepair(t *testing.T) {
	type args struct {
		domain string
	}
	tests := []struct {
		name string
		args args
		want string
	}{
		{
			name: "a",
			args: args{domain: "http://aaa"},
			want: "http://aaa/file/web/proxy/videoBodyV2",
		},
		{
			name: "b",
			args: args{domain: "http://ts.bbaazz.me"},
			want: "http://ts.bbaazz.me/file/web/proxy/videoBody",
		},
		{
			name: "c",
			args: args{domain: "http://resourcets.bbaazz.me"},
			want: "http://resourcets.bbaazz.me/file/web/proxy/videoBodyV2",
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			if got := VideoBodyAddressRepair(tt.args.domain); got != tt.want {
				t.Errorf("VideoBodyAddressRepair() = %v, want %v", got, tt.want)
			}
		})
	}
}

func TestVideoOkAddressRepair(t *testing.T) {
	type args struct {
		domain string
	}
	tests := []struct {
		name string
		args args
		want string
	}{
		{
			name: "a",
			args: args{domain: "http://aaa"},
			want: "http://aaa/file/web/proxy/videoOkv2",
		},
		{
			name: "b",
			args: args{domain: "http://ts.bbaazz.me"},
			want: "http://ts.bbaazz.me/file/web/proxy/videoOk",
		},
		{
			name: "c",
			args: args{domain: "http://resourcets.bbaazz.me"},
			want: "http://resourcets.bbaazz.me/file/web/proxy/videoOkv2",
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			if got := VideoOkAddressRepair(tt.args.domain); got != tt.want {
				t.Errorf("VideoOkAddressRepair() = %v, want %v", got, tt.want)
			}
		})
	}
}

func TestAppendV2PrefixToUrlPath(t *testing.T) {
	type args struct {
		name string
	}
	tests := []struct {
		name string
		args args
		want string
	}{
		{
			name: "a",
			args: args{
				name: "abc/aaa",
			},
			want: "v2/abc/aaa",
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			if got := AppendV2PrefixToUrlPath(tt.args.name); got != tt.want {
				t.Errorf("AppendV2PrefixToUrlPath() = %v, want %v", got, tt.want)
			}
		})
	}
}