API tests

mtojek · mtojek · commit 2c57c9671b8d · 2023-02-14T11:28:42.000Z
diff --git a/coderd/templateversions_test.go b/coderd/templateversions_test.go
@@ -1099,3 +1099,65 @@ func TestTemplateExamples(t *testing.T) {
 		require.EqualValues(t, ls, ex)
 	})
 }
+
+func TestTemplateVersionVariables(t *testing.T) {
+	t.Parallel()
+
+	templateVariables := []*proto.TemplateVariable{
+		{
+			Name:        "first_variable",
+			Description: "This is the first variable",
+			Type:        "string",
+			Required:    true,
+			Sensitive:   true,
+		},
+		{
+			Name:         "second_variable",
+			Description:  "This is the second variable",
+			DefaultValue: "123",
+			Type:         "number",
+		},
+	}
+
+	echoResponses := &echo.Responses{
+		Parse: []*proto.Parse_Response{
+			{
+				Type: &proto.Parse_Response_Complete{
+					Complete: &proto.Parse_Complete{
+						TemplateVariables: templateVariables,
+					},
+				},
+			},
+		},
+		ProvisionPlan: echo.ProvisionComplete,
+		ProvisionApply: []*proto.Provision_Response{{
+			Type: &proto.Provision_Response_Complete{
+				Complete: &proto.Provision_Complete{},
+			},
+		}},
+	}
+
+	client := coderdtest.New(t, &coderdtest.Options{IncludeProvisionerDaemon: true})
+	user := coderdtest.CreateFirstUser(t, client)
+	version := coderdtest.CreateTemplateVersion(t, client, user.OrganizationID, echoResponses)
+	templateVersion := coderdtest.AwaitTemplateVersionJob(t, client, version.ID)
+
+	ctx, cancel := context.WithTimeout(context.Background(), testutil.WaitShort)
+	defer cancel()
+
+	actualVariables, err := client.TemplateVersionVariables(ctx, templateVersion.ID)
+	require.NoError(t, err)
+
+	require.Len(t, actualVariables, 2)
+	for i := range templateVariables {
+		require.Equal(t, templateVariables[i].Name, actualVariables[i].Name)
+		require.Equal(t, templateVariables[i].Description, actualVariables[i].Description)
+		require.Equal(t, templateVariables[i].Type, actualVariables[i].Type)
+		require.Equal(t, templateVariables[i].DefaultValue, actualVariables[i].DefaultValue)
+		require.Equal(t, templateVariables[i].Required, actualVariables[i].Required)
+		require.Equal(t, templateVariables[i].Sensitive, actualVariables[i].Sensitive)
+	}
+
+	require.Equal(t, "", actualVariables[0].Value)
+	require.Equal(t, templateVariables[1].DefaultValue, actualVariables[1].Value)
+}
diff --git a/codersdk/templateversions.go b/codersdk/templateversions.go
@@ -150,6 +150,20 @@ func (c *Client) TemplateVersionResources(ctx context.Context, version uuid.UUID
 	return resources, json.NewDecoder(res.Body).Decode(&resources)
 }
 
+// TemplateVersionVariables returns resources a template version variables.
+func (c *Client) TemplateVersionVariables(ctx context.Context, version uuid.UUID) ([]TemplateVersionVariable, error) {
+	res, err := c.Request(ctx, http.MethodGet, fmt.Sprintf("/api/v2/templateversions/%s/variables", version), nil)
+	if err != nil {
+		return nil, err
+	}
+	defer res.Body.Close()
+	if res.StatusCode != http.StatusOK {
+		return nil, ReadBodyAsError(res)
+	}
+	var variables []TemplateVersionVariable
+	return variables, json.NewDecoder(res.Body).Decode(&variables)
+}
+
 // TemplateVersionLogsBefore returns logs that occurred before a specific log ID.
 func (c *Client) TemplateVersionLogsBefore(ctx context.Context, version uuid.UUID, before int64) ([]ProvisionerJobLog, error) {
 	return c.provisionerJobLogsBefore(ctx, fmt.Sprintf("/api/v2/templateversions/%s/logs", version), before)
