151c0b2f7Stbbdev /*
2*b15aabb3Stbbdev     Copyright (c) 2020-2021 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 
17*b15aabb3Stbbdev #if __INTEL_COMPILER && _MSC_VER
18*b15aabb3Stbbdev #pragma warning(disable : 2586) // decorated name length exceeded, name was truncated
19*b15aabb3Stbbdev #endif
2051c0b2f7Stbbdev 
2151c0b2f7Stbbdev #include "common/test.h"
2251c0b2f7Stbbdev 
2351c0b2f7Stbbdev #include "common/utils.h"
2451c0b2f7Stbbdev #include "common/graph_utils.h"
2551c0b2f7Stbbdev 
2649e08aacStbbdev #include "oneapi/tbb/flow_graph.h"
2749e08aacStbbdev #include "oneapi/tbb/task_arena.h"
2849e08aacStbbdev #include "oneapi/tbb/global_control.h"
2951c0b2f7Stbbdev 
3051c0b2f7Stbbdev #include "conformance_flowgraph.h"
3151c0b2f7Stbbdev 
3251c0b2f7Stbbdev //! \file conformance_buffer_node.cpp
3351c0b2f7Stbbdev //! \brief Test for [flow_graph.buffer_node] specification
3451c0b2f7Stbbdev 
3551c0b2f7Stbbdev /*
3651c0b2f7Stbbdev TODO: implement missing conformance tests for buffer_node:
3751c0b2f7Stbbdev   - [ ] The copy constructor is called for the node's type template parameter.
3851c0b2f7Stbbdev   - [ ] Improve `test_forwarding' by checking that the value passed is the actual one received.
3951c0b2f7Stbbdev   - [ ] Improve `test_buffering' by checking that additional `try_get()' does not receive the same
4051c0b2f7Stbbdev     value.
4151c0b2f7Stbbdev   - [ ] Improve tests of the constructors.
4251c0b2f7Stbbdev   - [ ] Based on the decision about the details for `try_put()' and `try_get()' write corresponding
4351c0b2f7Stbbdev     tests.
4451c0b2f7Stbbdev   - [ ] Fix description in `TEST_CASEs'.*/
4551c0b2f7Stbbdev template<typename T>
4651c0b2f7Stbbdev void test_inheritance(){
4749e08aacStbbdev     using namespace oneapi::tbb::flow;
4851c0b2f7Stbbdev 
4951c0b2f7Stbbdev     CHECK_MESSAGE( (std::is_base_of<graph_node, buffer_node<T>>::value), "buffer_node should be derived from graph_node");
5051c0b2f7Stbbdev     CHECK_MESSAGE( (std::is_base_of<receiver<T>, buffer_node<T>>::value), "buffer_node should be derived from receiver<T>");
5151c0b2f7Stbbdev     CHECK_MESSAGE( (std::is_base_of<sender<T>, buffer_node<T>>::value), "buffer_node should be derived from sender<T>");
5251c0b2f7Stbbdev }
5351c0b2f7Stbbdev 
5451c0b2f7Stbbdev void test_copies(){
5549e08aacStbbdev     using namespace oneapi::tbb::flow;
5651c0b2f7Stbbdev 
5751c0b2f7Stbbdev     graph g;
5851c0b2f7Stbbdev     buffer_node<int> n(g);
5951c0b2f7Stbbdev     buffer_node<int> n2(n);
6051c0b2f7Stbbdev 
6151c0b2f7Stbbdev }
6251c0b2f7Stbbdev 
6351c0b2f7Stbbdev void test_buffering(){
6449e08aacStbbdev     oneapi::tbb::flow::graph g;
6551c0b2f7Stbbdev 
6649e08aacStbbdev     oneapi::tbb::flow::buffer_node<int> node(g);
6749e08aacStbbdev     oneapi::tbb::flow::limiter_node<int> rejecter(g, 0);
6851c0b2f7Stbbdev 
6949e08aacStbbdev     oneapi::tbb::flow::make_edge(node, rejecter);
7051c0b2f7Stbbdev 
7151c0b2f7Stbbdev     int tmp = -1;
7251c0b2f7Stbbdev     CHECK_MESSAGE( (node.try_get(tmp) == false), "try_get before placemnt should not succeed");
7351c0b2f7Stbbdev 
7451c0b2f7Stbbdev     node.try_put(1);
7551c0b2f7Stbbdev 
7651c0b2f7Stbbdev     tmp = -1;
7751c0b2f7Stbbdev     CHECK_MESSAGE( (node.try_get(tmp) == true), "try_get after rejection should succeed");
7851c0b2f7Stbbdev     CHECK_MESSAGE( (tmp == 1), "try_get after rejection should set value");
7951c0b2f7Stbbdev     g.wait_for_all();
8051c0b2f7Stbbdev }
8151c0b2f7Stbbdev 
8251c0b2f7Stbbdev void test_forwarding(){
8349e08aacStbbdev     oneapi::tbb::flow::graph g;
8451c0b2f7Stbbdev 
8549e08aacStbbdev     oneapi::tbb::flow::buffer_node<int> node1(g);
8651c0b2f7Stbbdev     test_push_receiver<int> node2(g);
8751c0b2f7Stbbdev     test_push_receiver<int> node3(g);
8851c0b2f7Stbbdev 
8949e08aacStbbdev     oneapi::tbb::flow::make_edge(node1, node2);
9049e08aacStbbdev     oneapi::tbb::flow::make_edge(node1, node3);
9151c0b2f7Stbbdev 
9251c0b2f7Stbbdev     node1.try_put(1);
9351c0b2f7Stbbdev     g.wait_for_all();
9451c0b2f7Stbbdev 
9551c0b2f7Stbbdev     int c2 = get_count(node2), c3 = get_count(node3);
9651c0b2f7Stbbdev     CHECK_MESSAGE( (c2 != c3 ), "Only one descendant the node needs to receive");
9751c0b2f7Stbbdev     CHECK_MESSAGE( (c2 + c3 == 1 ), "All messages need to be received");
9851c0b2f7Stbbdev }
9951c0b2f7Stbbdev 
10051c0b2f7Stbbdev //! Test function_node buffering
10151c0b2f7Stbbdev //! \brief \ref requirement
10251c0b2f7Stbbdev TEST_CASE("buffer_node buffering"){
10351c0b2f7Stbbdev     test_forwarding();
10451c0b2f7Stbbdev }
10551c0b2f7Stbbdev 
10651c0b2f7Stbbdev //! Test function_node buffering
10751c0b2f7Stbbdev //! \brief \ref requirement
10851c0b2f7Stbbdev TEST_CASE("buffer_node buffering"){
10951c0b2f7Stbbdev     test_buffering();
11051c0b2f7Stbbdev }
11151c0b2f7Stbbdev 
11251c0b2f7Stbbdev //! Test copy constructor
11351c0b2f7Stbbdev //! \brief \ref interface
11451c0b2f7Stbbdev TEST_CASE("buffer_node copy constructor"){
11551c0b2f7Stbbdev     test_copies();
11651c0b2f7Stbbdev }
11751c0b2f7Stbbdev 
11851c0b2f7Stbbdev //! Test inheritance relations
11951c0b2f7Stbbdev //! \brief \ref interface
12051c0b2f7Stbbdev TEST_CASE("buffer_node superclasses"){
12151c0b2f7Stbbdev     test_inheritance<int>();
12251c0b2f7Stbbdev     test_inheritance<void*>();
12351c0b2f7Stbbdev }
12451c0b2f7Stbbdev 
125