Add timeout field

coder · Mar 15, 2023 · c50fa55 · c50fa55
1 parent 208a014
commit c50fa55
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 0 deletions.
diff --git a/docs/resources/agent.md b/docs/resources/agent.md
@@ -77,5 +77,6 @@ Required:
 Optional:
 
 - `display_name` (String) The user-facing name of this value.
+- `timeout` (Number) The maximum time the command is allowed to run in seconds.
 
 
diff --git a/provider/agent.go b/provider/agent.go
@@ -165,6 +165,12 @@ func agentResource() *schema.Resource {
 								Type: schema.TypeString,
 							},
 						},
+						"timeout": {
+							Type:        schema.TypeInt,
+							Description: "The maximum time the command is allowed to run in seconds.",
+							ForceNew:    true,
+							Optional:    true,
+						},
 						"interval": {
 							Type:        schema.TypeInt,
 							Description: "The interval in seconds at which to refresh this metadata item. ",

diff --git a/provider/agent_test.go b/provider/agent_test.go
@@ -132,6 +132,7 @@ func TestAgent_Metadata(t *testing.T) {
 						display_name = "Process Count"
 						cmd = ["sh", "-c", "ps aux | wc -l"]
 						interval = 5
+						timeout = 1
 					}
 				}
 				`,
@@ -153,6 +154,7 @@ func TestAgent_Metadata(t *testing.T) {
 				require.Equal(t, "-c", attr["metadata.0.cmd.1"])
 				require.Equal(t, "ps aux | wc -l", attr["metadata.0.cmd.2"])
 				require.Equal(t, "5", attr["metadata.0.interval"])
+				require.Equal(t, "1", attr["metadata.0.timeout"])
 				return nil
 			},
 		}},