From e7aeb4e2d7c2f49700eb837244c64b8a4c4011f7 Mon Sep 17 00:00:00 2001 From: Tanmoy Sarkar <57363826+tanmoysrt@users.noreply.github.com> Date: Tue, 22 Oct 2024 01:50:48 +0530 Subject: [PATCH] chore: ruff format --- agent/tests/test_database.py | 16 ++++------------ 1 file changed, 4 insertions(+), 12 deletions(-) diff --git a/agent/tests/test_database.py b/agent/tests/test_database.py index 911d6ea2..e115db82 100644 --- a/agent/tests/test_database.py +++ b/agent/tests/test_database.py @@ -12,9 +12,7 @@ class DatabaseTestInstance: # Test database instance with few utility functions def __init__(self) -> None: self.db_root_password = "123456" - self.db_container = MySqlContainer( - image="mysql:8.0", MYSQL_ROOT_PASSWORD=self.db_root_password - ) + self.db_container = MySqlContainer(image="mysql:8.0", MYSQL_ROOT_PASSWORD=self.db_root_password) self.db_container.start() @property @@ -46,9 +44,7 @@ def create_database_user(self, db_name: str, username: str, password: str): "FLUSH PRIVILEGES", ] for query in queries: - command = ( - f'mysql -h 127.0.0.1 -uroot -p{self.db_root_password} -e "{query}"' - ) + command = f'mysql -h 127.0.0.1 -uroot -p{self.db_root_password} -e "{query}"' self.execute_cmd(command) @@ -94,15 +90,11 @@ def tearDown(self) -> None: self.instance.destroy() def _db(self, db_name: str, username: str, password: str) -> Database: - return Database( - self.instance.host, self.instance.port, username, password, db_name - ) + return Database(self.instance.host, self.instance.port, username, password, db_name) def test_execute_query(self): """Basic test for `execute_query` function""" db = self._db(self.db1__name, self.db1__username, self.db1__password) - success, data = db.execute_query( - "SELECT * FROM Person", commit=False, as_dict=True - ) + success, data = db.execute_query("SELECT * FROM Person", commit=False, as_dict=True) self.assertEqual(success, True, "run sql query") self.assertEqual(data[0]["row_count"], 5)