diff --git a/src/tests/test_sqlthread.py b/src/tests/test_sqlthread.py index 1eddedb7..a612df3a 100644 --- a/src/tests/test_sqlthread.py +++ b/src/tests/test_sqlthread.py @@ -1,44 +1,44 @@ -# """Tests for SQL thread""" -# # flake8: noqa:E402 -# import os -# import tempfile -# import threading -# import unittest -# -# from .common import skip_python3 -# -# skip_python3() -# -# os.environ['BITMESSAGE_HOME'] = tempfile.gettempdir() -# -# from pybitmessage.helper_sql import ( -# sqlQuery, sql_ready, sqlStoredProcedure) # noqa:E402 -# from pybitmessage.class_sqlThread import sqlThread # noqa:E402 -# from pybitmessage.addresses import encodeAddress # noqa:E402 -# -# -# class TestSqlThread(unittest.TestCase): -# """Test case for SQL thread""" -# -# @classmethod -# def setUpClass(cls): -# # Start SQL thread -# sqlLookup = sqlThread() -# sqlLookup.daemon = True -# sqlLookup.start() -# sql_ready.wait() -# -# @classmethod -# def tearDownClass(cls): -# sqlStoredProcedure('exit') -# for thread in threading.enumerate(): -# if thread.name == "SQL": -# thread.join() -# -# def test_create_function(self): -# """Check the result of enaddr function""" -# encoded_str = encodeAddress(4, 1, "21122112211221122112") -# -# query = sqlQuery('SELECT enaddr(4, 1, "21122112211221122112")') -# self.assertEqual( -# query[0][-1], encoded_str, "test case fail for create_function") +"""Tests for SQL thread""" +# flake8: noqa:E402 +import os +import tempfile +import threading +import unittest + +from .common import skip_python3 + +skip_python3() + +os.environ['BITMESSAGE_HOME'] = tempfile.gettempdir() + +from pybitmessage.helper_sql import ( + sqlQuery, sql_ready, sqlStoredProcedure) # noqa:E402 +from pybitmessage.class_sqlThread import sqlThread # noqa:E402 +from pybitmessage.addresses import encodeAddress # noqa:E402 + + +class TestSqlThread(unittest.TestCase): + """Test case for SQL thread""" + + @classmethod + def setUpClass(cls): + # Start SQL thread + sqlLookup = sqlThread() + sqlLookup.daemon = True + sqlLookup.start() + sql_ready.wait() + + @classmethod + def tearDownClass(cls): + sqlStoredProcedure('exit') + for thread in threading.enumerate(): + if thread.name == "SQL": + thread.join() + + def test_create_function(self): + """Check the result of enaddr function""" + encoded_str = encodeAddress(4, 1, "21122112211221122112") + + query = sqlQuery('SELECT enaddr(4, 1, "21122112211221122112")') + self.assertEqual( + query[0][-1], encoded_str, "test case fail for create_function")