[datadog_synthetics_test] Correctly set grpc step request

datadog/resource_datadog_synthetics_test_.go

@@ -1565,17 +1565,14 @@ func buildSyntheticsAPITestStruct(d *schema.ResourceData) *datadogV1.SyntheticsA
 			requests := stepMap["request_definition"].([]interface{})
 			if len(requests) > 0 && requests[0] != nil {
 				requestMap := requests[0].(map[string]interface{})
-				request.SetMethod(requestMap["method"].(string))
-				request.SetUrl(requestMap["url"].(string))
-				request.SetBody(requestMap["body"].(string))
-				if v, ok := requestMap["body_type"].(string); ok && v != "" {
-					request.SetBodyType(datadogV1.SyntheticsTestRequestBodyType(v))
-				}
 				request.SetTimeout(float64(requestMap["timeout"].(int)))
 				request.SetAllowInsecure(requestMap["allow_insecure"].(bool))
-				request.SetFollowRedirects(requestMap["follow_redirects"].(bool))
-				request.SetPersistCookies(requestMap["persist_cookies"].(bool))
-				request.SetNoSavingResponseBody(requestMap["no_saving_response_body"].(bool))
+				if stepMap["subtype"] == "http" {
+					setHTTPStepRequest(requestMap, &request)
+				}
+				if stepMap["subtype"] == "grpc" {
+					setGRPCStepRequest(requestMap, &request)
+				}
 			}
 
 			request = *completeSyntheticsTestRequest(request, stepMap["request_headers"].(map[string]interface{}), stepMap["request_query"].(map[string]interface{}), stepMap["request_basicauth"].([]interface{}), stepMap["request_client_certificate"].([]interface{}), stepMap["request_proxy"].([]interface{}), map[string]interface{}{})
@@ -3451,3 +3448,48 @@ func convertStepParamsKey(key string) string {
 
 	return key
 }
+
+func setHTTPStepRequest(requestMap map[string]interface{}, request *datadogV1.SyntheticsTestRequest) {
+	request.SetMethod(requestMap["method"].(string))
+	request.SetUrl(requestMap["url"].(string))
+	request.SetBody(requestMap["body"].(string))
+	if v, ok := requestMap["body_type"].(string); ok && v != "" {
+		request.SetBodyType(datadogV1.SyntheticsTestRequestBodyType(v))
+	}
+	request.SetFollowRedirects(requestMap["follow_redirects"].(bool))
+	request.SetPersistCookies(requestMap["persist_cookies"].(bool))
+	request.SetNoSavingResponseBody(requestMap["no_saving_response_body"].(bool))
+}
+
+func setGRPCStepRequest(requestMap map[string]interface{}, request *datadogV1.SyntheticsTestRequest) {
+	request.SetCallType(datadogV1.SyntheticsTestCallType(requestMap["call_type"].(string)))
+	request.SetService(requestMap["service"].(string))
+	request.SetHost(requestMap["host"].(string))
+	request.SetPort(int64(requestMap["port"].(int)))
+	if requestMap["call_type"].(string) == "unary" {
+		request.SetMessage(requestMap["message"].(string))
+		request.SetMethod(requestMap["method"].(string))
+	}
+	if requestMap["plain_proto_file"] != nil {
+		stringifiedValue, _ := json.Marshal(requestMap["plain_proto_file"].(string))
+		var compressedValue bytes.Buffer
+		zl := zlib.NewWriter(&compressedValue)
+		zl.Write(stringifiedValue)
+		zl.Close()
+		encodedCompressedProtoFile := b64.StdEncoding.EncodeToString(compressedValue.Bytes())
+
+		request.SetCompressedProtoFile(encodedCompressedProtoFile)
+	}
+
+	// This is effectively useless as it will not work but is necessary for the state to be in sync
+	if requestMap["proto_json_descriptor"] != nil {
+		stringifiedValue, _ := json.Marshal(requestMap["proto_json_descriptor"].(string))
+		var compressedValue bytes.Buffer
+		zl := zlib.NewWriter(&compressedValue)
+		zl.Write(stringifiedValue)
+		zl.Close()
+		encodedCompressedJsonDescriptor := b64.StdEncoding.EncodeToString(compressedValue.Bytes())
+
+		request.SetCompressedJsonDescriptor(encodedCompressedJsonDescriptor)
+	}
+}

