def test_repr(monkeypatch, tmpdir):
t = "18:01:46 up 62 days, 18:27, 1 user, load average: 0.09, 0.04, 0.05"
monkeypatch.setattr(
ssh, "run", lambda x, combine_stderr, timeout: get_mock_ssh_text(t, 0))
ssh_ctx = ssh.SshContext("ubuntu", get_mock_key_file(tmpdir))
(ssh.SshCommandValidation(ssh_ctx, "name", "cmd", hosts=hosts).__repr__())
def test_generate_id_consistency_ssh(tmpdir):
pub = PagerDutyPublisher("url", "token")
ssh_ctx = ssh.SshContext("ubuntu", ssh_key_file(tmpdir))
ssh_ctx2 = ssh.SshContext("ubuntu", ssh_key_file(tmpdir))
v = ssh.SshCommandValidation(ssh_ctx, "name", "cmd", hosts=["a fake host"])
v2 = ssh.SshCommandValidation(ssh_ctx2,
"name",
"cmd",
hosts=["a fake host"])
failure = Failure("bar", v, "unable to transmogrify")
another = Failure("foo", v2, "to transmogrify")
assert pub._generate_id(failure) == pub._generate_id(another)
def test_ssh_expected_return_code(monkeypatch, tmpdir):
t = "18:01:46 up 62 days, 18:27, 1 user, load average: 0.09, 0.04, 0.05"
monkeypatch.setattr(
ssh, "run", lambda x, combine_stderr, timeout: get_mock_ssh_text(t, 1))
ssh_ctx = ssh.SshContext("ubuntu", get_mock_key_file(tmpdir))
(ssh.SshCommandValidation(ssh_ctx, "name", "cmd",
hosts=hosts).expect_exit_code(1).perform({}))
def test_exit_code_equals(monkeypatch, tmpdir):
t = "stopped/waiting"
monkeypatch.setattr(
ssh, "run", lambda x, combine_stderr, timeout: get_mock_ssh_text(t, 0))
ssh_ctx = ssh.SshContext("ubuntu", get_mock_key_file(tmpdir))
(ssh.SshCommandValidation(ssh_ctx, "citations", "command",
hosts=hosts).expect_exit_code(0).perform({}))
def test_ssh_expected_0_by_default(monkeypatch, tmpdir):
t = "18:01:46 up 62 days, 18:27, 1 user, load average: 0.09, 0.04, 0.05"
monkeypatch.setattr(
ssh, "run", lambda x, combine_stderr, timeout: get_mock_ssh_text(t, 1))
ssh_ctx = ssh.SshContext("ubuntu", get_mock_key_file(tmpdir))
with pytest.raises(ValidationFailure):
(ssh.SshCommandValidation(ssh_ctx, "name", "cmd",
hosts=hosts).perform({}))
def test_output_does_not_contain_correctly_fails(monkeypatch, tmpdir):
t = "stopped/waiting"
monkeypatch.setattr(
ssh, "run", lambda x, combine_stderr, timeout: get_mock_ssh_text(t, 0))
ssh_ctx = ssh.SshContext("ubuntu", get_mock_key_file(tmpdir))
with pytest.raises(ValidationFailure):
(ssh.SshCommandValidation(
ssh_ctx, "citations", "command",
hosts=hosts).expect_output_does_not_contain("stopped").perform({}))
def test_output_correct_on_ssh_failure(monkeypatch, tmpdir):
monkeypatch.setattr(ssh, "run",
lambda x, combine_stderr, timeout: broken_ssh())
ssh_ctx = ssh.SshContext("ubuntu", get_mock_key_file(tmpdir))
with pytest.raises(ValidationFailure):
validation = ssh.SshCommandValidation(ssh_ctx,
"citations",
"command",
hosts=hosts)
validation.add_expectation(ssh.OutputLessThan(validation, 90))
validation.perform({})
def test_output_less_than(monkeypatch, tmpdir):
t = "100"
monkeypatch.setattr(
ssh, "run", lambda x, combine_stderr, timeout: get_mock_ssh_text(t, 0))
ssh_ctx = ssh.SshContext("ubuntu", get_mock_key_file(tmpdir))
validation = ssh.SshCommandValidation(ssh_ctx,
"citations",
"command",
hosts=hosts)
validation.add_expectation(ssh.OutputLessThan(validation, 110))
validation.perform({})
def test_output_greater_than_correctly_fails(monkeypatch, tmpdir):
t = "100"
monkeypatch.setattr(
ssh, "run", lambda x, combine_stderr, timeout: get_mock_ssh_text(t, 0))
ssh_ctx = ssh.SshContext("ubuntu", get_mock_key_file(tmpdir))
with pytest.raises(ValidationFailure):
validation = ssh.SshCommandValidation(ssh_ctx,
"citations",
"command",
hosts=hosts)
validation.add_expectation(ssh.OutputGreaterThan(validation, 110))
validation.perform({})