Merge pull request #73 from mamclaughlin/userdataFeature

remove encoding for user data input
vultr · Aug 22, 2020 · 1273851 · 1273851
2 parents da44f60 + 5e6ad07
commit 1273851
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 8 deletions.
diff --git a/bare_metal_server.go b/bare_metal_server.go
@@ -767,11 +767,9 @@ func (b *BareMetalServerServiceHandler) SetTag(ctx context.Context, serverID, ta
 func (b *BareMetalServerServiceHandler) SetUserData(ctx context.Context, serverID, userData string) error {
 	uri := "/v1/baremetal/set_user_data"
 
-	encodedUserData := base64.StdEncoding.EncodeToString([]byte(userData))
-
 	values := url.Values{
 		"SUBID":    {serverID},
-		"userdata": {encodedUserData},
+		"userdata": {userData},
 	}
 
 	req, err := b.client.NewRequest(ctx, http.MethodPost, uri, values)

diff --git a/bare_metal_server_test.go b/bare_metal_server_test.go
@@ -854,7 +854,7 @@ func TestBareMetalServerServiceHandler_SetUserData(t *testing.T) {
 		fmt.Fprint(writer)
 	})
 
-	err := client.BareMetalServer.SetUserData(ctx, "900000", "user-test-data")
+	err := client.BareMetalServer.SetUserData(ctx, "900000", "ZWNobyBIZWxsbyBXb3JsZA==")
 
 	if err != nil {
 		t.Errorf("Server.SetUserData return %+v ", err)

diff --git a/server.go b/server.go
@@ -790,11 +790,9 @@ func (s *ServerServiceHandler) SetUserData(ctx context.Context, instanceID, user
 
 	uri := "/v1/server/set_user_data"
 
-	encodedUserData := base64.StdEncoding.EncodeToString([]byte(userData))
-
 	values := url.Values{
 		"SUBID":    {instanceID},
-		"userdata": {encodedUserData},
+		"userdata": {userData},
 	}
 
 	req, err := s.client.NewRequest(ctx, http.MethodPost, uri, values)

diff --git a/server_test.go b/server_test.go
@@ -452,7 +452,7 @@ func TestServerServiceHandler_SetUserData(t *testing.T) {
 		fmt.Fprint(writer)
 	})
 
-	err := client.Server.SetUserData(ctx, "1234", "user-test-data")
+	err := client.Server.SetUserData(ctx, "1234", "ZWNobyBIZWxsbyBXb3JsZA==")
 
 	if err != nil {
 		t.Errorf("Server.SetUserData return %+v ", err)