2022-02-07 15:11:05 +01:00
|
|
|
from click.testing import CliRunner
|
2023-03-20 11:44:19 +01:00
|
|
|
from pytest_mock import MockerFixture
|
2021-08-11 19:09:14 +02:00
|
|
|
|
2022-02-07 15:11:05 +01:00
|
|
|
from check_patroni.cli import main
|
2023-03-20 11:44:19 +01:00
|
|
|
|
|
|
|
from .tools import my_mock
|
2021-08-11 19:09:14 +02:00
|
|
|
|
|
|
|
|
2023-08-22 11:08:38 +02:00
|
|
|
def test_cluster_node_count_ok(
|
|
|
|
mocker: MockerFixture, use_old_replica_state: bool
|
|
|
|
) -> None:
|
2021-08-11 19:09:14 +02:00
|
|
|
runner = CliRunner()
|
|
|
|
|
2023-09-28 09:56:08 +02:00
|
|
|
my_mock(
|
|
|
|
mocker, "cluster_node_count_ok", use_old_replica_state=use_old_replica_state
|
|
|
|
)
|
2021-08-11 19:09:14 +02:00
|
|
|
result = runner.invoke(
|
|
|
|
main, ["-e", "https://10.20.199.3:8008", "cluster_node_count"]
|
|
|
|
)
|
|
|
|
assert result.exit_code == 0
|
2023-08-22 11:08:38 +02:00
|
|
|
if use_old_replica_state:
|
|
|
|
assert (
|
|
|
|
result.output
|
|
|
|
== "CLUSTERNODECOUNT OK - members is 3 | healthy_members=3 members=3 role_leader=1 role_replica=2 state_running=3\n"
|
|
|
|
)
|
|
|
|
else:
|
|
|
|
assert (
|
|
|
|
result.output
|
|
|
|
== "CLUSTERNODECOUNT OK - members is 3 | healthy_members=3 members=3 role_leader=1 role_replica=2 state_running=1 state_streaming=2\n"
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
|
|
def test_cluster_node_count_ok_with_thresholds(
|
|
|
|
mocker: MockerFixture, use_old_replica_state: bool
|
|
|
|
) -> None:
|
2021-08-11 19:09:14 +02:00
|
|
|
runner = CliRunner()
|
|
|
|
|
2023-09-28 09:56:08 +02:00
|
|
|
my_mock(
|
|
|
|
mocker, "cluster_node_count_ok", use_old_replica_state=use_old_replica_state
|
|
|
|
)
|
2021-08-11 19:09:14 +02:00
|
|
|
result = runner.invoke(
|
|
|
|
main,
|
|
|
|
[
|
|
|
|
"-e",
|
|
|
|
"https://10.20.199.3:8008",
|
|
|
|
"cluster_node_count",
|
|
|
|
"--warning",
|
|
|
|
"@0:1",
|
|
|
|
"--critical",
|
|
|
|
"@2",
|
2023-08-21 10:45:02 +02:00
|
|
|
"--healthy-warning",
|
2021-08-11 19:09:14 +02:00
|
|
|
"@2",
|
2023-08-21 10:45:02 +02:00
|
|
|
"--healthy-critical",
|
2021-08-11 19:09:14 +02:00
|
|
|
"@0:1",
|
|
|
|
],
|
|
|
|
)
|
|
|
|
assert result.exit_code == 0
|
2023-08-22 11:08:38 +02:00
|
|
|
if use_old_replica_state:
|
|
|
|
assert (
|
|
|
|
result.output
|
|
|
|
== "CLUSTERNODECOUNT OK - members is 3 | healthy_members=3;@2;@1 members=3;@1;@2 role_leader=1 role_replica=2 state_running=3\n"
|
|
|
|
)
|
|
|
|
else:
|
|
|
|
assert (
|
|
|
|
result.output
|
|
|
|
== "CLUSTERNODECOUNT OK - members is 3 | healthy_members=3;@2;@1 members=3;@1;@2 role_leader=1 role_replica=2 state_running=1 state_streaming=2\n"
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
|
|
def test_cluster_node_count_healthy_warning(
|
|
|
|
mocker: MockerFixture, use_old_replica_state: bool
|
|
|
|
) -> None:
|
2021-08-11 19:09:14 +02:00
|
|
|
runner = CliRunner()
|
|
|
|
|
2023-09-28 09:56:08 +02:00
|
|
|
my_mock(
|
|
|
|
mocker,
|
|
|
|
"cluster_node_count_healthy_warning",
|
|
|
|
use_old_replica_state=use_old_replica_state,
|
|
|
|
)
|
2021-08-11 19:09:14 +02:00
|
|
|
result = runner.invoke(
|
|
|
|
main,
|
|
|
|
[
|
|
|
|
"-e",
|
|
|
|
"https://10.20.199.3:8008",
|
|
|
|
"cluster_node_count",
|
2023-08-21 10:45:02 +02:00
|
|
|
"--healthy-warning",
|
2021-08-11 19:09:14 +02:00
|
|
|
"@2",
|
2023-08-21 10:45:02 +02:00
|
|
|
"--healthy-critical",
|
2021-08-11 19:09:14 +02:00
|
|
|
"@0:1",
|
|
|
|
],
|
|
|
|
)
|
|
|
|
assert result.exit_code == 1
|
2023-08-22 11:08:38 +02:00
|
|
|
if use_old_replica_state:
|
|
|
|
assert (
|
|
|
|
result.output
|
|
|
|
== "CLUSTERNODECOUNT WARNING - healthy_members is 2 (outside range @0:2) | healthy_members=2;@2;@1 members=2 role_leader=1 role_replica=1 state_running=2\n"
|
|
|
|
)
|
|
|
|
else:
|
|
|
|
assert (
|
|
|
|
result.output
|
|
|
|
== "CLUSTERNODECOUNT WARNING - healthy_members is 2 (outside range @0:2) | healthy_members=2;@2;@1 members=2 role_leader=1 role_replica=1 state_running=1 state_streaming=1\n"
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
|
|
def test_cluster_node_count_healthy_critical(
|
|
|
|
mocker: MockerFixture, use_old_replica_state: bool
|
|
|
|
) -> None:
|
2021-08-11 19:09:14 +02:00
|
|
|
runner = CliRunner()
|
|
|
|
|
2023-09-28 09:56:08 +02:00
|
|
|
my_mock(
|
|
|
|
mocker,
|
|
|
|
"cluster_node_count_healthy_critical",
|
|
|
|
use_old_replica_state=use_old_replica_state,
|
|
|
|
)
|
2021-08-11 19:09:14 +02:00
|
|
|
result = runner.invoke(
|
|
|
|
main,
|
|
|
|
[
|
|
|
|
"-e",
|
|
|
|
"https://10.20.199.3:8008",
|
|
|
|
"cluster_node_count",
|
2023-08-21 10:45:02 +02:00
|
|
|
"--healthy-warning",
|
2021-08-11 19:09:14 +02:00
|
|
|
"@2",
|
2023-08-21 10:45:02 +02:00
|
|
|
"--healthy-critical",
|
2021-08-11 19:09:14 +02:00
|
|
|
"@0:1",
|
|
|
|
],
|
|
|
|
)
|
|
|
|
assert result.exit_code == 2
|
2023-08-22 11:08:38 +02:00
|
|
|
assert (
|
|
|
|
result.output
|
|
|
|
== "CLUSTERNODECOUNT CRITICAL - healthy_members is 1 (outside range @0:1) | healthy_members=1;@2;@1 members=3 role_leader=1 role_replica=2 state_running=1 state_start_failed=2\n"
|
|
|
|
)
|
2021-08-11 19:09:14 +02:00
|
|
|
|
|
|
|
|
2023-08-22 11:08:38 +02:00
|
|
|
def test_cluster_node_count_warning(
|
|
|
|
mocker: MockerFixture, use_old_replica_state: bool
|
|
|
|
) -> None:
|
2021-08-11 19:09:14 +02:00
|
|
|
runner = CliRunner()
|
|
|
|
|
2023-09-28 09:56:08 +02:00
|
|
|
my_mock(
|
|
|
|
mocker,
|
|
|
|
"cluster_node_count_warning",
|
|
|
|
use_old_replica_state=use_old_replica_state,
|
|
|
|
)
|
2021-08-11 19:09:14 +02:00
|
|
|
result = runner.invoke(
|
|
|
|
main,
|
|
|
|
[
|
|
|
|
"-e",
|
|
|
|
"https://10.20.199.3:8008",
|
|
|
|
"cluster_node_count",
|
|
|
|
"--warning",
|
|
|
|
"@2",
|
|
|
|
"--critical",
|
|
|
|
"@0:1",
|
|
|
|
],
|
|
|
|
)
|
|
|
|
assert result.exit_code == 1
|
2023-08-22 11:08:38 +02:00
|
|
|
if use_old_replica_state:
|
|
|
|
assert (
|
|
|
|
result.stdout
|
|
|
|
== "CLUSTERNODECOUNT WARNING - members is 2 (outside range @0:2) | healthy_members=2 members=2;@2;@1 role_leader=1 role_replica=1 state_running=2\n"
|
|
|
|
)
|
|
|
|
else:
|
|
|
|
assert (
|
|
|
|
result.stdout
|
|
|
|
== "CLUSTERNODECOUNT WARNING - members is 2 (outside range @0:2) | healthy_members=2 members=2;@2;@1 role_leader=1 role_replica=1 state_running=1 state_streaming=1\n"
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
|
|
def test_cluster_node_count_critical(
|
|
|
|
mocker: MockerFixture, use_old_replica_state: bool
|
|
|
|
) -> None:
|
2021-08-11 19:09:14 +02:00
|
|
|
runner = CliRunner()
|
|
|
|
|
2023-09-28 09:56:08 +02:00
|
|
|
my_mock(
|
|
|
|
mocker,
|
|
|
|
"cluster_node_count_critical",
|
|
|
|
use_old_replica_state=use_old_replica_state,
|
|
|
|
)
|
2021-08-11 19:09:14 +02:00
|
|
|
result = runner.invoke(
|
|
|
|
main,
|
|
|
|
[
|
|
|
|
"-e",
|
|
|
|
"https://10.20.199.3:8008",
|
|
|
|
"cluster_node_count",
|
|
|
|
"--warning",
|
|
|
|
"@2",
|
|
|
|
"--critical",
|
|
|
|
"@0:1",
|
|
|
|
],
|
|
|
|
)
|
|
|
|
assert result.exit_code == 2
|
2023-08-22 11:08:38 +02:00
|
|
|
assert (
|
|
|
|
result.stdout
|
|
|
|
== "CLUSTERNODECOUNT CRITICAL - members is 1 (outside range @0:1) | healthy_members=1 members=1;@2;@1 role_leader=1 state_running=1\n"
|
|
|
|
)
|