mirror of
https://github.com/beetbox/beets.git
synced 2025-12-06 16:42:42 +01:00
a59e41a883
Moved tests related to ui into own folder. Moved 'modify' command tests into own file. Moved 'write' command tests into own file. Moved 'fields' command tests into own file. Moved 'do_query' test into own file. Moved 'list' command tests into own file. Moved 'remove' command tests into own file. Moved 'move' command tests into own file. Moved 'update' command tests into own file. Moved 'show_change' test into test_import file. Moved 'summarize_items' test into test_import file. Moved 'completion' command test into own file.
121 lines
3.9 KiB
Python
121 lines
3.9 KiB
Python
# This file is part of beets.
|
|
# Copyright 2016, Adrian Sampson.
|
|
#
|
|
# Permission is hereby granted, free of charge, to any person obtaining
|
|
# a copy of this software and associated documentation files (the
|
|
# "Software"), to deal in the Software without restriction, including
|
|
# without limitation the rights to use, copy, modify, merge, publish,
|
|
# distribute, sublicense, and/or sell copies of the Software, and to
|
|
# permit persons to whom the Software is furnished to do so, subject to
|
|
# the following conditions:
|
|
#
|
|
# The above copyright notice and this permission notice shall be
|
|
# included in all copies or substantial portions of the Software.
|
|
|
|
"""Test module for file ui/__init__.py"""
|
|
|
|
import os
|
|
import shutil
|
|
import unittest
|
|
from copy import deepcopy
|
|
from random import random
|
|
|
|
from beets import config, ui
|
|
from beets.test import _common
|
|
from beets.test.helper import BeetsTestCase, IOMixin, control_stdin
|
|
|
|
|
|
class InputMethodsTest(IOMixin, unittest.TestCase):
|
|
def _print_helper(self, s):
|
|
print(s)
|
|
|
|
def _print_helper2(self, s, prefix):
|
|
print(prefix, s)
|
|
|
|
def test_input_select_objects(self):
|
|
full_items = ["1", "2", "3", "4", "5"]
|
|
|
|
# Test no
|
|
self.io.addinput("n")
|
|
items = ui.input_select_objects(
|
|
"Prompt", full_items, self._print_helper
|
|
)
|
|
assert items == []
|
|
|
|
# Test yes
|
|
self.io.addinput("y")
|
|
items = ui.input_select_objects(
|
|
"Prompt", full_items, self._print_helper
|
|
)
|
|
assert items == full_items
|
|
|
|
# Test selective 1
|
|
self.io.addinput("s")
|
|
self.io.addinput("n")
|
|
self.io.addinput("y")
|
|
self.io.addinput("n")
|
|
self.io.addinput("y")
|
|
self.io.addinput("n")
|
|
items = ui.input_select_objects(
|
|
"Prompt", full_items, self._print_helper
|
|
)
|
|
assert items == ["2", "4"]
|
|
|
|
# Test selective 2
|
|
self.io.addinput("s")
|
|
self.io.addinput("y")
|
|
self.io.addinput("y")
|
|
self.io.addinput("n")
|
|
self.io.addinput("y")
|
|
self.io.addinput("n")
|
|
items = ui.input_select_objects(
|
|
"Prompt", full_items, lambda s: self._print_helper2(s, "Prefix")
|
|
)
|
|
assert items == ["1", "2", "4"]
|
|
|
|
# Test selective 3
|
|
self.io.addinput("s")
|
|
self.io.addinput("y")
|
|
self.io.addinput("n")
|
|
self.io.addinput("y")
|
|
self.io.addinput("q")
|
|
items = ui.input_select_objects(
|
|
"Prompt", full_items, self._print_helper
|
|
)
|
|
assert items == ["1", "3"]
|
|
|
|
|
|
class ParentalDirCreation(BeetsTestCase):
|
|
def test_create_yes(self):
|
|
non_exist_path = _common.os.fsdecode(
|
|
os.path.join(self.temp_dir, b"nonexist", str(random()).encode())
|
|
)
|
|
# Deepcopy instead of recovering because exceptions might
|
|
# occur; wish I can use a golang defer here.
|
|
test_config = deepcopy(config)
|
|
test_config["library"] = non_exist_path
|
|
with control_stdin("y"):
|
|
lib = ui._open_library(test_config)
|
|
lib._close()
|
|
|
|
def test_create_no(self):
|
|
non_exist_path_parent = _common.os.fsdecode(
|
|
os.path.join(self.temp_dir, b"nonexist")
|
|
)
|
|
non_exist_path = _common.os.fsdecode(
|
|
os.path.join(non_exist_path_parent.encode(), str(random()).encode())
|
|
)
|
|
test_config = deepcopy(config)
|
|
test_config["library"] = non_exist_path
|
|
|
|
with control_stdin("n"):
|
|
try:
|
|
lib = ui._open_library(test_config)
|
|
except ui.UserError:
|
|
if os.path.exists(non_exist_path_parent):
|
|
shutil.rmtree(non_exist_path_parent)
|
|
raise OSError("Parent directories should not be created.")
|
|
else:
|
|
if lib:
|
|
lib._close()
|
|
raise OSError("Parent directories should not be created.")
|