add unit test cases

private/protocol/rest/build_test.go

@@ -15,34 +15,54 @@ import (
 )
 
 func TestBuildURI(t *testing.T) {
-	in := struct {
-		Topic *string `location:"uri" locationName:"topic" type:"string" required:"true"`
+	cases := map[string]struct {
+		Topic         *string
+		ExpectPath    string
+		ExpectRawPath string
 	}{
-		Topic: aws.String("///devices/123/test"),
+		"path without prefix slash": {
+			Topic:         aws.String("devices/123/test"),
+			ExpectPath:    "/topics/devices/123/test",
+			ExpectRawPath: "/topics/devices%2F123%2Ftest",
+		},
+		"path containing single prefix slashes": {
+			Topic:         aws.String("/devices/123/test"),
+			ExpectPath:    "/topics//devices/123/test",
+			ExpectRawPath: "/topics/%2Fdevices%2F123%2Ftest",
+		},
+		"path containing prefix multi-slashes": {
+			Topic:         aws.String("///devices/123/test"),
+			ExpectPath:    "/topics////devices/123/test",
+			ExpectRawPath: "/topics/%2F%2F%2Fdevices%2F123%2Ftest",
+		},
 	}
 
-	uri := &url.URL{
-		Path:   "/topics/{topic}",
-		Scheme: "https",
-		Host:   "host",
-	}
+	for _, c := range cases {
+		uri := &url.URL{
+			Path:   "/topics/{topic}",
+			Scheme: "https",
+			Host:   "host",
+		}
 
-	req := &request.Request{
-		HTTPRequest: &http.Request{
-			URL: uri,
-		},
-		Params: &in,
-	}
+		req := &request.Request{
+			HTTPRequest: &http.Request{
+				URL: uri,
+			},
+			Params: &struct {
+				Topic *string `location:"uri" locationName:"topic" type:"string" required:"true"`
+			}{
+				c.Topic,
+			},
+		}
 
-	Build(req)
+		Build(req)
 
-	expectedPath := "/topics////devices/123/test"
-	expectedRawPath := "/topics/%2F%2F%2Fdevices%2F123%2Ftest"
-	if a, e := uri.Path, expectedPath; a != e {
-		t.Errorf("expect %q Path, got %q", e, a)
-	}
-	if a, e := uri.RawPath, expectedRawPath; a != e {
-		t.Errorf("expect %q RawPath, got %q", e, a)
+		if a, e := uri.Path, c.ExpectPath; a != e {
+			t.Errorf("expect %q Path, got %q", e, a)
+		}
+		if a, e := uri.RawPath, c.ExpectRawPath; a != e {
+			t.Errorf("expect %q RawPath, got %q", e, a)
+		}
 	}
 }