Update tests
This commit is contained in:
parent
91ca5fc14a
commit
3a0d2b5881
@ -2,7 +2,7 @@ import gc
|
|||||||
import pytest
|
import pytest
|
||||||
import os
|
import os
|
||||||
|
|
||||||
from server.storage import Storage, open_db
|
from server.storage import Storage, db_class
|
||||||
from lib.util import subclasses
|
from lib.util import subclasses
|
||||||
|
|
||||||
# Find out which db engines to test
|
# Find out which db engines to test
|
||||||
@ -21,7 +21,7 @@ for c in subclasses(Storage):
|
|||||||
def db(tmpdir, request):
|
def db(tmpdir, request):
|
||||||
cwd = os.getcwd()
|
cwd = os.getcwd()
|
||||||
os.chdir(str(tmpdir))
|
os.chdir(str(tmpdir))
|
||||||
db = open_db("db", request.param, False)
|
db = db_class(request.param)("db", False)
|
||||||
yield db
|
yield db
|
||||||
os.chdir(cwd)
|
os.chdir(cwd)
|
||||||
# Make sure all the locks and handles are closed
|
# Make sure all the locks and handles are closed
|
||||||
@ -72,5 +72,5 @@ def test_iterator_reverse(db):
|
|||||||
def test_close(db):
|
def test_close(db):
|
||||||
db.put(b"a", b"b")
|
db.put(b"a", b"b")
|
||||||
db.close()
|
db.close()
|
||||||
db = open_db("db", db.__class__.__name__, False)
|
db = db_class(db.__class__.__name__)("db", False)
|
||||||
assert db.get(b"a") == b"b"
|
assert db.get(b"a") == b"b"
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user