Backup_Repos/beets
1
0
Fork 0
mirror of https://github.com/beetbox/beets.git synced 2026-01-09 01:15:38 +01:00
Code Issues Projects Releases Wiki Activity

switch to control_stdin

Browse source
This commit is contained in:
alicezou 2022-03-29 20:04:56 -04:00
parent a0b0028874
commit 67e778fec6
1 changed files with 7 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
test/test_dbcore.py
View file

@ -23,6 +23,7 @@ from random import random
from unittest import mock
from test import _common
from test.helper import control_stdin
from beets import dbcore
from tempfile import mkstemp
@ -763,19 +764,19 @@ class ResultsIteratorTest(unittest.TestCase):
class ParentalDirCreation(_common.TestCase):
@mock.patch('builtins.input', side_effect=['y', ])
def test_create_yes(self, _):
def test_create_yes(self):
non_exist_path = _common.util.py3_path(os.path.join(
self.temp_dir, b'nonexist', str(random()).encode()))
dbcore.Database(non_exist_path)
with control_stdin('y'):
dbcore.Database(non_exist_path)
@mock.patch('builtins.input', side_effect=['n', ])
def test_create_no(self, _):
def test_create_no(self):
non_exist_path_parent = _common.util.py3_path(
os.path.join(self.temp_dir, b'nonexist'))
non_exist_path = _common.util.py3_path(os.path.join(
non_exist_path_parent.encode(), str(random()).encode()))
dbcore.Database(non_exist_path)
with control_stdin('n'):
dbcore.Database(non_exist_path)
if os.path.exists(non_exist_path_parent):
shutil.rmtree(non_exist_path_parent)
raise OSError("Should not create dir")