aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorGravatar ionite34 <[email protected]>2022-12-04 09:49:15 +0800
committerGravatar ionite34 <[email protected]>2022-12-04 09:49:15 +0800
commita71e1234862cc00741cf2ec03ea32e6f5332289e (patch)
treec6f2567e2574b36ea60ba91a3df2ab4ecf8be378
parentRemove readme implementation details (diff)
Add refactors for Size enum rename
-rw-r--r--snekbox/nsjail.py2
-rw-r--r--tests/test_nsjail.py4
2 files changed, 3 insertions, 3 deletions
diff --git a/snekbox/nsjail.py b/snekbox/nsjail.py
index 887bc9a..4933a0a 100644
--- a/snekbox/nsjail.py
+++ b/snekbox/nsjail.py
@@ -51,7 +51,7 @@ class NsJail:
config_path: str = "./config/snekbox.cfg",
max_output_size: int = 1_000_000,
read_chunk_size: int = 10_000,
- memfs_instance_size: int = 48 * Size.MB,
+ memfs_instance_size: int = 48 * Size.MiB,
files_limit: int | None = 100,
files_timeout: float | None = 8,
files_pattern: str = "**/*",
diff --git a/tests/test_nsjail.py b/tests/test_nsjail.py
index 927b8c0..71405c7 100644
--- a/tests/test_nsjail.py
+++ b/tests/test_nsjail.py
@@ -19,7 +19,7 @@ class NsJailTests(unittest.TestCase):
super().setUp()
# Specify lower limits for unit tests to complete within time limits
- self.nsjail = NsJail(memfs_instance_size=2 * Size.MB)
+ self.nsjail = NsJail(memfs_instance_size=2 * Size.MiB)
self.logger = logging.getLogger("snekbox.nsjail")
self.logger.setLevel(logging.WARNING)
@@ -200,7 +200,7 @@ class NsJailTests(unittest.TestCase):
"""
).strip()
- nsjail = NsJail(memfs_instance_size=32 * Size.MB, files_timeout=1)
+ nsjail = NsJail(memfs_instance_size=32 * Size.MiB, files_timeout=1)
result = nsjail.python3(["-c", code])
self.assertEqual(result.returncode, None)
self.assertEqual(