35 lines
1.1 KiB
Python
35 lines
1.1 KiB
Python
import pytest
|
|
|
|
from cibuildwheel.__main__ import print_new_wheels
|
|
|
|
|
|
def test_printout_wheels(tmp_path, capsys):
|
|
tmp_path.joinpath("example.0").touch()
|
|
with print_new_wheels("TEST_MSG: {n}", tmp_path):
|
|
tmp_path.joinpath("example.1").write_bytes(b"0" * 1023)
|
|
tmp_path.joinpath("example.2").write_bytes(b"0" * 1025)
|
|
|
|
captured = capsys.readouterr()
|
|
assert captured.err == ""
|
|
|
|
assert "example.0" not in captured.out
|
|
assert "example.1 1 kB\n" in captured.out
|
|
assert "example.2 2 kB\n" in captured.out
|
|
assert "TEST_MSG:" in captured.out
|
|
assert "TEST_MSG: 2\n" in captured.out
|
|
|
|
|
|
def test_no_printout_on_error(tmp_path, capsys):
|
|
tmp_path.joinpath("example.0").touch()
|
|
with pytest.raises(RuntimeError), print_new_wheels("TEST_MSG: {n}", tmp_path): # noqa: PT012
|
|
tmp_path.joinpath("example.1").touch()
|
|
raise RuntimeError()
|
|
|
|
captured = capsys.readouterr()
|
|
assert captured.err == ""
|
|
|
|
assert "example.0" not in captured.out
|
|
assert "example.1" not in captured.out
|
|
assert "example.2" not in captured.out
|
|
assert "TEST_MSG:" not in captured.out
|