]>
Commit | Line | Data |
---|---|---|
9f95a23c TL |
1 | /* |
2 | * This file is open source software, licensed to you under the terms | |
3 | * of the Apache License, Version 2.0 (the "License"). See the NOTICE file | |
4 | * distributed with this work for additional information regarding copyright | |
5 | * ownership. You may not use this file except in compliance with the License. | |
6 | * | |
7 | * You may obtain a copy of the License at | |
8 | * | |
9 | * http://www.apache.org/licenses/LICENSE-2.0 | |
10 | * | |
11 | * Unless required by applicable law or agreed to in writing, | |
12 | * software distributed under the License is distributed on an | |
13 | * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY | |
14 | * KIND, either express or implied. See the License for the | |
15 | * specific language governing permissions and limitations | |
16 | * under the License. | |
17 | */ | |
18 | /* | |
19 | * Copyright (C) 2018 Scylladb, Ltd. | |
20 | */ | |
21 | ||
22 | #include <seastar/testing/thread_test_case.hh> | |
23 | ||
24 | #include <seastar/core/sharded.hh> | |
25 | ||
26 | using namespace seastar; | |
27 | ||
28 | namespace { | |
29 | class invoke_on_during_stop final : public peering_sharded_service<invoke_on_during_stop> { | |
30 | bool flag = false; | |
31 | ||
32 | public: | |
33 | future<> stop() { | |
34 | return container().invoke_on(0, [] (invoke_on_during_stop& instance) { | |
35 | instance.flag = true; | |
36 | }); | |
37 | } | |
38 | ||
39 | ~invoke_on_during_stop() { | |
40 | if (engine().cpu_id() == 0) { | |
41 | assert(flag); | |
42 | } | |
43 | } | |
44 | }; | |
45 | } | |
46 | ||
47 | SEASTAR_THREAD_TEST_CASE(invoke_on_during_stop_test) { | |
48 | sharded<invoke_on_during_stop> s; | |
49 | s.start().get(); | |
50 | s.stop().get(); | |
51 | } | |
52 | ||
53 | class mydata { | |
54 | public: | |
55 | int x = 1; | |
56 | future<> stop() { | |
57 | return make_ready_future<>(); | |
58 | } | |
59 | }; | |
60 | ||
61 | SEASTAR_THREAD_TEST_CASE(invoke_map_returns_non_future_value) { | |
62 | seastar::sharded<mydata> s; | |
63 | s.start().get(); | |
64 | s.map([] (mydata& m) { | |
65 | return m.x; | |
66 | }).then([] (std::vector<int> results) { | |
67 | for (auto& x : results) { | |
68 | assert(x == 1); | |
69 | } | |
70 | }).get(); | |
71 | s.stop().get(); | |
72 | }; | |
73 | ||
74 | SEASTAR_THREAD_TEST_CASE(invoke_map_returns_future_value) { | |
75 | seastar::sharded<mydata> s; | |
76 | s.start().get(); | |
77 | s.map([] (mydata& m) { | |
78 | return make_ready_future<int>(m.x); | |
79 | }).then([] (std::vector<int> results) { | |
80 | for (auto& x : results) { | |
81 | assert(x == 1); | |
82 | } | |
83 | }).get(); | |
84 | s.stop().get(); | |
85 | } | |
86 | ||
87 | SEASTAR_THREAD_TEST_CASE(invoke_map_returns_future_value_from_thread) { | |
88 | seastar::sharded<mydata> s; | |
89 | s.start().get(); | |
90 | s.map([] (mydata& m) { | |
91 | return seastar::async([&m] { | |
92 | return m.x; | |
93 | }); | |
94 | }).then([] (std::vector<int> results) { | |
95 | for (auto& x : results) { | |
96 | assert(x == 1); | |
97 | } | |
98 | }).get(); | |
99 | s.stop().get(); | |
100 | } |