blob: 0d5a8c9f619a567d2e75b4e6b74ee96d4a221ca4 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
|
// -*- C++ -*-
//===----------------------------------------------------------------------===//
//
// The LLVM Compiler Infrastructure
//
// This file is dual licensed under the MIT and the University of Illinois Open
// Source Licenses. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
// UNSUPPORTED: c++98, c++03, c++11, c++14
#include <experimental/task>
#include <string>
#include <vector>
#include <memory>
#include "../sync_wait.hpp"
void test_returning_move_only_type()
{
auto move_only_async =
[](bool x) -> std::experimental::task<std::unique_ptr<int>> {
if (x) {
auto p = std::make_unique<int>(123);
co_return p; // Should be implicit std::move(p) here.
}
co_return std::make_unique<int>(456);
};
assert(*sync_wait(move_only_async(true)) == 123);
assert(*sync_wait(move_only_async(false)) == 456);
}
void test_co_return_with_curly_braces()
{
auto t = []() -> std::experimental::task<std::tuple<int, std::string>>
{
co_return { 123, "test" };
}();
auto result = sync_wait(std::move(t));
assert(std::get<0>(result) == 123);
assert(std::get<1>(result) == "test");
}
void test_co_return_by_initialiser_list()
{
auto t = []() -> std::experimental::task<std::vector<int>>
{
co_return { 2, 10, -1 };
}();
auto result = sync_wait(std::move(t));
assert(result.size() == 3);
assert(result[0] == 2);
assert(result[1] == 10);
assert(result[2] == -1);
}
int main()
{
test_returning_move_only_type();
test_co_return_with_curly_braces();
test_co_return_by_initialiser_list();
return 0;
}
|
