using System.CommandLine; using System.Net; using System.Net.Sockets; using System.Text; using System.Text.Json; using ZB.MOM.WW.ScadaBridge.CLI.Commands; namespace ZB.MOM.WW.ScadaBridge.CLI.Tests; /// /// deploy area (Gitea #40) sends MgmtDeployArea with the area id and /// force flag, and derives its exit code from the summary's failureCount /// exactly like deploy site: a per-instance failure is a successful HTTP /// call, so without that a script could not detect it. /// public class DeployAreaExitCodeTests { /// Verifies the wire payload and the exit code derived from the summary. /// The JSON response body served by the fake management endpoint. /// Whether --force is passed. /// The expected process exit code. /// A task that represents the asynchronous operation. [Theory] [InlineData("{\"successCount\":2,\"failureCount\":0}", false, 0)] [InlineData("{\"successCount\":1,\"failureCount\":1}", true, 1)] [InlineData("{\"unexpected\":true}", false, 1)] public async Task DeployArea_SendsAreaCommand_AndReturnsDomainExitCode(string response, bool force, int expected) { var listener = new TcpListener(IPAddress.Loopback, 0); listener.Start(); try { var port = ((IPEndPoint)listener.LocalEndpoint).Port; string? requestBody = null; var server = Task.Run(async () => { using var client = await listener.AcceptTcpClientAsync(); await using var stream = client.GetStream(); using var reader = new StreamReader(stream, Encoding.UTF8, leaveOpen: true); var length = 0; while (await reader.ReadLineAsync() is { Length: > 0 } line) if (line.StartsWith("Content-Length:", StringComparison.OrdinalIgnoreCase)) length = int.Parse(line[15..].Trim()); var buffer = new char[length]; var read = 0; while (read < length) { var count = await reader.ReadAsync(buffer.AsMemory(read)); if (count == 0) throw new EndOfStreamException(); read += count; } requestBody = new string(buffer); var body = Encoding.UTF8.GetBytes(response); var headers = Encoding.ASCII.GetBytes($"HTTP/1.1 200 OK\r\nContent-Type: application/json\r\nContent-Length: {body.Length}\r\nConnection: close\r\n\r\n"); await stream.WriteAsync(headers); await stream.WriteAsync(body); }); var root = new RootCommand(); var url = new Option("--url") { Recursive = true }; var format = new Option("--format") { Recursive = true }; var user = new Option("--username") { Recursive = true }; var password = new Option("--password") { Recursive = true }; root.Add(url); root.Add(format); root.Add(user); root.Add(password); root.Add(DeployCommands.Build(url, format, user, password)); string[] args = ["deploy", "area", "--id", "12", "--url", $"http://127.0.0.1:{port}", "--username", "test", "--password", "test", "--format", "json"]; if (force) args = [.. args, "--force"]; var exit = await root.Parse(args).InvokeAsync(); await server.WaitAsync(TimeSpan.FromSeconds(5)); Assert.Equal(expected, exit); using var payload = JsonDocument.Parse(requestBody!); Assert.Equal("MgmtDeployArea", payload.RootElement.GetProperty("command").GetString()); var command = payload.RootElement.GetProperty("payload"); Assert.Equal(12, GetPropertyIgnoreCase(command, "areaId").GetInt32()); Assert.Equal(force, GetPropertyIgnoreCase(command, "force").GetBoolean()); } finally { listener.Stop(); } } private static JsonElement GetPropertyIgnoreCase(JsonElement element, string name) => element.EnumerateObject().First(p => string.Equals(p.Name, name, StringComparison.OrdinalIgnoreCase)).Value; }