fix: fix tests

Author: 35C4n0r

registry/coder/modules/claude-code/README.md

@@ -1,6 +1,6 @@
 ---
 display_name: Claude Code
-description: Run the Claude Code agent in your workspace to generate code and perform tasks.
+description: Run the Claude Code agent in your workspace.
 icon: ../../../../.icons/claude.svg
 verified: true
 tags: [agent, claude-code, ai, tasks, anthropic]
@@ -33,12 +33,6 @@ module "claude-code" {
 This example shows how to configure the Claude Code module with a task prompt, API key, and other custom settings.
 
 ```tf
-variable "anthropic_api_key" {
-  type        = string
-  description = "The Anthropic API key."
-  sensitive   = true
-}
-
 data "coder_parameter" "task_prompt" {
   type        = "string"
   name        = "AI Task Prompt"
@@ -47,21 +41,16 @@ data "coder_parameter" "task_prompt" {
   mutable     = true
 }
 
-module "coder-login" {
-  count    = data.coder_workspace.me.start_count
-  source   = "registry.coder.com/coder/coder-login/coder"
-  version  = "1.0.31" # Use a recent version
-  agent_id = coder_agent.example.id
-}
-
 module "claude-code" {
   source   = "registry.coder.com/coder/claude-code/coder"
   version  = "3.0.0"
   agent_id = coder_agent.example.id
   workdir  = "/home/coder/project"
 
-  # --- Authentication ---
-  claude_api_key = var.anthropic_api_key # required for tasks
+  # --- Authentication --- required for tasks
+  claude_api_key = "xxxx-xxxxx-xxxx"
+  # OR
+  claude_code_oauth_token = "xxxxx-xxxx-xxxx"
 
   # --- Versioning ---
   claude_code_version = "1.0.82" # Pin to a specific version

registry/coder/modules/claude-code/main.test.ts

@@ -134,6 +134,7 @@ describe("claude-code", async () => {
       },
     });
     const { id } = await setup({
+      skipClaudeMock: true,
       moduleVariables: {
         mcp: mcpConfig,
       },
@@ -142,7 +143,7 @@ describe("claude-code", async () => {
 
     const resp = await readFileContainer(
       id,
-      "/home/coder/.claude-module/install.log",
+      "/home/coder/.claude.json",
     );
     expect(resp).toContain("test-cmd");
   });
@@ -151,7 +152,7 @@ describe("claude-code", async () => {
     const prompt = "This is a task prompt for Claude.";
     const { id } = await setup({
       moduleVariables: {
-        task_prompt: prompt,
+        ai_prompt: prompt,
       },
     });
     await execModuleScript(id);

registry/coder/modules/claude-code/main.tf

@@ -89,6 +89,12 @@ variable "agentapi_version" {
   default     = "v0.6.1"
 }
 
+variable "ai_prompt" {
+  type        = string
+  description = "Initial task prompt for Claude Code."
+  default     = ""
+}
+
 # ---------------------------------------------------------------
 
 variable "install_claude_code" {
@@ -162,13 +168,21 @@ variable "disallowed_tools" {
 
 }
 
-variable "ai_prompt" {
+variable "claude_code_oauth_token" {
   type        = string
-  description = "Task prompt for the Claude Code CLI"
-  default     = ""
+  description = "Set up a long-lived authentication token (requires Claude subscription). Generated using `claude setup-token` command"
+  sensitive   = true
+}
+
+resource "coder_env" "claude_code_oauth_token" {
+  agent_id = var.agent_id
+  name     = "CLAUDE_CODE_OAUTH_TOKEN"
+  value    = var.claude_code_oauth_token
 }
 
 resource "coder_env" "claude_api_key" {
+  count = length(var.claude_api_key) > 0 ? 1 : 0
+
   agent_id = var.agent_id
   name     = "CLAUDE_API_KEY"
   value    = var.claude_api_key

registry/coder/modules/claude-code/scripts/install.sh

@@ -67,7 +67,7 @@ function setup_claude_configurations() {
     claude mcp add "$server_name" "$server_json"
     echo "------------------------"
     echo ""
-  done < <(echo "${MCP:-{}}" | jq -r '.mcpServers | to_entries[] | .key, (.value | @json)')
+  done < <(echo "${ARG_MCP:-{}}" | jq -r '.mcpServers | to_entries[] | .key, (.value | @json)')
 
   if [ -n "$ARG_ALLOWED_TOOLS" ]; then
     coder --allowedTools "$ARG_ALLOWED_TOOLS"