1 // Copyright 2024 The Pigweed Authors
2 //
3 // Licensed under the Apache License, Version 2.0 (the "License"); you may not
4 // use this file except in compliance with the License. You may obtain a copy of
5 // the License at
6 //
7 // https://www.apache.org/licenses/LICENSE-2.0
8 //
9 // Unless required by applicable law or agreed to in writing, software
10 // distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
11 // WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
12 // License for the specific language governing permissions and limitations under
13 // the License.
14
15 #include "pw_allocator/layout.h"
16
17 #include <cstddef>
18
19 #include "pw_result/result.h"
20 #include "pw_status/status.h"
21 #include "pw_unit_test/framework.h"
22
23 namespace {
24
25 using ::pw::allocator::Layout;
26
TEST(LayoutTest,DefaultConstructor)27 TEST(LayoutTest, DefaultConstructor) {
28 Layout layout;
29 EXPECT_EQ(layout.size(), 0U);
30 EXPECT_EQ(layout.alignment(), alignof(std::max_align_t));
31 }
32
TEST(LayoutTest,SizeOnlyConstructor)33 TEST(LayoutTest, SizeOnlyConstructor) {
34 constexpr size_t kSize = 512;
35 Layout layout(kSize);
36 EXPECT_EQ(layout.size(), kSize);
37 EXPECT_EQ(layout.alignment(), alignof(std::max_align_t));
38 }
39
TEST(LayoutTest,FullConstructor)40 TEST(LayoutTest, FullConstructor) {
41 constexpr size_t kSize = 2048;
42 constexpr size_t kAlignment = 4;
43 Layout layout(kSize, kAlignment);
44 EXPECT_EQ(layout.size(), kSize);
45 EXPECT_EQ(layout.alignment(), kAlignment);
46 }
47
TEST(LayoutTest,ConstructUsingInitializer)48 TEST(LayoutTest, ConstructUsingInitializer) {
49 constexpr size_t kSize = 1024;
50 constexpr size_t kAlignment = 8;
51 Layout layout = {kSize, kAlignment};
52 EXPECT_EQ(layout.size(), kSize);
53 EXPECT_EQ(layout.alignment(), kAlignment);
54 }
55
TEST(LayoutTest,ConstructFromType)56 TEST(LayoutTest, ConstructFromType) {
57 struct Values {
58 uint8_t u8;
59 uint16_t u16;
60 uint32_t u32;
61 };
62 Layout layout = Layout::Of<Values>();
63 EXPECT_EQ(layout.size(), sizeof(Values));
64 EXPECT_EQ(layout.alignment(), alignof(Values));
65 }
66
TEST(LayoutTest,Extend)67 TEST(LayoutTest, Extend) {
68 constexpr size_t kSize1 = 2048;
69 constexpr size_t kSize2 = 1024;
70 constexpr size_t kAlignment = 2;
71 Layout layout1 = {kSize1, kAlignment};
72 EXPECT_EQ(layout1.size(), kSize1);
73 EXPECT_EQ(layout1.alignment(), kAlignment);
74
75 Layout layout2 = layout1.Extend(kSize2);
76 EXPECT_EQ(layout2.size(), kSize1 + kSize2);
77 EXPECT_EQ(layout2.alignment(), kAlignment);
78 }
79
TEST(LayoutTest,UnwrapOk)80 TEST(LayoutTest, UnwrapOk) {
81 constexpr size_t kSize = 1024;
82 constexpr size_t kAlignment = 8;
83 pw::Result<Layout> result = Layout(kSize, kAlignment);
84 Layout layout = Layout::Unwrap(result);
85 EXPECT_EQ(layout.size(), kSize);
86 EXPECT_EQ(layout.alignment(), kAlignment);
87 }
88
TEST(LayoutTest,UnwrapError)89 TEST(LayoutTest, UnwrapError) {
90 pw::Result<Layout> result = pw::Status::Unimplemented();
91 Layout layout = Layout::Unwrap(result);
92 EXPECT_EQ(layout.size(), 0U);
93 EXPECT_EQ(layout.alignment(), alignof(std::max_align_t));
94 }
95
96 } // namespace
97