check-patroni/tests/test_node_is_primary.py

from click.testing import CliRunner

from check_patroni.cli import main


def test_node_is_primary_ok(runner: CliRunner, fake_restapi) -> None:
    fake_restapi("node_is_primary_ok")
    result = runner.invoke(main, ["-e", "https://10.20.199.3:8008", "node_is_primary"])
    assert result.exit_code == 0
    assert (
        result.stdout
        == "NODEISPRIMARY OK - This node is the primary with the leader lock. | is_primary=1;;@0\n"
    )


def test_node_is_primary_ko(runner: CliRunner, fake_restapi) -> None:
    fake_restapi("node_is_primary_ko", status=503)
    result = runner.invoke(main, ["-e", "https://10.20.199.3:8008", "node_is_primary"])
    assert result.exit_code == 2
    assert (
        result.stdout
        == "NODEISPRIMARY CRITICAL - This node is not the primary with the leader lock. | is_primary=0;;@0\n"
    )
An attempt at correctly ordering imports 2022-02-07 15:11:05 +01:00			`from click.testing import CliRunner`
First working version 2021-08-11 19:09:14 +02:00
An attempt at correctly ordering imports 2022-02-07 15:11:05 +01:00			`from check_patroni.cli import main`
Use isort to automatically sort imports 2023-03-20 11:44:19 +01:00
First working version 2021-08-11 19:09:14 +02:00
Introduce a 'runner' test fixture Instead of defining the CliRunner value in each test, we use a fixture. The CliRunner is also configured with stdout and stderr separated because mixing them will pose problem if we use stderr for other purposes in tests, e.g. to emit log messages from a forth-coming HTTP server. 2023-09-28 09:41:33 +02:00			`def test_node_is_primary_ok(runner: CliRunner, fake_restapi) -> None:`
Introduce a fake_restapi test fixture This fixture itself uses the 'use_old_replica_state' fixture, so that it's no longer needed to use it explicitly in test functions. 2023-09-28 10:57:24 +02:00			`fake_restapi("node_is_primary_ok")`
First working version 2021-08-11 19:09:14 +02:00			`result = runner.invoke(main, ["-e", "https://10.20.199.3:8008", "node_is_primary"])`
			`assert result.exit_code == 0`
Add tests for the output of the script and support pre/post 3.0.4 * Change all replica status from `running` to `streaming` * Add an option to pytest to change the state back to `running` * Also tests the output of the script * Add a quick test script for live clusters 2023-08-22 11:08:38 +02:00			`assert (`
			`result.stdout`
			`== "NODEISPRIMARY OK - This node is the primary with the leader lock. \| is_primary=1;;@0\n"`
			`)`
First working version 2021-08-11 19:09:14 +02:00

Introduce a 'runner' test fixture Instead of defining the CliRunner value in each test, we use a fixture. The CliRunner is also configured with stdout and stderr separated because mixing them will pose problem if we use stderr for other purposes in tests, e.g. to emit log messages from a forth-coming HTTP server. 2023-09-28 09:41:33 +02:00			`def test_node_is_primary_ko(runner: CliRunner, fake_restapi) -> None:`
Introduce a fake_restapi test fixture This fixture itself uses the 'use_old_replica_state' fixture, so that it's no longer needed to use it explicitly in test functions. 2023-09-28 10:57:24 +02:00			`fake_restapi("node_is_primary_ko", status=503)`
First working version 2021-08-11 19:09:14 +02:00			`result = runner.invoke(main, ["-e", "https://10.20.199.3:8008", "node_is_primary"])`
			`assert result.exit_code == 2`
Add tests for the output of the script and support pre/post 3.0.4 * Change all replica status from `running` to `streaming` * Add an option to pytest to change the state back to `running` * Also tests the output of the script * Add a quick test script for live clusters 2023-08-22 11:08:38 +02:00			`assert (`
			`result.stdout`
			`== "NODEISPRIMARY CRITICAL - This node is not the primary with the leader lock. \| is_primary=0;;@0\n"`
			`)`