Add test cases

Emyrk · Emyrk · commit 40586651b487 · 2022-08-25T09:14:38.000-04:00
diff --git a/coderd/httpmw/workspaceparam_test.go b/coderd/httpmw/workspaceparam_test.go
@@ -147,6 +147,14 @@ func TestWorkspaceAgentByNameParam(t *testing.T) {
 			ExpectedError:      "No agents exist",
 			ExpectedStatusCode: http.StatusBadRequest,
 		},
+		{
+			Name:               "NoAgentsSpecify",
+			WorkspaceName:      "dev",
+			Agents:             map[string][]string{},
+			URLParam:           "dev.agent",
+			ExpectedError:      "No agents exist",
+			ExpectedStatusCode: http.StatusBadRequest,
+		},
 		{
 			Name:          "MultipleAgents",
 			WorkspaceName: "dev",
@@ -232,6 +240,18 @@ func TestWorkspaceAgentByNameParam(t *testing.T) {
 			ExpectedAgent:      "agent-one",
 			ExpectedStatusCode: http.StatusOK,
 		},
+		{
+			Name:          "OneAgentSelected",
+			WorkspaceName: "dev",
+			Agents: map[string][]string{
+				"resource-a": {
+					"agent-one",
+				},
+			},
+			URLParam:           "dev.agent-one",
+			ExpectedAgent:      "agent-one",
+			ExpectedStatusCode: http.StatusOK,
+		},
 		{
 			Name:          "MultipleAgentSelectOne",
 			WorkspaceName: "dev",
