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) } }) } }