mysql: add support for RocksDB

neo/storage/database/mysqldb.py

@@ -49,7 +49,7 @@ class MySQLDatabaseManager(DatabaseManager):
     """This class manages a database on MySQL."""
 
     VERSION = 1
-    ENGINES = "InnoDB", "TokuDB"
+    ENGINES = "InnoDB", "RocksDB", "TokuDB"
     _engine = ENGINES[0] # default engine
 
     # Disabled even on MySQL 5.1-5.5 and MariaDB 5.2-5.3 because

neo/tests/storage/testStorageMySQL.py

@@ -15,7 +15,8 @@
 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
 import unittest
-from MySQLdb import OperationalError
+from MySQLdb import NotSupportedError, OperationalError
+from MySQLdb.constants.ER import UNKNOWN_STORAGE_ENGINE
 from ..mock import Mock
 from neo.lib.exception import DatabaseFailure
 from neo.lib.util import p64
@@ -35,7 +36,13 @@ class StorageMySQLdbTests(StorageDBTests):
         db = MySQLDatabaseManager(database, self.engine)
         self.assertEqual(db.db, DB_PREFIX + '0')
         self.assertEqual(db.user, DB_USER)
+        try:
             db.setup(reset)
+        except NotSupportedError as m:
+            code, m = m.args
+            if code != UNKNOWN_STORAGE_ENGINE:
+                raise
+            raise unittest.SkipTest(m)
         return db
 
     def test_query1(self):
@@ -119,6 +126,11 @@ class StorageMySQLdbTests(StorageDBTests):
             self.assertEqual(len(query_list), count)
 
 
+class StorageMySQLdbRocksDBTests(StorageMySQLdbTests):
+
+    engine = "RocksDB"
+
+
 class StorageMySQLdbTokuDBTests(StorageMySQLdbTests):
 
     engine = "TokuDB"