Skip to content

Commit 6aab1e9

Browse files
committed
Updated context to select default storage engine based on version.
Replicates kaizendorks#119
1 parent e5b2221 commit 6aab1e9

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

pymongo_inmemory/context.py

+7-2
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
from os import path
66
import platform
77

8-
from ._utils import mkdir_ifnot_exist
8+
from ._utils import mkdir_ifnot_exist, make_semver
99
from .downloader._urls import best_url
1010

1111
DEFAULT_CONF = {}
@@ -127,7 +127,7 @@ def __init__(
127127
)
128128
self.archive_folder = mkdir_ifnot_exist(self.download_folder, self.url_hash)
129129
self.extracted_folder = mkdir_ifnot_exist(self.extract_folder, self.url_hash)
130-
self.storage_engine = conf("storage_engine", "ephemeralForTest")
130+
self.storage_engine = self._build_storage_engine()
131131

132132
def __str__(self):
133133
return (
@@ -165,3 +165,8 @@ def _build_download_url(self):
165165

166166
self.downloaded_version = downloaded_version
167167
return dl_url
168+
169+
def _build_storage_engine(self):
170+
major, minor, patch = make_semver(self.downloaded_version)
171+
storage_engine_fallback = "wiredTiger" if major > 6 else "ephemeralForTest"
172+
return conf("storage_engine", storage_engine_fallback)

0 commit comments

Comments
 (0)