fivem/code/tests/shared/TestMemoryScriptBufferPool.cpp at master · NGcodeX/fivem · GitHub

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
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
#include <type_traits>
#include <StdInc.h>

#include <catch_amalgamated.hpp>

#include <fxScripting.h>
#include <fxScriptBuffer.h>

#include "TestUtils.h"

TEST_CASE("MemoryScriptBufferPool tests")
{
	SECTION("Constructor test")
	{
		fx::MemoryScriptBuffer::MemoryScriptBufferPool pool(1024);
		REQUIRE(pool.GetResultBuffer(0) == nullptr);
	}

	SECTION("GetResultBuffer test")
	{
		fx::MemoryScriptBuffer::MemoryScriptBufferPool pool(1024);
		char* buffer = pool.GetResultBuffer(512);
		REQUIRE(pool.GetFreeBlockCount() == 1);
		REQUIRE(buffer != nullptr);
		pool.FreeResultBuffer(buffer, 512);
		REQUIRE(pool.GetFreeBlockCount() == 1);
	}

	SECTION("FreeResultBuffer test")
	{
		fx::MemoryScriptBuffer::MemoryScriptBufferPool pool(1024);
		char* buffer = pool.GetResultBuffer(512);
		pool.FreeResultBuffer(buffer, 512);
		char* buffer2 = pool.GetResultBuffer(512);
		REQUIRE(buffer2 != nullptr);
		REQUIRE(buffer == buffer2);
	}

	SECTION("Pool exhaustion test")
	{
		fx::MemoryScriptBuffer::MemoryScriptBufferPool pool(1024);
		char* buffer = pool.GetResultBuffer(1025);
		REQUIRE(buffer == nullptr);
	}

	SECTION("Pool to small fragmentation test")
	{
		fx::MemoryScriptBuffer::MemoryScriptBufferPool pool(1024);
		char* buffer1 = pool.GetResultBuffer(256);
		char* buffer2 = pool.GetResultBuffer(256);
		char* buffer3 = pool.GetResultBuffer(256);
		REQUIRE(pool.GetFreeBlockCount() == 1);
		// Free the middle buffer
		pool.FreeResultBuffer(buffer2, 256);
		REQUIRE(pool.GetFreeBlockCount() == 2);
		char* buffer4 = pool.GetResultBuffer(512);
		REQUIRE(buffer4 == nullptr);
	}

	SECTION("Pool fragmentation test")
	{
		fx::MemoryScriptBuffer::MemoryScriptBufferPool pool(1280);
		char* buffer1 = pool.GetResultBuffer(256);
		REQUIRE(pool.GetFreeBlockCount() == 1);
		char* buffer2 = pool.GetResultBuffer(256);
		REQUIRE(pool.GetFreeBlockCount() == 1);
		char* buffer3 = pool.GetResultBuffer(256);
		REQUIRE(pool.GetFreeBlockCount() == 1);
		char* buffer4 = pool.GetResultBuffer(256);
		REQUIRE(pool.GetFreeBlockCount() == 1);
		// Free the middle buffers
		pool.FreeResultBuffer(buffer2, 256);
		REQUIRE(pool.GetFreeBlockCount() == 2);
		pool.FreeResultBuffer(buffer3, 256);
		REQUIRE(pool.GetFreeBlockCount() == 2);
		char* buffer5 = pool.GetResultBuffer(512);
		REQUIRE(pool.GetFreeBlockCount() == 1);
		REQUIRE(buffer5 != nullptr);
	}

	SECTION("Pool empty fragmentation test")
	{
		fx::MemoryScriptBuffer::MemoryScriptBufferPool pool(1024);
		char* buffer1 = pool.GetResultBuffer(256);
		REQUIRE(pool.GetFreeBlockCount() == 1);
		char* buffer2 = pool.GetResultBuffer(256);
		REQUIRE(pool.GetFreeBlockCount() == 1);
		char* buffer3 = pool.GetResultBuffer(256);
		REQUIRE(pool.GetFreeBlockCount() == 1);
		char* buffer4 = pool.GetResultBuffer(256);
		REQUIRE(pool.GetFreeBlockCount() == 0);
		pool.FreeResultBuffer(buffer1, 256);
		REQUIRE(pool.GetFreeBlockCount() == 1);
		pool.FreeResultBuffer(buffer2, 256);
		REQUIRE(pool.GetFreeBlockCount() == 1);
		pool.FreeResultBuffer(buffer3, 256);
		REQUIRE(pool.GetFreeBlockCount() == 1);
		pool.FreeResultBuffer(buffer4, 256);
		REQUIRE(pool.GetFreeBlockCount() == 1);
	}
}