datadog/tests/cassettes/TestAccDatadogSyntheticsTestMultistepApi_Basic.freeze

@@ -1 +1 @@
-2024-01-19T11:58:30.290388+01:00
+2024-03-11T15:52:13.316572+01:00

datadog/tests/resource_datadog_synthetics_test_test.go

@@ -4103,7 +4103,7 @@ func createSyntheticsMultistepAPITest(ctx context.Context, accProvider func() (*
 			resource.TestCheckResourceAttr(
 				"datadog_synthetics_test.multi", "status", "paused"),
 			resource.TestCheckResourceAttr(
-				"datadog_synthetics_test.multi", "api_step.#", "4"),
+				"datadog_synthetics_test.multi", "api_step.#", "5"),
 			resource.TestCheckResourceAttr(
 				"datadog_synthetics_test.multi", "api_step.0.name", "First api step"),
 			resource.TestCheckResourceAttr(
@@ -4248,6 +4248,20 @@ func createSyntheticsMultistepAPITest(ctx context.Context, accProvider func() (*
 				"datadog_synthetics_test.multi", "api_step.3.request_basicauth.0.username", "username"),
 			resource.TestCheckResourceAttr(
 				"datadog_synthetics_test.multi", "api_step.3.request_basicauth.0.password", "password"),
+			resource.TestCheckResourceAttr(
+				"datadog_synthetics_test.multi", "api_step.4.request_definition.0.host", "fakehost.io"),
+			resource.TestCheckResourceAttr(
+				"datadog_synthetics_test.multi", "api_step.4.request_definition.0.service", "OboService"),
+			resource.TestCheckResourceAttr(
+				"datadog_synthetics_test.multi", "api_step.4.request_definition.0.port", "8443"),
+			resource.TestCheckResourceAttr(
+				"datadog_synthetics_test.multi", "api_step.4.request_definition.0.method", "GetOboUserJwt"),
+			resource.TestCheckResourceAttr(
+				"datadog_synthetics_test.multi", "api_step.4.request_definition.0.message", "{\"user_uuid\": \"1234\"}"),
+			resource.TestCheckResourceAttr(
+				"datadog_synthetics_test.multi", "api_step.4.request_definition.0.call_type", "unary"),
+			resource.TestCheckResourceAttr(
+				"datadog_synthetics_test.multi", "api_step.4.request_definition.0.plain_proto_file", "syntax = \"proto3\";\npackage pb;\noption go_package = \"github.com/DataDog/dd-source/domains/aaa_authn/apps/apis/obo/pb\";\nmessage GetOboUserJwtRequest {\n  string user_uuid = 1; \n}\nmessage GetOboJwtResponse {\n  string jwt = 1;\n}\nservice OboService {\n  rpc GetOboUserJwt(GetOboUserJwtRequest) returns (GetOboJwtResponse);\n}\n"),
 			resource.TestCheckResourceAttr(
 				"datadog_synthetics_test.multi", "config_variable.0.type", "global"),
 			resource.TestCheckResourceAttr(
@@ -4429,6 +4443,33 @@ resource "datadog_synthetics_test" "multi" {
       target   = "200"
     }
   }
+
+  api_step {
+    name    = "GRPC step"
+    subtype = "grpc"
+    request_definition {
+      call_type = "unary"
+      method           = "GetOboUserJwt"
+      host             = "fakehost.io"
+      port             = 8443
+      service          = "OboService"
+      message          = "{\"user_uuid\": \"1234\"}"
+      plain_proto_file = <<EOT
+syntax = "proto3";
+package pb;
+option go_package = "github.com/DataDog/dd-source/domains/aaa_authn/apps/apis/obo/pb";
+message GetOboUserJwtRequest {
+  string user_uuid = 1; 
+}
+message GetOboJwtResponse {
+  string jwt = 1;
+}
+service OboService {
+  rpc GetOboUserJwt(GetOboUserJwtRequest) returns (GetOboJwtResponse);
+}
+EOT
+    }
+  }
 }
 `, testName, variableName)
 }