[apple/xnu.git] / tests / intrusive_shared_ptr_src / assign.move.cpp

//
// Tests for
//  template <typename U>
//  intrusive_shared_ptr& operator=(intrusive_shared_ptr<U, RefcountPolicy>&& other);
//
//  intrusive_shared_ptr& operator=(intrusive_shared_ptr&& other);
//

#include <libkern/c++/intrusive_shared_ptr.h>
#include <darwintest.h>
#include <type_traits>
#include <utility>
#include "test_policy.h"

struct Base { int i; };
struct Derived : Base { };

struct Base1 { int i; };
struct Base2 { long l; };
struct DerivedMultiple : Base1, Base2 {
        DerivedMultiple(int i) : Base1{i}, Base2{i + 10}
        {
        }
};

struct Unrelated { };

template <typename Stored, typename From, typename To>
static void
tests()
{
        Stored obj1{1};
        Stored obj2{2};

        // Move-assign non-null to non-null
        {
                tracked_shared_ptr<From> from(&obj1, libkern::retain);
                tracked_shared_ptr<To> to(&obj2, libkern::retain);
                tracking_policy::reset();
                tracked_shared_ptr<To>& ref = (to = std::move(from));
                CHECK(tracking_policy::releases == 1);
                CHECK(tracking_policy::retains == 0);
                CHECK(&ref == &to);
                CHECK(from.get() == nullptr);
                CHECK(to.get() == &obj1);
        }

        // Move-assign non-null to null
        {
                tracked_shared_ptr<From> from(&obj1, libkern::retain);
                tracked_shared_ptr<To> to = nullptr;
                tracking_policy::reset();
                tracked_shared_ptr<To>& ref = (to = std::move(from));
                CHECK(tracking_policy::releases == 0);
                CHECK(tracking_policy::retains == 0);
                CHECK(&ref == &to);
                CHECK(from.get() == nullptr);
                CHECK(to.get() == &obj1);
        }

        // Move-assign null to non-null
        {
                tracked_shared_ptr<From> from = nullptr;
                tracked_shared_ptr<To> to(&obj2, libkern::retain);
                tracking_policy::reset();
                tracked_shared_ptr<To>& ref = (to = std::move(from));
                CHECK(tracking_policy::releases == 1);
                CHECK(tracking_policy::retains == 0);
                CHECK(&ref == &to);
                CHECK(from.get() == nullptr);
                CHECK(to.get() == nullptr);
        }

        // Move-assign null to null
        {
                tracked_shared_ptr<From> from = nullptr;
                tracked_shared_ptr<To> to = nullptr;
                tracking_policy::reset();
                tracked_shared_ptr<To>& ref = (to = std::move(from));
                CHECK(tracking_policy::releases == 0);
                CHECK(tracking_policy::retains == 0);
                CHECK(&ref == &to);
                CHECK(from.get() == nullptr);
                CHECK(to.get() == nullptr);
        }
}

T_DECL(assign_move, "intrusive_shared_ptr.assign.move") {
        tests</*stored*/ Derived, /*from*/ Derived, /*to*/ Derived>();
        tests</*stored*/ Derived, /*from*/ Derived, /*to*/ Derived const>();
        tests</*stored*/ Derived, /*from*/ Derived const, /*to*/ Derived const>();

        tests</*stored*/ Derived, /*from*/ Derived, /*to*/ Base>();
        tests</*stored*/ Derived, /*from*/ Derived, /*to*/ Base const>();
        tests</*stored*/ Derived, /*from*/ Derived const, /*to*/ Base const>();

        tests</*stored*/ DerivedMultiple, /*from*/ DerivedMultiple, /*to*/ Base1>();
        tests</*stored*/ DerivedMultiple, /*from*/ DerivedMultiple const, /*to*/ Base1 const>();

        tests</*stored*/ DerivedMultiple, /*from*/ DerivedMultiple, /*to*/ Base2>();
        tests</*stored*/ DerivedMultiple, /*from*/ DerivedMultiple const, /*to*/ Base2 const>();

        // Make sure basic trait querying works
        static_assert(std::is_move_assignable_v<test_shared_ptr<Derived> >);

        // Make sure downcasts are disabled
        static_assert(!std::is_assignable_v</*to*/ test_shared_ptr<Derived>, /*from*/ test_shared_ptr<Base>&&>);
        static_assert(!std::is_assignable_v</*to*/ test_shared_ptr<DerivedMultiple>, /*from*/ test_shared_ptr<Base1>&&>);
        static_assert(!std::is_assignable_v</*to*/ test_shared_ptr<DerivedMultiple>, /*from*/ test_shared_ptr<Base2>&&>);
        static_assert(!std::is_assignable_v</*to*/ test_shared_ptr<Base2>, /*from*/ test_shared_ptr<Base1>&&>);

        // Make sure const-casting away doesn't work
        static_assert(!std::is_assignable_v</*to*/ test_shared_ptr<Derived>, /*from*/ test_shared_ptr<Derived const>&&>);

        // Make sure casting to unrelated types doesn't work
        static_assert(!std::is_assignable_v</*to*/ test_shared_ptr<char>, /*from*/ test_shared_ptr<Derived>&&>);
        static_assert(!std::is_assignable_v</*to*/ test_shared_ptr<Unrelated>, /*from*/ test_shared_ptr<Derived>&&>);
        static_assert(!std::is_assignable_v</*to*/ test_shared_ptr<Base2>, /*from*/ test_shared_ptr<Base1>&&>);

        // Make sure constructing with different policies doesn't work
        static_assert(!std::is_assignable_v</*to*/ libkern::intrusive_shared_ptr<Derived, dummy_policy<2> >, /*from*/ libkern::intrusive_shared_ptr<Derived, dummy_policy<1> >&&>);
}