151c0b2f7Stbbdev /*
251c0b2f7Stbbdev     Copyright (c) 2020 Intel Corporation
351c0b2f7Stbbdev 
451c0b2f7Stbbdev     Licensed under the Apache License, Version 2.0 (the "License");
551c0b2f7Stbbdev     you may not use this file except in compliance with the License.
651c0b2f7Stbbdev     You may obtain a copy of the License at
751c0b2f7Stbbdev 
851c0b2f7Stbbdev         http://www.apache.org/licenses/LICENSE-2.0
951c0b2f7Stbbdev 
1051c0b2f7Stbbdev     Unless required by applicable law or agreed to in writing, software
1151c0b2f7Stbbdev     distributed under the License is distributed on an "AS IS" BASIS,
1251c0b2f7Stbbdev     WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
1351c0b2f7Stbbdev     See the License for the specific language governing permissions and
1451c0b2f7Stbbdev     limitations under the License.
1551c0b2f7Stbbdev */
1651c0b2f7Stbbdev 
1751c0b2f7Stbbdev 
1851c0b2f7Stbbdev #include "common/test.h"
1951c0b2f7Stbbdev 
2051c0b2f7Stbbdev #include "common/utils.h"
2151c0b2f7Stbbdev #include "common/graph_utils.h"
2251c0b2f7Stbbdev 
23*49e08aacStbbdev #include "oneapi/tbb/flow_graph.h"
24*49e08aacStbbdev #include "oneapi/tbb/task_arena.h"
25*49e08aacStbbdev #include "oneapi/tbb/global_control.h"
2651c0b2f7Stbbdev 
2751c0b2f7Stbbdev #include "conformance_flowgraph.h"
2851c0b2f7Stbbdev 
2951c0b2f7Stbbdev //! \file conformance_sequencer_node.cpp
3051c0b2f7Stbbdev //! \brief Test for [flow_graph.sequencer_node] specification
3151c0b2f7Stbbdev 
3251c0b2f7Stbbdev /*
3351c0b2f7Stbbdev TODO: implement missing conformance tests for sequencer_node:
3451c0b2f7Stbbdev   - [ ] The copy constructor and copy assignment are called for the node's type template parameter.
3551c0b2f7Stbbdev   - [ ] Explicit test that `Sequencer' requirements are necessary.
3651c0b2f7Stbbdev   - [ ] Write tests for the constructors.
3751c0b2f7Stbbdev   - [ ] Add CTAD test.
3851c0b2f7Stbbdev   - [ ] Improve `test_buffering' by checking that additional `try_get()' does not receive the same
3951c0b2f7Stbbdev     value.
4051c0b2f7Stbbdev   - [ ] Add explicit test on the example from the specification.
4151c0b2f7Stbbdev */
4251c0b2f7Stbbdev 
4351c0b2f7Stbbdev template<typename T>
4451c0b2f7Stbbdev void test_inheritance(){
45*49e08aacStbbdev     using namespace oneapi::tbb::flow;
4651c0b2f7Stbbdev 
4751c0b2f7Stbbdev     CHECK_MESSAGE( (std::is_base_of<graph_node, sequencer_node<T>>::value), "sequencer_node should be derived from graph_node");
4851c0b2f7Stbbdev     CHECK_MESSAGE( (std::is_base_of<receiver<T>, sequencer_node<T>>::value), "sequencer_node should be derived from receiver<T>");
4951c0b2f7Stbbdev     CHECK_MESSAGE( (std::is_base_of<sender<T>, sequencer_node<T>>::value), "sequencer_node should be derived from sender<T>");
5051c0b2f7Stbbdev }
5151c0b2f7Stbbdev 
5251c0b2f7Stbbdev template<typename T>
5351c0b2f7Stbbdev struct id_sequencer{
5451c0b2f7Stbbdev     using input_type = T;
5551c0b2f7Stbbdev 
5651c0b2f7Stbbdev     T operator()(T v){
5751c0b2f7Stbbdev         return v;
5851c0b2f7Stbbdev     }
5951c0b2f7Stbbdev };
6051c0b2f7Stbbdev 
6151c0b2f7Stbbdev void test_copies(){
62*49e08aacStbbdev     using namespace oneapi::tbb::flow;
6351c0b2f7Stbbdev 
6451c0b2f7Stbbdev     graph g;
6551c0b2f7Stbbdev     id_sequencer<int> sequencer;
6651c0b2f7Stbbdev 
6751c0b2f7Stbbdev     sequencer_node<int> n(g, sequencer);
6851c0b2f7Stbbdev     sequencer_node<int> n2(n);
6951c0b2f7Stbbdev 
7051c0b2f7Stbbdev }
7151c0b2f7Stbbdev 
7251c0b2f7Stbbdev void test_buffering(){
73*49e08aacStbbdev     oneapi::tbb::flow::graph g;
7451c0b2f7Stbbdev 
7551c0b2f7Stbbdev     id_sequencer<int> sequencer;
7651c0b2f7Stbbdev 
77*49e08aacStbbdev     oneapi::tbb::flow::sequencer_node<int> node(g, sequencer);
78*49e08aacStbbdev     oneapi::tbb::flow::limiter_node<int> rejecter(g, 0);
7951c0b2f7Stbbdev 
80*49e08aacStbbdev     oneapi::tbb::flow::make_edge(node, rejecter);
8151c0b2f7Stbbdev     node.try_put(1);
8251c0b2f7Stbbdev     g.wait_for_all();
8351c0b2f7Stbbdev 
8451c0b2f7Stbbdev     int tmp = -1;
8551c0b2f7Stbbdev     CHECK_MESSAGE( (node.try_get(tmp) == false), "try_get after rejection should not succeed");
8651c0b2f7Stbbdev     CHECK_MESSAGE( (tmp == -1), "try_get after rejection should not set value");
8751c0b2f7Stbbdev }
8851c0b2f7Stbbdev 
8951c0b2f7Stbbdev #if __TBB_CPP17_DEDUCTION_GUIDES_PRESENT
9051c0b2f7Stbbdev void test_deduction_guides(){
91*49e08aacStbbdev     // oneapi::tbb::flow::graph g;
9251c0b2f7Stbbdev     // id_sequencer<int> sequ;
93*49e08aacStbbdev     // oneapi::tbb::flow::sequencer_node node1(g, sequ);
9451c0b2f7Stbbdev }
9551c0b2f7Stbbdev #endif
9651c0b2f7Stbbdev 
9751c0b2f7Stbbdev void test_forwarding(){
98*49e08aacStbbdev     oneapi::tbb::flow::graph g;
9951c0b2f7Stbbdev     id_sequencer<int> sequencer;
10051c0b2f7Stbbdev 
101*49e08aacStbbdev     oneapi::tbb::flow::sequencer_node<int> node1(g, sequencer);
10251c0b2f7Stbbdev     test_push_receiver<int> node2(g);
10351c0b2f7Stbbdev     test_push_receiver<int> node3(g);
10451c0b2f7Stbbdev 
105*49e08aacStbbdev     oneapi::tbb::flow::make_edge(node1, node2);
106*49e08aacStbbdev     oneapi::tbb::flow::make_edge(node1, node3);
10751c0b2f7Stbbdev 
10851c0b2f7Stbbdev     node1.try_put(0);
10951c0b2f7Stbbdev 
11051c0b2f7Stbbdev     g.wait_for_all();
11151c0b2f7Stbbdev 
11251c0b2f7Stbbdev     int c2 = get_count(node2), c3 = get_count(node3);
11351c0b2f7Stbbdev     CHECK_MESSAGE( (c2 != c3 ), "Only one descendant the node needs to receive");
11451c0b2f7Stbbdev     CHECK_MESSAGE( (c2 + c3 == 1 ), "Messages need to be received");
11551c0b2f7Stbbdev }
11651c0b2f7Stbbdev 
11751c0b2f7Stbbdev void test_sequencer(){
118*49e08aacStbbdev     oneapi::tbb::flow::graph g;
11951c0b2f7Stbbdev     id_sequencer<int> sequencer;
12051c0b2f7Stbbdev 
121*49e08aacStbbdev     oneapi::tbb::flow::sequencer_node<int> node(g, sequencer);
12251c0b2f7Stbbdev 
12351c0b2f7Stbbdev     node.try_put(1);
12451c0b2f7Stbbdev     node.try_put(0);
12551c0b2f7Stbbdev     node.try_put(1);
12651c0b2f7Stbbdev     g.wait_for_all();
12751c0b2f7Stbbdev 
12851c0b2f7Stbbdev     int tmp = -1;
12951c0b2f7Stbbdev     CHECK_MESSAGE((node.try_get(tmp) == true), "Getting from sequencer should succeed");
13051c0b2f7Stbbdev     CHECK_MESSAGE((tmp == 0), "Received value should be correct");
13151c0b2f7Stbbdev 
13251c0b2f7Stbbdev     tmp = -1;
13351c0b2f7Stbbdev     CHECK_MESSAGE((node.try_get(tmp) == true), "Getting from sequencer should succeed");
13451c0b2f7Stbbdev     CHECK_MESSAGE((tmp == 1), "Received value should be correct");
13551c0b2f7Stbbdev 
13651c0b2f7Stbbdev     tmp = -1;
13751c0b2f7Stbbdev     CHECK_MESSAGE((node.try_get(tmp) == false), "Getting from sequencer should not succeed");
13851c0b2f7Stbbdev 
13951c0b2f7Stbbdev }
14051c0b2f7Stbbdev 
14151c0b2f7Stbbdev //! Test function_node buffering
14251c0b2f7Stbbdev //! \brief \ref requirement
14351c0b2f7Stbbdev TEST_CASE("sequencer_node buffering"){
14451c0b2f7Stbbdev     test_sequencer();
14551c0b2f7Stbbdev }
14651c0b2f7Stbbdev 
14751c0b2f7Stbbdev //! Test function_node buffering
14851c0b2f7Stbbdev //! \brief \ref requirement
14951c0b2f7Stbbdev TEST_CASE("sequencer_node buffering"){
15051c0b2f7Stbbdev     test_forwarding();
15151c0b2f7Stbbdev }
15251c0b2f7Stbbdev 
15351c0b2f7Stbbdev //! Test deduction guides
15451c0b2f7Stbbdev //! \brief \ref interface \ref requirement
15551c0b2f7Stbbdev TEST_CASE("Deduction guides"){
15651c0b2f7Stbbdev #if __TBB_CPP17_DEDUCTION_GUIDES_PRESENT
15751c0b2f7Stbbdev     test_deduction_guides();
15851c0b2f7Stbbdev #endif
15951c0b2f7Stbbdev }
16051c0b2f7Stbbdev 
16151c0b2f7Stbbdev //! Test priority_queue_node buffering
16251c0b2f7Stbbdev //! \brief \ref requirement
16351c0b2f7Stbbdev TEST_CASE("sequencer_node buffering"){
16451c0b2f7Stbbdev     test_buffering();
16551c0b2f7Stbbdev }
16651c0b2f7Stbbdev 
16751c0b2f7Stbbdev //! Test copy constructor
16851c0b2f7Stbbdev //! \brief \ref interface
16951c0b2f7Stbbdev TEST_CASE("sequencer_node copy constructor"){
17051c0b2f7Stbbdev     test_copies();
17151c0b2f7Stbbdev }
17251c0b2f7Stbbdev 
17351c0b2f7Stbbdev //! Test inheritance relations
17451c0b2f7Stbbdev //! \brief \ref interface
17551c0b2f7Stbbdev TEST_CASE("sequencer_node superclasses"){
17651c0b2f7Stbbdev     test_inheritance<int>();
17751c0b2f7Stbbdev     test_inheritance<void*>();
17851c0b2f7Stbbdev }
17951c0b2f7Stbbdev 
180