WIP: Temporary commit for branch move

This commit is contained in:
2025-08-26 02:14:11 +03:00
parent ee9eafa09f
commit 7b6abe2a0e
44 changed files with 4783 additions and 2383 deletions

View File

@@ -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()