Ejemplos de main en Python

Ejemplo n.º 1

Archivo: test_update_config.py Proyecto: LinuxHaus/checkmk

def test_main(monkeypatch: pytest.MonkeyPatch) -> None:
    buf = io.StringIO()
    monkeypatch.setattr(sys, "stdout", buf)
    monkeypatch.setattr(update_config.UpdateConfig, "run",
                        lambda self: mock_run())
    assert update_config.main([]) == 0
    assert "XYZ" in buf.getvalue()

Ejemplo n.º 2

Archivo: test_update_config.py Proyecto: tboerger/checkmk

def test_main(monkeypatch):
    buf = io.StringIO()
    monkeypatch.setattr(sys, "stdout", buf)
    monkeypatch.setattr(update_config.UpdateConfig, "run",
                        lambda self: sys.stdout.write("XYZ\n"))
    assert update_config.main([]) == 0
    assert "XYZ" in buf.getvalue()