aboutsummaryrefslogtreecommitdiffstats
path: root/vespalib/src/tests/util/file_area_freelist/file_area_freelist_test.cpp
blob: 470680e0f8f4e2de636dd1e9f57d50b43956a1b5 (plain) (blame)
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
102
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.

#include <vespa/vespalib/util/file_area_freelist.h>
#include <vespa/vespalib/gtest/gtest.h>

using vespalib::alloc::FileAreaFreeList;

class FileAreaFreeListTest : public ::testing::Test
{
protected:
    FileAreaFreeList _freelist;
    static constexpr auto bad_offset = FileAreaFreeList::bad_offset;

public:
    FileAreaFreeListTest();
    ~FileAreaFreeListTest();
    void test_merge_area_blocked(bool previous);
};

FileAreaFreeListTest::FileAreaFreeListTest()
    : _freelist()
{
}

FileAreaFreeListTest::~FileAreaFreeListTest() = default;

void
FileAreaFreeListTest::test_merge_area_blocked(bool previous)
{
    _freelist.add_premmapped_area(4, 1);
    _freelist.add_premmapped_area(5, 1);
    EXPECT_EQ(4, _freelist.alloc(1));
    EXPECT_EQ(5, _freelist.alloc(1));
    if (previous) {
        _freelist.free(4, 1);
        _freelist.free(5, 1);
    } else {
        _freelist.free(5, 1);
        _freelist.free(4, 1);
    }
    EXPECT_EQ(bad_offset, _freelist.alloc(2));
    _freelist.remove_premmapped_area(4, 1);
    _freelist.remove_premmapped_area(5, 1);
}

TEST_F(FileAreaFreeListTest, empty_freelist_is_ok)
{
    EXPECT_EQ(bad_offset, _freelist.alloc(1));
}

TEST_F(FileAreaFreeListTest, can_reuse_free_area)
{
    _freelist.free(4, 1);
    EXPECT_EQ(4, _freelist.alloc(1));
    EXPECT_EQ(bad_offset, _freelist.alloc(1));
}

TEST_F(FileAreaFreeListTest, merge_area_with_next_area)
{
    _freelist.free(5, 1);
    _freelist.free(4, 1);
    EXPECT_EQ(4, _freelist.alloc(2));
    EXPECT_EQ(bad_offset, _freelist.alloc(1));
}

TEST_F(FileAreaFreeListTest, merge_area_with_next_area_blocked_by_fence)
{
    test_merge_area_blocked(false);
}

TEST_F(FileAreaFreeListTest, merge_area_with_previous_area)
{
    _freelist.free(3, 1);
    _freelist.free(4, 1);
    EXPECT_EQ(3, _freelist.alloc(2));
    EXPECT_EQ(bad_offset, _freelist.alloc(1));
}

TEST_F(FileAreaFreeListTest, merge_area_with_previous_area_blocked_by_fence)
{
    test_merge_area_blocked(true);
}

TEST_F(FileAreaFreeListTest, merge_area_with_previous_and_next_area)
{
    _freelist.free(5, 1);
    _freelist.free(3, 1);
    _freelist.free(4, 1);
    EXPECT_EQ(3, _freelist.alloc(3));
    EXPECT_EQ(bad_offset, _freelist.alloc(1));
}

TEST_F(FileAreaFreeListTest, can_use_part_of_free_area)
{
    _freelist.free(4, 2);
    EXPECT_EQ(4, _freelist.alloc(1));
    EXPECT_EQ(5, _freelist.alloc(1));
    EXPECT_EQ(bad_offset, _freelist.alloc(1));
}


GTEST_MAIN_RUN_ALL_TESTS()