import os import pytest from moulinette import m18n from moulinette.core import MoulinetteError from moulinette.utils.filesystem import ( append_to_file, read_file, read_json, rm, write_to_file, write_to_json, ) def test_read_file(test_file): content = read_file(str(test_file)) assert content == "foo\nbar\n" def test_read_file_missing_file(): bad_file = "doesnt-exist" with pytest.raises(MoulinetteError) as exception: read_file(bad_file) translation = m18n.g("file_not_exist", path=bad_file) expected_msg = translation.format(path=bad_file) assert expected_msg in str(exception) def test_read_file_cannot_read_ioerror(test_file, mocker): error = "foobar" mocker.patch("builtins.open", side_effect=IOError(error)) with pytest.raises(MoulinetteError) as exception: read_file(str(test_file)) translation = m18n.g("cannot_open_file", file=str(test_file), error=error) expected_msg = translation.format(file=str(test_file), error=error) assert expected_msg in str(exception) def test_read_json(test_json): content = read_json(str(test_json)) assert "foo" in content.keys() assert content["foo"] == "bar" def test_read_json_cannot_read(test_json, mocker): error = "foobar" mocker.patch("json.loads", side_effect=ValueError(error)) with pytest.raises(MoulinetteError) as exception: read_json(str(test_json)) translation = m18n.g("corrupted_json", ressource=str(test_json), error=error) expected_msg = translation.format(ressource=str(test_json), error=error) assert expected_msg in str(exception) def test_write_to_existing_file(test_file): write_to_file(str(test_file), "yolo\nswag") assert read_file(str(test_file)) == "yolo\nswag" def test_write_to_new_file(tmp_path): new_file = tmp_path / "newfile.txt" write_to_file(str(new_file), "yolo\nswag") assert os.path.exists(str(new_file)) assert read_file(str(new_file)) == "yolo\nswag" def test_write_to_existing_file_bad_perms(test_file, mocker): error = "foobar" mocker.patch("builtins.open", side_effect=IOError(error)) with pytest.raises(MoulinetteError) as exception: write_to_file(str(test_file), "yolo\nswag") translation = m18n.g("cannot_write_file", file=str(test_file), error=error) expected_msg = translation.format(file=str(test_file), error=error) assert expected_msg in str(exception) def test_write_cannot_write_folder(tmp_path): with pytest.raises(AssertionError): write_to_file(str(tmp_path), "yolo\nswag") def test_write_cannot_write_to_non_existant_folder(): with pytest.raises(AssertionError): write_to_file("/toto/test", "yolo\nswag") def test_write_to_file_with_a_list(test_file): write_to_file(str(test_file), ["yolo", "swag"]) assert read_file(str(test_file)) == "yolo\nswag" def test_append_to_existing_file(test_file): append_to_file(str(test_file), "yolo\nswag") assert read_file(str(test_file)) == "foo\nbar\nyolo\nswag" def test_append_to_new_file(tmp_path): new_file = tmp_path / "newfile.txt" append_to_file(str(new_file), "yolo\nswag") assert os.path.exists(str(new_file)) assert read_file(str(new_file)) == "yolo\nswag" def text_write_dict_to_json(tmp_path): new_file = tmp_path / "newfile.json" dummy_dict = {"foo": 42, "bar": ["a", "b", "c"]} write_to_json(str(new_file), dummy_dict) _json = read_json(str(new_file)) assert "foo" in _json.keys() assert "bar" in _json.keys() assert _json["foo"] == 42 assert _json["bar"] == ["a", "b", "c"] def text_write_list_to_json(tmp_path): new_file = tmp_path / "newfile.json" dummy_list = ["foo", "bar", "baz"] write_to_json(str(new_file), dummy_list) _json = read_json(str(new_file)) assert _json == ["foo", "bar", "baz"] def test_write_to_json_bad_perms(test_json, mocker): error = "foobar" mocker.patch("builtins.open", side_effect=IOError(error)) with pytest.raises(MoulinetteError) as exception: write_to_json(str(test_json), {"a": 1}) translation = m18n.g("cannot_write_file", file=str(test_json), error=error) expected_msg = translation.format(file=str(test_json), error=error) assert expected_msg in str(exception) def test_write_json_cannot_write_to_non_existant_folder(): with pytest.raises(AssertionError): write_to_json("/toto/test.json", ["a", "b"]) def test_remove_file(test_file): assert os.path.exists(str(test_file)) rm(str(test_file)) assert not os.path.exists(str(test_file)) def test_remove_file_bad_perms(test_file, mocker): error = "foobar" mocker.patch("os.remove", side_effect=OSError(error)) with pytest.raises(MoulinetteError) as exception: rm(str(test_file)) translation = m18n.g("error_removing", path=str(test_file), error=error) expected_msg = translation.format(path=str(test_file), error=error) assert expected_msg in str(exception) def test_remove_directory(tmp_path): test_dir = tmp_path / "foo" test_dir.mkdir() assert os.path.exists(str(test_dir)) rm(str(test_dir), recursive=True) assert not os.path.exists(str(test_dir))