87 lines
3.1 KiB
C#
87 lines
3.1 KiB
C#
|
using w4c_workflows.Models;
|
||
|
|
using w4c_workflows.Services.Execution;
|
||
|
|
using Xunit;
|
||
|
|
|
||
|
|
namespace w4c_workflows.Tests;
|
||
|
|
|
||
|
|
public class TaskRunMessageTests
|
||
|
|
{
|
||
|
|
private static WorkflowRun NewRun() => new()
|
||
|
|
{
|
||
|
|
Id = Guid.Parse("11111111-1111-1111-1111-111111111111"),
|
||
|
|
WorkflowId = Guid.NewGuid(),
|
||
|
|
TenantId = "t1",
|
||
|
|
Status = RunStatus.Running,
|
||
|
|
};
|
||
|
|
|
||
|
|
private static WorkflowTask NewTask() => new()
|
||
|
|
{
|
||
|
|
Id = Guid.Parse("22222222-2222-2222-2222-222222222222"),
|
||
|
|
WorkflowId = Guid.NewGuid(),
|
||
|
|
Key = "root",
|
||
|
|
Language = "shell",
|
||
|
|
Mode = WorkflowMode.Function,
|
||
|
|
EntryJson = "{\"file\":\"root.sh\",\"function\":\"main\"}",
|
||
|
|
EnvJson = "{\"FOO\":\"bar\"}",
|
||
|
|
};
|
||
|
|
|
||
|
|
[Fact]
|
||
|
|
public void Serializes_fields_the_worker_can_parse_back()
|
||
|
|
{
|
||
|
|
var fields = TaskRunMessage.ToFields(
|
||
|
|
NewRun(), NewTask(), "{\"x\":1}", "/repo/workflows", 2);
|
||
|
|
|
||
|
|
var invocation = TaskInvocation.FromFields(fields);
|
||
|
|
|
||
|
|
Assert.Equal(TaskInvocation.TypeValue, invocation.Type);
|
||
|
|
Assert.Equal("11111111-1111-1111-1111-111111111111", invocation.RunId);
|
||
|
|
Assert.Equal("22222222-2222-2222-2222-222222222222", invocation.TaskId);
|
||
|
|
Assert.Equal("root", invocation.TaskKey);
|
||
|
|
Assert.Equal("shell", invocation.Language);
|
||
|
|
Assert.Equal("root.sh", invocation.EntryFile);
|
||
|
|
Assert.Equal("main", invocation.EntryFunction);
|
||
|
|
Assert.Equal("{\"x\":1}", invocation.Input);
|
||
|
|
Assert.Equal("/repo/workflows", invocation.WorkingDir);
|
||
|
|
Assert.Equal("t1", invocation.TenantId);
|
||
|
|
Assert.Equal(2, invocation.Attempt);
|
||
|
|
Assert.Equal("bar", invocation.Env["FOO"]);
|
||
|
|
}
|
||
|
|
|
||
|
|
[Fact]
|
||
|
|
public void Omits_optional_fields_when_absent()
|
||
|
|
{
|
||
|
|
var task = NewTask();
|
||
|
|
task.EntryJson = "{\"file\":\"a.sh\"}"; // no function
|
||
|
|
task.EnvJson = null;
|
||
|
|
|
||
|
|
var fields = TaskRunMessage.ToFields(NewRun(), task, null, null, 1);
|
||
|
|
var invocation = TaskInvocation.FromFields(fields);
|
||
|
|
|
||
|
|
Assert.Null(invocation.EntryFunction);
|
||
|
|
Assert.Null(invocation.Input);
|
||
|
|
Assert.Null(invocation.WorkingDir);
|
||
|
|
Assert.Empty(invocation.Env);
|
||
|
|
Assert.Equal(1, invocation.Attempt);
|
||
|
|
}
|
||
|
|
|
||
|
|
[Fact]
|
||
|
|
public void Carries_server_reference_when_set_and_omits_when_absent()
|
||
|
|
{
|
||
|
|
// S8: a task targeting a managed server serializes the server id through the
|
||
|
|
// wire contract so the worker can execute it remotely instead of locally.
|
||
|
|
var withServer = NewTask();
|
||
|
|
withServer.Server = "srv-07";
|
||
|
|
var fields = TaskRunMessage.ToFields(NewRun(), withServer, null, null, 1);
|
||
|
|
|
||
|
|
Assert.Equal("srv-07", fields["server"]);
|
||
|
|
var invocation = TaskInvocation.FromFields(fields);
|
||
|
|
Assert.Equal("srv-07", invocation.Server);
|
||
|
|
|
||
|
|
// No server field => local subprocess path (the default) => null.
|
||
|
|
var withoutServer = NewTask();
|
||
|
|
var bareFields = TaskRunMessage.ToFields(NewRun(), withoutServer, null, null, 1);
|
||
|
|
Assert.False(bareFields.ContainsKey("server"));
|
||
|
|
Assert.Null(TaskInvocation.FromFields(bareFields).Server);
|
||
|
|
}
|
||
|
|
}
|