Refactor and add tests

Author: matt-richardson

README.md

@@ -1,28 +1,61 @@
 # TeamCityCloudAgentUpdater
 
-When a new TeamCity agent image is built (e.g., via [packer](packer.io)), you need to tell TeamCity to start using it. 
+When a new TeamCity agent image is built (e.g., via [packer](packer.io)), you need to tell TeamCity to start using it.
 
 This is a simple NodeJS app that can:
-1. update TeamCity Cloud Agents images and disable any agents that are running that are based on the old image
-2. remove any agents that were disabled during update that are no longer running a build
+1. Update TeamCity Cloud Agents images and disable any agents that are running that are based on the old image
+2. Remove any agents that were disabled during update that are no longer running a build
 
 ## Usage
 
 When you have a new agent, update the cloud profile:
-```
-node index.js [update-cloud-image] --token XXXXX --server https://teamcity.example.com --image ami-XXXXXXX --cloudprofile "AWS Agents" --agentprefix "Ubuntu" [--dryrun]
+```bash
+node index.js update-cloud-profile --token XXXXX --server https://teamcity.example.com --image ami-XXXXXXX --cloudprofile "AWS Agents" --agentprefix "Ubuntu" [--dryrun]
 ```
 
 To remove any agents that were disabled as part of the update, and are no longer running a build (run on a schedule):
-```
+```bash
 node index.js remove-disabled-agents --token XXXXX --server https://teamcity.example.com [--dryrun]
 ```
 
-The `--dryrun` flag allows you to check what actions the script would have taken with out any real modifications.
+The `--dryrun` flag allows you to check what actions the script would have taken without any real modifications.
+
+## Requirements
+
+- Node.js >= 22.0.0
+- npm >= 11.0.0
+- TeamCity `2019.1` or newer (for user access tokens)
+
+## Development
+
+### Installation
+```bash
+npm install
+```
+
+### Testing
+```bash
+# Run all tests
+npm test
+
+# Run tests in watch mode
+npm run test:watch
 
-## Requiremens
+# Run tests with coverage
+npm run test:coverage
+```
 
-This app uses features (user access tokens) that require TeamCity `2019.1` or newer.
+### Project Structure
+```
+.
+├── index.js                 # Entry point
+├── cli.js                   # CLI command definitions
+├── lib/
+│   ├── agents.js           # Agent management operations
+│   ├── cloud-profiles.js   # Cloud profile management
+│   └── utils.js            # Utility functions
+└── *.test.js               # Test files co-located with source
+```
 
 ## License
 

cli.js

@@ -0,0 +1,42 @@
+'use strict';
+
+const { Command } = require('commander');
+const lib = require('./lib/index');
+
+function createProgram() {
+  const program = new Command();
+
+  program
+    .name('TeamCity Cloud Agent Updater')
+    .description('Simple NodeJS app to update images for TeamCity Cloud Agents, via the 2017.1 API.')
+    .version('1.0.0');
+
+  program
+    .command('update-cloud-profile', { isDefault: true })
+    .requiredOption('--token <string>', 'A valid TeamCity user access token (requires TC 2019.1)')
+    .requiredOption('--server <string>', 'The url of the TeamCity server, eg "http://teamcity.example.com"')
+    .requiredOption('--image <string>', 'The AMI id (for AWS), or full url to the VHD / resource id of the managed image (for Azure)')
+    .requiredOption('--cloudprofile <string>', 'The name of the TeamCity Cloud Profile to modify')
+    .requiredOption('--agentprefix <string>', 'The agent prefix used in the Cloud Profile image that should be updated')
+    .option('--dryrun', 'Output what changes the app would make, but dont actually make the changes')
+    .action((options) => lib.updateCloudImage(options.server, "Bearer " + options.token, options.cloudprofile, options.agentprefix, options.image, options.dryrun));
+
+  program
+    .command('remove-disabled-agents')
+    .requiredOption('--token <string>', 'A valid TeamCity user access token (requires TC 2019.1)')
+    .requiredOption('--server <string>', 'The url of the TeamCity server, eg "http://teamcity.example.com"')
+    .option('--dryrun', 'Output what changes the app would make, but dont actually make the changes')
+    .action((options) => lib.removeDisabledAgents(options.server, "Bearer " + options.token, options.dryrun));
+
+  return program;
+}
+
+function run(argv) {
+  const program = createProgram();
+  return program.parse(argv);
+}
+
+module.exports = {
+  createProgram,
+  run
+};

