singlestoredb 1.16.1__py3-none-any.whl

singlestoredb/tests/test_dbapi.py

@@ -0,0 +1,27 @@
+# type: ignore
+import os
+
+import singlestoredb as s2
+from . import utils
+from singlestoredb.mysql.tests.thirdparty.test_MySQLdb import test_MySQLdb_dbapi20
+
+
+class TestDBAPI(test_MySQLdb_dbapi20.test_MySQLdb):
+
+    driver = s2
+
+    dbname: str = ''
+    dbexisted: bool = False
+
+    @classmethod
+    def setUpClass(cls):
+        sql_file = os.path.join(os.path.dirname(__file__), 'empty.sql')
+        cls.dbname, cls.dbexisted = utils.load_sql(sql_file)
+
+    @classmethod
+    def tearDownClass(cls):
+        if not cls.dbexisted:
+            utils.drop_database(cls.dbname)
+
+    def _connect(self):
+        return s2.connect(database=type(self).dbname)

singlestoredb/tests/test_exceptions.py

@@ -0,0 +1,45 @@
+#!/usr/bin/env python
+# type: ignore
+"""Test SingleStoreDB exceptions."""
+import unittest
+
+import singlestoredb as s2
+
+
+class TestExceptions(unittest.TestCase):
+
+    def test_str(self):
+        exc = s2.Error()
+        assert str(exc) == 'Unknown error', str(exc)
+
+        exc = s2.Error(errno=1)
+        assert str(exc) == '1', str(exc)
+
+        exc = s2.Error(msg='hi there')
+        assert str(exc) == 'hi there', str(exc)
+
+        exc = s2.Error(errno=1, msg='hi there')
+        assert str(exc) == '1: hi there', str(exc)
+
+        exc = s2.Error(errno=1, msg='hi there', sqlstate=9)
+        assert str(exc) == '1 (9): hi there'
+
+        exc = s2.Error(msg='hi there', sqlstate=9)
+        assert str(exc) == '(9): hi there'
+
+        exc = s2.Error(sqlstate=9)
+        assert str(exc) == '(9)'
+
+    def test_repr(self):
+        exc = s2.Error(errno=1, msg='hi there')
+        assert str(exc) == '1: hi there', str(exc)
+        assert str(exc) == repr(exc)
+
+    def test_msg(self):
+        exc = s2.Error(errno=1, msg='hi there')
+        assert exc.msg == exc.errmsg
+
+
+if __name__ == '__main__':
+    import nose2
+    nose2.main()