third_party/llvm-project/libcxx/test/std/ranges/range.adaptors/range.take.while/base.pass.cpp

//===----------------------------------------------------------------------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//

// UNSUPPORTED: c++03, c++11, c++14, c++17

// constexpr V base() const & requires copy_constructible<V> { return base_; }
// constexpr V base() && { return std::move(base_); }

#include <cassert>
#include <ranges>
#include <type_traits>
#include <utility>

#include "MoveOnly.h"
#include "types.h"

struct View : std::ranges::view_interface<View> {
  int i;
  int* begin() const;
  int* end() const;
};

struct MoveOnlyView : View {
  MoveOnly mo;
};

template <class T>
concept HasBase = requires(T&& t) { std::forward<T>(t).base(); };

struct Pred {
  constexpr bool operator()(int i) const { return i > 5; }
};

static_assert(HasBase<std::ranges::take_while_view<View, Pred> const&>);
static_assert(HasBase<std::ranges::take_while_view<View, Pred>&&>);

static_assert(!HasBase<std::ranges::take_while_view<MoveOnlyView, Pred> const&>);
static_assert(HasBase<std::ranges::take_while_view<MoveOnlyView, Pred>&&>);

constexpr bool test() {
  // const &
  {
    const std::ranges::take_while_view<View, Pred> twv{View{{}, 5}, {}};
    std::same_as<View> decltype(auto) v = twv.base();
    assert(v.i == 5);
  }

  // &
  {
    std::ranges::take_while_view<View, Pred> twv{View{{}, 5}, {}};
    std::same_as<View> decltype(auto) v = twv.base();
    assert(v.i == 5);
  }

  // &&
  {
    std::ranges::take_while_view<View, Pred> twv{View{{}, 5}, {}};
    std::same_as<View> decltype(auto) v = std::move(twv).base();
    assert(v.i == 5);
  }

  // const &&
  {
    const std::ranges::take_while_view<View, Pred> twv{View{{}, 5}, {}};
    std::same_as<View> decltype(auto) v = std::move(twv).base();
    assert(v.i == 5);
  }

  // move only
  {
    std::ranges::take_while_view<MoveOnlyView, Pred> twv{MoveOnlyView{{}, 5}, {}};
    std::same_as<MoveOnlyView> decltype(auto) v = std::move(twv).base();
    assert(v.mo.get() == 5);
  }
  return true;
}

int main(int, char**) {
  test();
  static_assert(test());
  return 0;
}