def test_clean_dirs_not_isdir(self, mock_path): mock_path.isdir.return_value = False dirs = ['dir1', 'dir2', 'dir3'] mock_path.join.side_effect = dirs bu.clean_dirs('chroot', dirs) self.assertEqual([mock.call('chroot', d) for d in dirs], mock_path.join.call_args_list)
def test_clean_dirs(self, mock_path, mock_mkdir, mock_rmtree): mock_path.isdir.return_value = True dirs = ['dir1', 'dir2', 'dir3'] mock_path.join.side_effect = dirs bu.clean_dirs('chroot', dirs) for m in (mock_rmtree, mock_mkdir): self.assertEqual([mock.call(d) for d in dirs], m.call_args_list)