mirror of
https://github.com/python-telegram-bot/python-telegram-bot.git
synced 2025-03-17 04:39:55 +01:00
Use temporary directories in test_persistence (#1808)
This commit is contained in:
parent
8d6970ab02
commit
f7ec7a7c4c
1 changed files with 11 additions and 25 deletions
|
@ -37,6 +37,17 @@ from telegram.ext import BasePersistence, Updater, ConversationHandler, MessageH
|
|||
PicklePersistence, CommandHandler, DictPersistence, TypeHandler
|
||||
|
||||
|
||||
@pytest.fixture(autouse=True)
|
||||
def change_directory(tmp_path):
|
||||
orig_dir = os.getcwd()
|
||||
# Switch to a temporary directory so we don't have to worry about cleaning up files
|
||||
# (str() for py<3.6)
|
||||
os.chdir(str(tmp_path))
|
||||
yield
|
||||
# Go back to original directory
|
||||
os.chdir(orig_dir)
|
||||
|
||||
|
||||
@pytest.fixture(scope="function")
|
||||
def base_persistence():
|
||||
return BasePersistence(store_chat_data=True, store_user_data=True, store_bot_data=True)
|
||||
|
@ -310,9 +321,6 @@ def bad_pickle_files():
|
|||
with open(name, 'w') as f:
|
||||
f.write('(())')
|
||||
yield True
|
||||
for name in ['pickletest_user_data', 'pickletest_chat_data', 'pickletest_bot_data',
|
||||
'pickletest_conversations', 'pickletest']:
|
||||
os.remove(name)
|
||||
|
||||
|
||||
@pytest.fixture(scope='function')
|
||||
|
@ -330,9 +338,6 @@ def good_pickle_files(user_data, chat_data, bot_data, conversations):
|
|||
with open('pickletest', 'wb') as f:
|
||||
pickle.dump(data, f)
|
||||
yield True
|
||||
for name in ['pickletest_user_data', 'pickletest_chat_data', 'pickletest_bot_data',
|
||||
'pickletest_conversations', 'pickletest']:
|
||||
os.remove(name)
|
||||
|
||||
|
||||
@pytest.fixture(scope='function')
|
||||
|
@ -347,9 +352,6 @@ def pickle_files_wo_bot_data(user_data, chat_data, conversations):
|
|||
with open('pickletest', 'wb') as f:
|
||||
pickle.dump(data, f)
|
||||
yield True
|
||||
for name in ['pickletest_user_data', 'pickletest_chat_data',
|
||||
'pickletest_conversations', 'pickletest']:
|
||||
os.remove(name)
|
||||
|
||||
|
||||
@pytest.fixture(scope='function')
|
||||
|
@ -776,9 +778,6 @@ class TestPickelPersistence(object):
|
|||
assert pickle_persistence_2.get_bot_data()['test'] == 'Working3!'
|
||||
|
||||
def test_flush_on_stop_only_bot(self, bot, update, pickle_persistence_only_bot):
|
||||
os.remove('pickletest_user_data')
|
||||
os.remove('pickletest_chat_data')
|
||||
os.remove('pickletest_bot_data')
|
||||
u = Updater(bot=bot, persistence=pickle_persistence_only_bot)
|
||||
dp = u.dispatcher
|
||||
u.running = True
|
||||
|
@ -800,7 +799,6 @@ class TestPickelPersistence(object):
|
|||
assert pickle_persistence_2.get_bot_data()['my_test3'] == 'Working3!'
|
||||
|
||||
def test_flush_on_stop_only_chat(self, bot, update, pickle_persistence_only_chat):
|
||||
os.remove('pickletest_bot_data')
|
||||
u = Updater(bot=bot, persistence=pickle_persistence_only_chat)
|
||||
dp = u.dispatcher
|
||||
u.running = True
|
||||
|
@ -821,7 +819,6 @@ class TestPickelPersistence(object):
|
|||
assert pickle_persistence_2.get_bot_data() == {}
|
||||
|
||||
def test_flush_on_stop_only_user(self, bot, update, pickle_persistence_only_user):
|
||||
os.remove('pickletest_chat_data')
|
||||
u = Updater(bot=bot, persistence=pickle_persistence_only_user)
|
||||
dp = u.dispatcher
|
||||
u.running = True
|
||||
|
@ -923,17 +920,6 @@ class TestPickelPersistence(object):
|
|||
assert nested_ch.conversations[nested_ch._get_key(update)] == 1
|
||||
assert nested_ch.conversations == pickle_persistence.conversations['name3']
|
||||
|
||||
@classmethod
|
||||
def teardown_class(cls):
|
||||
try:
|
||||
for name in ['pickletest_user_data', 'pickletest_chat_data',
|
||||
'pickletest_bot_data',
|
||||
'pickletest_conversations',
|
||||
'pickletest']:
|
||||
os.remove(name)
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
|
||||
@pytest.fixture(scope='function')
|
||||
def user_data_json(user_data):
|
||||
|
|
Loading…
Add table
Reference in a new issue