1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
|
# SPDX-License-Identifier: (Apache-2.0 OR MIT)
# Copyright ijl (2025)
import os
import pytest
import orjson
ORJSON_RUNNER_MEMORY_GIB = os.getenv("ORJSON_RUNNER_MEMORY_GIB", "")
@pytest.mark.skipif(
not ORJSON_RUNNER_MEMORY_GIB,
reason="ORJSON_RUNNER_MEMORY_GIB not defined",
)
def test_memory_loads():
buffer_factor = 12
segment = "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"
size = (
(int(ORJSON_RUNNER_MEMORY_GIB) * 1024 * 1024 * 1024)
// buffer_factor
// len(segment)
)
doc = "".join(segment for _ in range(size))
with pytest.raises(orjson.JSONDecodeError) as exc_info:
_ = orjson.loads(doc)
assert (
str(exc_info.value)
== "Not enough memory to allocate buffer for parsing: line 1 column 1 (char 0)"
)
|