cli.test.js

@@ -0,0 +1,107 @@
+const { spawn } = require('child_process');
+const path = require('path');
+const cli = require('./cli');
+
+describe('CLI Commands', () => {
+  const indexPath = path.join(__dirname, 'index.js');
+
+  const runCommand = (args) => {
+    return new Promise((resolve, reject) => {
+      const process = spawn('node', [indexPath, ...args]);
+      let stdout = '';
+      let stderr = '';
+
+      process.stdout.on('data', (data) => {
+        stdout += data.toString();
+      });
+
+      process.stderr.on('data', (data) => {
+        stderr += data.toString();
+      });
+
+      process.on('close', (code) => {
+        resolve({ code, stdout, stderr });
+      });
+
+      process.on('error', (error) => {
+        reject(error);
+      });
+    });
+  };
+
+  describe('update-cloud-profile command', () => {
+    it('should show error when required options are missing', async () => {
+      const result = await runCommand(['update-cloud-profile']);
+      expect(result.code).not.toBe(0);
+      expect(result.stderr).toContain('required option');
+    });
+
+    it('should show help with --help flag', async () => {
+      const result = await runCommand(['update-cloud-profile', '--help']);
+      expect(result.code).toBe(0);
+      expect(result.stdout).toContain('--token');
+      expect(result.stdout).toContain('--server');
+      expect(result.stdout).toContain('--image');
+      expect(result.stdout).toContain('--cloudprofile');
+      expect(result.stdout).toContain('--agentprefix');
+      expect(result.stdout).toContain('--dryrun');
+    });
+
+    it('should validate required parameters', async () => {
+      const result = await runCommand([
+        'update-cloud-profile',
+        '--token', 'test-token'
+      ]);
+      expect(result.code).not.toBe(0);
+      expect(result.stderr).toContain('required option');
+    });
+  });
+
+  describe('remove-disabled-agents command', () => {
+    it('should show error when required options are missing', async () => {
+      const result = await runCommand(['remove-disabled-agents']);
+      expect(result.code).not.toBe(0);
+      expect(result.stderr).toContain('required option');
+    });
+
+    it('should show help with --help flag', async () => {
+      const result = await runCommand(['remove-disabled-agents', '--help']);
+      expect(result.code).toBe(0);
+      expect(result.stdout).toContain('--token');
+      expect(result.stdout).toContain('--server');
+      expect(result.stdout).toContain('--dryrun');
+    });
+
+    it('should validate required parameters', async () => {
+      const result = await runCommand([
+        'remove-disabled-agents',
+        '--token', 'test-token'
+      ]);
+      expect(result.code).not.toBe(0);
+      expect(result.stderr).toContain('required option');
+    });
+  });
+
+  describe('General CLI', () => {
+    it('should show version with --version flag', async () => {
+      const result = await runCommand(['--version']);
+      expect(result.code).toBe(0);
+      expect(result.stdout).toContain('1.0.0');
+    });
+
+    it('should show help with --help flag', async () => {
+      const result = await runCommand(['--help']);
+      expect(result.code).toBe(0);
+      expect(result.stdout).toContain('TeamCity Cloud Agent Updater');
+      expect(result.stdout).toContain('update-cloud-profile');
+      expect(result.stdout).toContain('remove-disabled-agents');
+    });
+
+    it('should use update-cloud-profile as default command', async () => {
+      const result = await runCommand([]);
+      expect(result.code).not.toBe(0);
+      // Should show error for missing required options from update-cloud-profile
+      expect(result.stderr).toContain('required option');
+    });
+  });
+});