From 5115496659e6129ef72bea5f699377df7d53ffc0 Mon Sep 17 00:00:00 2001 From: Nik Kantar Date: Fri, 23 Oct 2020 20:09:32 -0700 Subject: [PATCH] Add ensure_dir tests --- CHANGELOG.md | 3 +++ tests/test_ensure_dir.py | 24 ++++++++++++++++++++++++ 2 files changed, 27 insertions(+) create mode 100644 tests/test_ensure_dir.py diff --git a/CHANGELOG.md b/CHANGELOG.md index b0953f7..740b835 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -12,6 +12,9 @@ ## [Unreleased] +### Aded +- `_ensure_dir` tests + ## [1.1.0] - 2020-09-30 diff --git a/tests/test_ensure_dir.py b/tests/test_ensure_dir.py new file mode 100644 index 0000000..9f44050 --- /dev/null +++ b/tests/test_ensure_dir.py @@ -0,0 +1,24 @@ +from pathlib import Path + +from src import sus +from src.sus import _ensure_dir + + +def test_ensure_dir_not_found(mocker): + mock_rmtree = mocker.patch("src.sus.rmtree") + mock_path_mkdir = mocker.patch.object(sus.Path, "mkdir") + + _ensure_dir(Path("foo")) + + mock_rmtree.assert_called_once_with(Path("foo")) + mock_path_mkdir.assert_called_once() + + +def test_ensure_dir_exists(mocker): + mock_rmtree = mocker.patch("src.sus.rmtree", side_effect=FileNotFoundError()) + mock_path_mkdir = mocker.patch.object(sus.Path, "mkdir") + + _ensure_dir(Path("foo")) + + mock_rmtree.assert_called_once_with(Path("foo")) + mock_path_mkdir.assert_called_once()