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_overwrite_node.cpp
3351c0b2f7Stbbdev //! \brief Test for [flow_graph.overwrite_node] specification
3451c0b2f7Stbbdev 
3551c0b2f7Stbbdev /*
3651c0b2f7Stbbdev TODO: implement missing conformance tests for overwrite_node:
3751c0b2f7Stbbdev   - [ ] The copy constructor and copy assignment are called for the node's type template parameter.
3851c0b2f7Stbbdev   - [ ] Improve copy constructor test and general constructor tests.
3951c0b2f7Stbbdev   - [ ] Write test checking the value is initially invalid.
4051c0b2f7Stbbdev   - [ ] Write test checking that the gets from the node are non-destructive, but the first `try_get'
4151c0b2f7Stbbdev     fails.
4251c0b2f7Stbbdev   - [ ] Test that first `try_get' fails.
4351c0b2f7Stbbdev   - [ ] Add test on `overwrite_node::is_valid()' method.
4451c0b2f7Stbbdev   - [ ] Add test on `overwrite_node::clear()' method.
4551c0b2f7Stbbdev   - [ ] Add test with reserving `join_node' as node's successor. Use example from the spec.
4651c0b2f7Stbbdev */
4751c0b2f7Stbbdev 
4851c0b2f7Stbbdev template<typename T>
4951c0b2f7Stbbdev void test_inheritance(){
5049e08aacStbbdev     using namespace oneapi::tbb::flow;
5151c0b2f7Stbbdev 
5251c0b2f7Stbbdev     CHECK_MESSAGE( (std::is_base_of<graph_node, overwrite_node<T>>::value), "overwrite_node should be derived from graph_node");
5351c0b2f7Stbbdev     CHECK_MESSAGE( (std::is_base_of<receiver<T>, overwrite_node<T>>::value), "overwrite_node should be derived from receiver<T>");
5451c0b2f7Stbbdev     CHECK_MESSAGE( (std::is_base_of<sender<T>, overwrite_node<T>>::value), "overwrite_node should be derived from sender<T>");
5551c0b2f7Stbbdev }
5651c0b2f7Stbbdev 
5751c0b2f7Stbbdev void test_copies(){
5849e08aacStbbdev     using namespace oneapi::tbb::flow;
5951c0b2f7Stbbdev 
6051c0b2f7Stbbdev     graph g;
6151c0b2f7Stbbdev     overwrite_node<int> fn(g);
6251c0b2f7Stbbdev     overwrite_node<int> fn2(fn);
6351c0b2f7Stbbdev 
6451c0b2f7Stbbdev }
6551c0b2f7Stbbdev 
6651c0b2f7Stbbdev void test_buffering(){
6749e08aacStbbdev     oneapi::tbb::flow::graph g;
6851c0b2f7Stbbdev 
6949e08aacStbbdev     oneapi::tbb::flow::overwrite_node<int> node(g);
7049e08aacStbbdev     oneapi::tbb::flow::limiter_node<int> rejecter(g, 0);
7151c0b2f7Stbbdev 
7249e08aacStbbdev     oneapi::tbb::flow::make_edge(node, rejecter);
7351c0b2f7Stbbdev     node.try_put(1);
7451c0b2f7Stbbdev     g.wait_for_all();
7551c0b2f7Stbbdev 
7651c0b2f7Stbbdev     int 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 }
8051c0b2f7Stbbdev 
8151c0b2f7Stbbdev void test_forwarding(){
8249e08aacStbbdev     oneapi::tbb::flow::graph g;
8351c0b2f7Stbbdev 
8449e08aacStbbdev     oneapi::tbb::flow::overwrite_node<int> node1(g);
8551c0b2f7Stbbdev     test_push_receiver<int> node2(g);
8651c0b2f7Stbbdev     test_push_receiver<int> node3(g);
8751c0b2f7Stbbdev 
8849e08aacStbbdev     oneapi::tbb::flow::make_edge(node1, node2);
8949e08aacStbbdev     oneapi::tbb::flow::make_edge(node1, node3);
9051c0b2f7Stbbdev 
9151c0b2f7Stbbdev     node1.try_put(1);
9251c0b2f7Stbbdev     g.wait_for_all();
9351c0b2f7Stbbdev 
9451c0b2f7Stbbdev     CHECK_MESSAGE( (get_count(node2) == 1), "Descendant of the node needs to be receive N messages");
9551c0b2f7Stbbdev     CHECK_MESSAGE( (get_count(node3) == 1), "Descendant of the node must receive one message.");
9651c0b2f7Stbbdev }
9751c0b2f7Stbbdev 
9851c0b2f7Stbbdev void test_overwriting(){
9949e08aacStbbdev     oneapi::tbb::flow::graph g;
10051c0b2f7Stbbdev 
10149e08aacStbbdev     oneapi::tbb::flow::overwrite_node<int> node1(g);
10251c0b2f7Stbbdev 
10351c0b2f7Stbbdev     int tmp = -1;
10451c0b2f7Stbbdev     node1.try_put(1);
10551c0b2f7Stbbdev     g.wait_for_all();
10651c0b2f7Stbbdev 
10751c0b2f7Stbbdev     CHECK_MESSAGE( (node1.try_get(tmp) == true), "Descendant needs to receive a message");
10851c0b2f7Stbbdev     CHECK_MESSAGE( (tmp == 1), "Descendant needs to receive a correct value");
10951c0b2f7Stbbdev 
11051c0b2f7Stbbdev     node1.try_put(2);
11151c0b2f7Stbbdev     g.wait_for_all();
11251c0b2f7Stbbdev 
11351c0b2f7Stbbdev     CHECK_MESSAGE( (node1.try_get(tmp) == true), "Descendant needs to receive a message");
11451c0b2f7Stbbdev     CHECK_MESSAGE( (tmp == 2), "Descendant needs to receive a correct value");
11551c0b2f7Stbbdev }
11651c0b2f7Stbbdev 
11751c0b2f7Stbbdev //! Test overwrite_node behavior
11851c0b2f7Stbbdev //! \brief \ref requirement
11951c0b2f7Stbbdev TEST_CASE("overwrite_node messages"){
12051c0b2f7Stbbdev     test_overwriting();
12151c0b2f7Stbbdev }
12251c0b2f7Stbbdev 
12351c0b2f7Stbbdev //! Test function_node broadcast
12451c0b2f7Stbbdev //! \brief \ref requirement
12551c0b2f7Stbbdev TEST_CASE("overwrite_node broadcast"){
12651c0b2f7Stbbdev     test_forwarding();
12751c0b2f7Stbbdev }
12851c0b2f7Stbbdev 
12951c0b2f7Stbbdev //! Test function_node buffering
13051c0b2f7Stbbdev //! \brief \ref requirement
13151c0b2f7Stbbdev TEST_CASE("overwrite_node buffering"){
13251c0b2f7Stbbdev     test_buffering();
13351c0b2f7Stbbdev }
13451c0b2f7Stbbdev 
13551c0b2f7Stbbdev //! Test copy constructor
13651c0b2f7Stbbdev //! \brief \ref interface
13751c0b2f7Stbbdev TEST_CASE("overwrite_node copy constructor"){
13851c0b2f7Stbbdev     test_copies();
13951c0b2f7Stbbdev }
14051c0b2f7Stbbdev 
14151c0b2f7Stbbdev //! Test inheritance relations
14251c0b2f7Stbbdev //! \brief \ref interface
14351c0b2f7Stbbdev TEST_CASE("overwrite_node superclasses"){
14451c0b2f7Stbbdev     test_inheritance<int>();
14551c0b2f7Stbbdev     test_inheritance<void*>();
14651c0b2f7Stbbdev }
147