94 lines
2.6 KiB
Elixir
94 lines
2.6 KiB
Elixir
defmodule ServerWeb.HostChannelTest do
|
|
use ServerWeb.ChannelCase, async: false
|
|
|
|
alias Server.Hosts
|
|
alias ServerWeb.AgentSocket
|
|
|
|
setup do
|
|
{:ok, {host, token}} = Hosts.create_host("pve-01")
|
|
%{host: host, token: token}
|
|
end
|
|
|
|
describe "join" do
|
|
test "succeeds with valid token and marks host online", %{host: host, token: token} do
|
|
{:ok, socket} = connect(AgentSocket, %{})
|
|
|
|
assert {:ok, _reply, socket} =
|
|
subscribe_and_join(socket, "host:pve-01", %{
|
|
"token" => token,
|
|
"agent_version" => "0.1.0"
|
|
})
|
|
|
|
assert socket.assigns.host_id == host.id
|
|
|
|
reloaded = Server.Repo.reload!(host)
|
|
assert reloaded.status == "online"
|
|
assert reloaded.agent_version == "0.1.0"
|
|
assert reloaded.last_seen_at != nil
|
|
end
|
|
|
|
test "rejects invalid token" do
|
|
{:ok, socket} = connect(AgentSocket, %{})
|
|
|
|
assert {:error, %{reason: "invalid_token"}} =
|
|
subscribe_and_join(socket, "host:pve-01", %{
|
|
"token" => "garbage",
|
|
"agent_version" => "0.1.0"
|
|
})
|
|
end
|
|
|
|
test "rejects unknown host name" do
|
|
{:ok, socket} = connect(AgentSocket, %{})
|
|
|
|
assert {:error, %{reason: "unknown_host"}} =
|
|
subscribe_and_join(socket, "host:nope", %{
|
|
"token" => "x",
|
|
"agent_version" => "0.1.0"
|
|
})
|
|
end
|
|
end
|
|
|
|
describe "metric:fast event" do
|
|
setup %{token: token} do
|
|
{:ok, socket} = connect(AgentSocket, %{})
|
|
|
|
{:ok, _reply, joined} =
|
|
subscribe_and_join(socket, "host:pve-01", %{
|
|
"token" => token,
|
|
"agent_version" => "0.1.0"
|
|
})
|
|
|
|
%{socket: joined}
|
|
end
|
|
|
|
test "accepts metric payload and replies :ok", %{socket: socket} do
|
|
ref =
|
|
push(socket, "metric:fast", %{
|
|
"collected_at" => "2026-04-21T12:00:00Z",
|
|
"data" => %{"cpu_percent" => 12.3, "load1" => 0.2}
|
|
})
|
|
|
|
assert_reply ref, :ok
|
|
end
|
|
end
|
|
|
|
describe "terminate" do
|
|
test "marks host offline when channel process exits", %{host: host, token: token} do
|
|
{:ok, socket} = connect(AgentSocket, %{})
|
|
|
|
{:ok, _, joined} =
|
|
subscribe_and_join(socket, "host:pve-01", %{
|
|
"token" => token,
|
|
"agent_version" => "0.1.0"
|
|
})
|
|
|
|
Process.unlink(joined.channel_pid)
|
|
ref = Process.monitor(joined.channel_pid)
|
|
close(joined)
|
|
assert_receive {:DOWN, ^ref, :process, _, _}, 1_000
|
|
|
|
reloaded = Server.Repo.reload!(host)
|
|
assert reloaded.status == "offline"
|
|
end
|
|
end
|
|
end
|