WIP: Temporary commit for branch move
This commit is contained in:
@@ -11,7 +11,7 @@ from database.db import BotDB
|
||||
def bot():
|
||||
"""Фикстура для создания объекта BotDB."""
|
||||
current_dir = os.getcwd()
|
||||
return BotDB(current_dir, "test.db")
|
||||
return BotDB(current_dir, "database/test.db")
|
||||
|
||||
|
||||
@pytest.fixture(autouse=True, )
|
||||
@@ -38,7 +38,7 @@ def setup_db():
|
||||
# Other data
|
||||
date = "2024-07-10"
|
||||
next_date = "2024-07-11"
|
||||
conn = sqlite3.connect("test.db")
|
||||
conn = sqlite3.connect("database/test.db")
|
||||
cursor = conn.cursor()
|
||||
cursor.execute("""
|
||||
CREATE TABLE IF NOT EXISTS "admins" (
|
||||
@@ -139,12 +139,12 @@ def setup_db():
|
||||
conn.commit()
|
||||
conn.close()
|
||||
yield
|
||||
os.remove('test.db')
|
||||
os.remove('database/test.db')
|
||||
|
||||
|
||||
def test_bot_init(bot):
|
||||
"""Проверяет, что объект BotDB инициализируется с правильным именем файла."""
|
||||
assert bot.db_file == os.path.join(os.getcwd(), "test.db")
|
||||
assert bot.db_file == os.path.join(os.getcwd(), "database", "test.db")
|
||||
# Проверьте, что соединения с базой данных нет, так как оно не устанавливается в init
|
||||
assert bot.conn is None
|
||||
assert bot.cursor is None
|
||||
@@ -174,7 +174,7 @@ def test_create_table_success(bot):
|
||||
bot.create_table(sql_script)
|
||||
|
||||
# Проверяем, что таблица создана
|
||||
conn = sqlite3.connect('test.db')
|
||||
conn = sqlite3.connect('database/test.db')
|
||||
cursor = conn.cursor()
|
||||
cursor.execute("SELECT name FROM sqlite_master WHERE type='table' AND name='test_table'")
|
||||
result = cursor.fetchone()
|
||||
@@ -192,7 +192,7 @@ def test_create_table_error(bot):
|
||||
|
||||
|
||||
def test_get_current_version_success(bot):
|
||||
conn = sqlite3.connect('test.db')
|
||||
conn = sqlite3.connect('database/test.db')
|
||||
cursor = conn.cursor()
|
||||
cursor.execute("INSERT INTO migrations (version, script_name) VALUES (123, 'test')")
|
||||
conn.commit()
|
||||
@@ -216,7 +216,7 @@ def test_update_version_success(bot):
|
||||
bot.update_version(new_version, script_name)
|
||||
|
||||
# Проверяем, что данные записаны в таблицу
|
||||
conn = sqlite3.connect('test.db')
|
||||
conn = sqlite3.connect('database/test.db')
|
||||
cursor = conn.cursor()
|
||||
cursor.execute("SELECT * FROM migrations WHERE version = ?", (new_version,))
|
||||
result = cursor.fetchone()
|
||||
@@ -228,7 +228,7 @@ def test_update_version_success(bot):
|
||||
|
||||
|
||||
def test_update_version_integrity_error(bot):
|
||||
conn = sqlite3.connect('test.db')
|
||||
conn = sqlite3.connect('database/test.db')
|
||||
cursor = conn.cursor()
|
||||
cursor.execute("INSERT INTO migrations (version, script_name) VALUES (123, 'test')")
|
||||
conn.commit()
|
||||
@@ -261,7 +261,7 @@ def test_add_new_user_in_db(bot):
|
||||
)
|
||||
|
||||
# Проверяем наличие записи в базе данных
|
||||
conn = sqlite3.connect('test.db')
|
||||
conn = sqlite3.connect('database/test.db')
|
||||
cursor = conn.cursor()
|
||||
cursor.execute("SELECT * FROM our_users WHERE user_id = ?", (user_id,))
|
||||
result = cursor.fetchone()
|
||||
@@ -306,7 +306,7 @@ def test_add_new_user_in_db_empty_first_name(bot):
|
||||
)
|
||||
|
||||
# Проверяем наличие записи в базе данных
|
||||
conn = sqlite3.connect('test.db')
|
||||
conn = sqlite3.connect('database/test.db')
|
||||
cursor = conn.cursor()
|
||||
cursor.execute(f"SELECT * FROM our_users WHERE user_id = ?", (user_id,))
|
||||
result = cursor.fetchone()
|
||||
@@ -388,7 +388,7 @@ def test_get_username_error(bot):
|
||||
|
||||
def test_get_all_user_id_empty(bot):
|
||||
"""Проверяет, что функция возвращает пустой список, если в базе нет пользователей."""
|
||||
conn = sqlite3.connect('test.db')
|
||||
conn = sqlite3.connect('database/test.db')
|
||||
cursor = conn.cursor()
|
||||
cursor.execute("DELETE FROM our_users")
|
||||
conn.commit()
|
||||
@@ -481,7 +481,7 @@ def test_update_info_about_stickers_success(bot):
|
||||
bot.update_info_about_stickers(user_id)
|
||||
|
||||
# Проверяем, что информация обновлена
|
||||
conn = sqlite3.connect('test.db')
|
||||
conn = sqlite3.connect('database/test.db')
|
||||
cursor = conn.cursor()
|
||||
cursor.execute("SELECT has_stickers FROM our_users WHERE user_id = ?", (user_id,))
|
||||
result = cursor.fetchone()
|
||||
@@ -495,7 +495,7 @@ def test_update_info_about_stickers_not_found(bot):
|
||||
bot.update_info_about_stickers(user_id)
|
||||
|
||||
# Проверяем, что база данных не изменилась
|
||||
conn = sqlite3.connect('test.db')
|
||||
conn = sqlite3.connect('database/test.db')
|
||||
cursor = conn.cursor()
|
||||
cursor.execute("SELECT COUNT(*) FROM our_users WHERE user_id = ?", (user_id,))
|
||||
result = cursor.fetchone()
|
||||
@@ -512,7 +512,7 @@ def test_update_info_about_stickers_error(bot):
|
||||
|
||||
def test_get_users_blacklist_empty(bot):
|
||||
"""Проверяет, что функция возвращает пустой словарь, если в черном списке нет пользователей."""
|
||||
conn = sqlite3.connect('test.db')
|
||||
conn = sqlite3.connect('database/test.db')
|
||||
cursor = conn.cursor()
|
||||
cursor.execute("DELETE FROM blacklist")
|
||||
conn.commit()
|
||||
@@ -619,7 +619,7 @@ def test_set_user_blacklist_success(bot):
|
||||
assert bot.set_user_blacklist(user_id, user_name, message_for_user, date_to_unban) is None
|
||||
|
||||
# Проверяем, что запись добавлена в базу
|
||||
conn = sqlite3.connect('test.db')
|
||||
conn = sqlite3.connect('database/test.db')
|
||||
cursor = conn.cursor()
|
||||
cursor.execute("SELECT * FROM blacklist WHERE user_id = ?", (user_id,))
|
||||
result = cursor.fetchone()
|
||||
@@ -658,7 +658,7 @@ def test_delete_user_blacklist_success(bot):
|
||||
|
||||
@pytest.mark.xfail
|
||||
def test_delete_user_blacklist_not_found(bot):
|
||||
conn = sqlite3.connect('test.db')
|
||||
conn = sqlite3.connect('database/test.db')
|
||||
cursor = conn.cursor()
|
||||
cursor.execute("INSERT INTO blacklist (user_id, user_name, date_to_unban) VALUES (?, ?, ?)",
|
||||
(12345, "JohnDoe", "2023-12-26"))
|
||||
@@ -691,7 +691,7 @@ def test_add_new_message_in_db_error(bot):
|
||||
def test_update_date_for_user_success(bot):
|
||||
bot.update_date_for_user('2024-07-15', 12345)
|
||||
|
||||
conn = sqlite3.connect('test.db')
|
||||
conn = sqlite3.connect('database/test.db')
|
||||
cursor = conn.cursor()
|
||||
cursor.execute("SELECT date_changed FROM our_users WHERE user_id = ?", (12345,))
|
||||
new_date = cursor.fetchone()[0]
|
||||
@@ -741,7 +741,7 @@ def test_get_last_users_from_db_success(bot):
|
||||
|
||||
|
||||
def test_get_last_users_from_db_empty(bot):
|
||||
conn = sqlite3.connect('test.db')
|
||||
conn = sqlite3.connect('database/test.db')
|
||||
cursor = conn.cursor()
|
||||
cursor.execute("DELETE FROM our_users")
|
||||
conn.commit()
|
||||
@@ -768,7 +768,7 @@ def test_get_banned_users_from_db_success(bot):
|
||||
|
||||
|
||||
def test_get_banned_users_from_db_empty(bot):
|
||||
conn = sqlite3.connect('test.db')
|
||||
conn = sqlite3.connect('database/test.db')
|
||||
cursor = conn.cursor()
|
||||
cursor.execute("DELETE FROM blacklist")
|
||||
conn.commit()
|
||||
@@ -801,7 +801,7 @@ def test_get_banned_users_from_db_with_limits_success_offset(bot):
|
||||
|
||||
|
||||
def test_get_banned_users_from_db_with_limits_empty(bot):
|
||||
conn = sqlite3.connect('test.db')
|
||||
conn = sqlite3.connect('database/test.db')
|
||||
cursor = conn.cursor()
|
||||
cursor.execute("DELETE FROM blacklist")
|
||||
conn.commit()
|
||||
@@ -818,7 +818,7 @@ def test_get_banned_users_from_db_with_limits_error(bot):
|
||||
|
||||
|
||||
def __drop_table(table_name: str):
|
||||
conn = sqlite3.connect('test.db')
|
||||
conn = sqlite3.connect('database/test.db')
|
||||
cursor = conn.cursor()
|
||||
cursor.execute(f"DROP TABLE {table_name}")
|
||||
conn.commit()
|
||||
|
||||
Reference in New Issue
Block a user