|
| 1 | +import { describe, it, expect } from "bun:test"; |
| 2 | +import { |
| 3 | + runTerraformApply, |
| 4 | + runTerraformInit, |
| 5 | + testRequiredVariables, |
| 6 | + findResourceInstance, |
| 7 | +} from "~test"; |
| 8 | +import path from "path"; |
| 9 | + |
| 10 | +const moduleDir = path.resolve(__dirname); |
| 11 | + |
| 12 | +const requiredVars = { |
| 13 | + agent_id: "dummy-agent-id", |
| 14 | +}; |
| 15 | + |
| 16 | +const fullConfigVars = { |
| 17 | + agent_id: "dummy-agent-id", |
| 18 | + install_amazon_q: true, |
| 19 | + install_agentapi: true, |
| 20 | + agentapi_version: "v0.6.0", |
| 21 | + amazon_q_version: "1.14.1", |
| 22 | + q_install_url: "https://desktop-release.q.us-east-1.amazonaws.com", |
| 23 | + trust_all_tools: false, |
| 24 | + ai_prompt: "Build a comprehensive test suite", |
| 25 | + auth_tarball: "dGVzdEF1dGhUYXJiYWxs", // base64 "testAuthTarball" |
| 26 | + system_prompt: "You are a helpful AI assistant for testing", |
| 27 | + order: 1, |
| 28 | + group: "AI Tools", |
| 29 | + icon: "/icon/custom-amazon-q.svg", |
| 30 | + pre_install_script: "echo 'Starting pre-install'", |
| 31 | + post_install_script: "echo 'Completed post-install'", |
| 32 | + agent_config: JSON.stringify({ |
| 33 | + name: "test-agent", |
| 34 | + description: "Test agent configuration for comprehensive testing", |
| 35 | + prompt: "Custom test prompt", |
| 36 | + tools: ["fs_read", "fs_write", "execute_bash", "use_aws", "knowledge"], |
| 37 | + allowedTools: ["fs_read"], |
| 38 | + resources: ["file://README.md", "file://.amazonq/rules/**/*.md"] |
| 39 | + }) |
| 40 | +}; |
| 41 | + |
| 42 | +describe("amazon-q module v2.0.0", async () => { |
| 43 | + await runTerraformInit(moduleDir); |
| 44 | + |
| 45 | + // 1. Required variables test |
| 46 | + testRequiredVariables(moduleDir, requiredVars); |
| 47 | + |
| 48 | + // 2. Environment variables are created correctly |
| 49 | + it("creates required environment variables", async () => { |
| 50 | + const state = await runTerraformApply(moduleDir, fullConfigVars); |
| 51 | + |
| 52 | + // Check status slug environment variable |
| 53 | + const statusSlugEnv = findResourceInstance(state, "coder_env", "status_slug"); |
| 54 | + expect(statusSlugEnv).toBeDefined(); |
| 55 | + expect(statusSlugEnv.name).toBe("CODER_MCP_APP_STATUS_SLUG"); |
| 56 | + expect(statusSlugEnv.value).toBe("amazonq"); |
| 57 | + |
| 58 | + // Check auth tarball environment variable |
| 59 | + const authTarballEnv = findResourceInstance(state, "coder_env", "auth_tarball"); |
| 60 | + expect(authTarballEnv).toBeDefined(); |
| 61 | + expect(authTarballEnv.name).toBe("AMAZON_Q_AUTH_TARBALL"); |
| 62 | + expect(authTarballEnv.value).toBe("dGVzdEF1dGhUYXJiYWxs"); |
| 63 | + }); |
| 64 | + |
| 65 | + // 3. Empty auth tarball handling |
| 66 | + it("handles empty auth tarball correctly", async () => { |
| 67 | + const noAuthVars = { |
| 68 | + ...requiredVars, |
| 69 | + auth_tarball: "" |
| 70 | + }; |
| 71 | + |
| 72 | + const state = await runTerraformApply(moduleDir, noAuthVars); |
| 73 | + |
| 74 | + // Auth tarball environment variable should not be created when empty |
| 75 | + const authTarballEnv = state.resources?.find(r => |
| 76 | + r.type === "coder_env" && r.name === "auth_tarball" |
| 77 | + ); |
| 78 | + expect(authTarballEnv).toBeUndefined(); |
| 79 | + }); |
| 80 | + |
| 81 | + // 4. Status slug is always created |
| 82 | + it("creates status slug environment variable", async () => { |
| 83 | + const state = await runTerraformApply(moduleDir, requiredVars); |
| 84 | + |
| 85 | + // Status slug should always be configured |
| 86 | + const statusSlugEnv = findResourceInstance(state, "coder_env", "status_slug"); |
| 87 | + expect(statusSlugEnv).toBeDefined(); |
| 88 | + expect(statusSlugEnv.name).toBe("CODER_MCP_APP_STATUS_SLUG"); |
| 89 | + expect(statusSlugEnv.value).toBe("amazonq"); |
| 90 | + }); |
| 91 | + |
| 92 | + // 5. Install options configuration |
| 93 | + it("respects install option flags", async () => { |
| 94 | + const noInstallVars = { |
| 95 | + ...requiredVars, |
| 96 | + install_amazon_q: false, |
| 97 | + install_agentapi: false |
| 98 | + }; |
| 99 | + |
| 100 | + const state = await runTerraformApply(moduleDir, noInstallVars); |
| 101 | + |
| 102 | + // Status slug should still be configured even when install options are disabled |
| 103 | + const statusSlugEnv = findResourceInstance(state, "coder_env", "status_slug"); |
| 104 | + expect(statusSlugEnv).toBeDefined(); |
| 105 | + expect(statusSlugEnv.value).toBe("amazonq"); |
| 106 | + }); |
| 107 | + |
| 108 | + // 6. Custom system prompt handling (without JSON parsing issues) |
| 109 | + it("handles custom system prompt", async () => { |
| 110 | + const customPromptVars = { |
| 111 | + ...requiredVars, |
| 112 | + system_prompt: "Custom system prompt for testing" |
| 113 | + }; |
| 114 | + |
| 115 | + const state = await runTerraformApply(moduleDir, customPromptVars); |
| 116 | + |
| 117 | + // Should create the basic resources |
| 118 | + const statusSlugEnv = findResourceInstance(state, "coder_env", "status_slug"); |
| 119 | + expect(statusSlugEnv).toBeDefined(); |
| 120 | + }); |
| 121 | + |
| 122 | + // 7. Configurable installation URL |
| 123 | + it("uses configurable q_install_url parameter", async () => { |
| 124 | + const customUrlVars = { |
| 125 | + ...requiredVars, |
| 126 | + q_install_url: "https://internal-mirror.company.com/amazon-q" |
| 127 | + }; |
| 128 | + |
| 129 | + const state = await runTerraformApply(moduleDir, customUrlVars); |
| 130 | + |
| 131 | + // Should create the basic resources |
| 132 | + const statusSlugEnv = findResourceInstance(state, "coder_env", "status_slug"); |
| 133 | + expect(statusSlugEnv).toBeDefined(); |
| 134 | + }); |
| 135 | + |
| 136 | + // 8. Version configuration |
| 137 | + it("uses specified versions", async () => { |
| 138 | + const versionVars = { |
| 139 | + ...requiredVars, |
| 140 | + amazon_q_version: "1.14.1", |
| 141 | + agentapi_version: "v0.6.0" |
| 142 | + }; |
| 143 | + |
| 144 | + const state = await runTerraformApply(moduleDir, versionVars); |
| 145 | + |
| 146 | + // Should create the basic resources |
| 147 | + const statusSlugEnv = findResourceInstance(state, "coder_env", "status_slug"); |
| 148 | + expect(statusSlugEnv).toBeDefined(); |
| 149 | + }); |
| 150 | + |
| 151 | + // 9. UI configuration options |
| 152 | + it("supports UI customization options", async () => { |
| 153 | + const uiCustomVars = { |
| 154 | + ...requiredVars, |
| 155 | + order: 5, |
| 156 | + group: "Custom AI Tools", |
| 157 | + icon: "/icon/custom-amazon-q-icon.svg" |
| 158 | + }; |
| 159 | + |
| 160 | + const state = await runTerraformApply(moduleDir, uiCustomVars); |
| 161 | + |
| 162 | + // Should create the basic resources |
| 163 | + const statusSlugEnv = findResourceInstance(state, "coder_env", "status_slug"); |
| 164 | + expect(statusSlugEnv).toBeDefined(); |
| 165 | + }); |
| 166 | + |
| 167 | + // 10. Pre and post install scripts |
| 168 | + it("supports pre and post install scripts", async () => { |
| 169 | + const scriptVars = { |
| 170 | + ...requiredVars, |
| 171 | + pre_install_script: "echo 'Pre-install setup'", |
| 172 | + post_install_script: "echo 'Post-install cleanup'" |
| 173 | + }; |
| 174 | + |
| 175 | + const state = await runTerraformApply(moduleDir, scriptVars); |
| 176 | + |
| 177 | + // Should create the basic resources |
| 178 | + const statusSlugEnv = findResourceInstance(state, "coder_env", "status_slug"); |
| 179 | + expect(statusSlugEnv).toBeDefined(); |
| 180 | + }); |
| 181 | + |
| 182 | + // 11. Valid agent_config JSON (simple structure to avoid parsing issues) |
| 183 | + it("handles valid agent_config JSON", async () => { |
| 184 | + const validAgentConfigVars = { |
| 185 | + ...requiredVars, |
| 186 | + agent_config: JSON.stringify({ |
| 187 | + name: "production-agent", |
| 188 | + description: "Production Amazon Q agent" |
| 189 | + }) |
| 190 | + }; |
| 191 | + |
| 192 | + const state = await runTerraformApply(moduleDir, validAgentConfigVars); |
| 193 | + |
| 194 | + // Should create the basic resources |
| 195 | + const statusSlugEnv = findResourceInstance(state, "coder_env", "status_slug"); |
| 196 | + expect(statusSlugEnv).toBeDefined(); |
| 197 | + }); |
| 198 | + |
| 199 | + // 12. Air-gapped installation support |
| 200 | + it("supports air-gapped installation with custom URL", async () => { |
| 201 | + const airGappedVars = { |
| 202 | + ...requiredVars, |
| 203 | + q_install_url: "https://artifacts.internal.corp/amazon-q-releases", |
| 204 | + amazon_q_version: "1.14.1" |
| 205 | + }; |
| 206 | + |
| 207 | + const state = await runTerraformApply(moduleDir, airGappedVars); |
| 208 | + |
| 209 | + // Should create the basic resources |
| 210 | + const statusSlugEnv = findResourceInstance(state, "coder_env", "status_slug"); |
| 211 | + expect(statusSlugEnv).toBeDefined(); |
| 212 | + }); |
| 213 | + |
| 214 | + // 13. Trust all tools configuration |
| 215 | + it("handles trust_all_tools configuration", async () => { |
| 216 | + const trustVars = { |
| 217 | + ...requiredVars, |
| 218 | + trust_all_tools: true |
| 219 | + }; |
| 220 | + |
| 221 | + const state = await runTerraformApply(moduleDir, trustVars); |
| 222 | + |
| 223 | + // Should create the basic resources |
| 224 | + const statusSlugEnv = findResourceInstance(state, "coder_env", "status_slug"); |
| 225 | + expect(statusSlugEnv).toBeDefined(); |
| 226 | + }); |
| 227 | + |
| 228 | + // 14. AI prompt configuration |
| 229 | + it("handles AI prompt configuration", async () => { |
| 230 | + const promptVars = { |
| 231 | + ...requiredVars, |
| 232 | + ai_prompt: "Create a comprehensive test suite for the application" |
| 233 | + }; |
| 234 | + |
| 235 | + const state = await runTerraformApply(moduleDir, promptVars); |
| 236 | + |
| 237 | + // Should create the basic resources |
| 238 | + const statusSlugEnv = findResourceInstance(state, "coder_env", "status_slug"); |
| 239 | + expect(statusSlugEnv).toBeDefined(); |
| 240 | + }); |
| 241 | + |
| 242 | + // 15. Module directory name configuration |
| 243 | + it("handles module directory name configuration", async () => { |
| 244 | + const dirVars = { |
| 245 | + ...requiredVars, |
| 246 | + module_dir_name: ".custom-amazonq" |
| 247 | + }; |
| 248 | + |
| 249 | + const state = await runTerraformApply(moduleDir, dirVars); |
| 250 | + |
| 251 | + // Should create the basic resources |
| 252 | + const statusSlugEnv = findResourceInstance(state, "coder_env", "status_slug"); |
| 253 | + expect(statusSlugEnv).toBeDefined(); |
| 254 | + }); |
| 255 | +}); |
0 commit comments