azalea/kotlin-fork
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[K/N] Refactor allocators ^KT-52130

Browse Source 
Merge-request: KT-MR-6135
Merged-by: Alexander Shabalin <Alexander.Shabalin@jetbrains.com>
This commit is contained in:
Alexander Shabalin
2022-04-27 14:00:27 +00:00
committed by Space
parent a05bb3c77c
commit 663e843544
49 changed files with 1345 additions and 500 deletions
Download Patch File Download Diff File Expand all files Collapse all files
kotlin-native/runtime/src/main/cpp/MultiSourceQueueTest.cpp
+42
View File
@@ -10,12 +10,15 @@
#include "gmock/gmock.h"
#include "gtest/gtest.h"
#include "AllocatorTestSupport.hpp"
#include "ScopedThread.hpp"
#include "TestSupport.hpp"
#include "Types.h"
using namespace kotlin;
using ::testing::_;
namespace {
template <typename T, typename Mutex>
@@ -312,3 +315,42 @@ TEST(MultiSourceQueueTest, ConcurrentPublishAndApplyDeletions) {
auto actual = Collect(queue);
EXPECT_THAT(actual, testing::IsEmpty());
}
TEST(MultiSourceQueueTest, CustomAllocator) {
testing::StrictMock<test_support::SpyAllocatorCore> allocatorCore;
auto allocator = test_support::MakeAllocator<int>(allocatorCore);
using Queue = MultiSourceQueue<int, SpinLock<MutexThreadStateHandling::kIgnore>, decltype(allocator)>;
Queue queue(allocator);
Queue::Producer producer1(queue);
Queue::Producer producer2(queue);
EXPECT_CALL(allocatorCore, allocate(_)).Times(5);
auto* node11 = producer1.Insert(1);
auto* node12 = producer1.Insert(2);
auto* node21 = producer2.Insert(10);
auto* node22 = producer2.Insert(20);
auto* node23 = producer2.Insert(30);
testing::Mock::VerifyAndClearExpectations(&allocatorCore);
EXPECT_CALL(allocatorCore, deallocate(_, _));
producer2.Erase(node22);
testing::Mock::VerifyAndClearExpectations(&allocatorCore);
producer1.Publish();
producer2.Publish();
EXPECT_CALL(allocatorCore, allocate(_)).Times(4);
producer1.Erase(node11);
producer1.Erase(node23);
producer2.Erase(node12);
producer2.Erase(node21);
testing::Mock::VerifyAndClearExpectations(&allocatorCore);
producer1.Publish();
producer2.Publish();
EXPECT_CALL(allocatorCore, deallocate(_, _)).Times(8);
queue.ApplyDeletions();
testing::Mock::VerifyAndClearExpectations(&allocatorCore);
}