test_tracking_allocator.cpp

Go to the documentation of this file.

#include "gtest/gtest.h"
 
#include "kokkos-execution/stdexec.hpp"
 
#include "tests/utils/tracking_allocator.hpp"

 
namespace Tests {

TEST(TrackingAllocatorTest, write_read_env) {
    std::atomic<size_t> count = 0;
 
    stdexec::sender auto sndr =
        stdexec::read_env(stdexec::get_allocator) | stdexec::then([](auto allocator) {
            int value = 42;
            return Tests::Utils::round_trip_allocate(allocator, static_cast<int&>(value));
        })
        | stdexec::write_env(stdexec::prop{stdexec::get_allocator, Tests::Utils::TrackingAllocator<int>{&count}});
 
    const auto [val] = stdexec::sync_wait(std::move(sndr)).value(); // NOLINT(performance-move-const-arg)
 
    ASSERT_EQ(val, 42);
    ASSERT_EQ(count, 1);
}

TEST(TrackingAllocatorTest, forwarded_through_then) {
    static_assert(stdexec::forwarding_query(stdexec::get_allocator));
 
    std::atomic<size_t> count = 0;
 
    stdexec::sender auto sndr =
        stdexec::read_env(stdexec::get_allocator)
        | stdexec::then([](auto allocator) { return Tests::Utils::round_trip_allocate(allocator, 42); })
        | stdexec::then([](auto&& value) { return value; })
        | stdexec::write_env(stdexec::prop{stdexec::get_allocator, Tests::Utils::TrackingAllocator<int>{&count}});
 
    const auto [val] = stdexec::sync_wait(std::move(sndr)).value(); // NOLINT(performance-move-const-arg)
 
    ASSERT_EQ(val, 42);
    ASSERT_EQ(count, 1);
}
 
} // namespace Tests