blob: 260a1965ca538245d474fb48de98e3916544cbff [file] [log] [blame]
Junxiao Shi727ed292014-02-19 23:26:45 -07001/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
2/**
Junxiao Shi82e7f582014-09-07 15:15:40 -07003 * Copyright (c) 2014, Regents of the University of California,
4 * Arizona Board of Regents,
5 * Colorado State University,
6 * University Pierre & Marie Curie, Sorbonne University,
7 * Washington University in St. Louis,
8 * Beijing Institute of Technology,
9 * The University of Memphis
Alexander Afanasyev9bcbc7c2014-04-06 19:37:37 -070010 *
11 * This file is part of NFD (Named Data Networking Forwarding Daemon).
12 * See AUTHORS.md for complete list of NFD authors and contributors.
13 *
14 * NFD is free software: you can redistribute it and/or modify it under the terms
15 * of the GNU General Public License as published by the Free Software Foundation,
16 * either version 3 of the License, or (at your option) any later version.
17 *
18 * NFD is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
19 * without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
20 * PURPOSE. See the GNU General Public License for more details.
21 *
22 * You should have received a copy of the GNU General Public License along with
23 * NFD, e.g., in COPYING.md file. If not, see <http://www.gnu.org/licenses/>.
Junxiao Shi82e7f582014-09-07 15:15:40 -070024 */
Junxiao Shi727ed292014-02-19 23:26:45 -070025
Alexander Afanasyev613e2a92014-04-15 13:36:58 -070026#ifndef NFD_TESTS_NFD_FW_STRATEGY_TESTER_HPP
27#define NFD_TESTS_NFD_FW_STRATEGY_TESTER_HPP
Junxiao Shi727ed292014-02-19 23:26:45 -070028
29#include <boost/tuple/tuple_comparison.hpp>
30#include "fw/strategy.hpp"
31
32namespace nfd {
Junxiao Shid9ee45c2014-02-27 15:38:11 -070033namespace tests {
Junxiao Shi727ed292014-02-19 23:26:45 -070034
35/** \class StrategyTester
36 * \brief extends strategy S for unit testing
37 *
38 * Actions invoked by S are recorded but not passed to forwarder
39 */
40template<typename S>
41class StrategyTester : public S
42{
43public:
44 explicit
45 StrategyTester(Forwarder& forwarder)
Junxiao Shif3c07812014-03-11 21:48:49 -070046 : S(forwarder, Name(S::STRATEGY_NAME).append("tester"))
Junxiao Shi727ed292014-02-19 23:26:45 -070047 {
48 }
Alexander Afanasyev9bcbc7c2014-04-06 19:37:37 -070049
Junxiao Shi0b5fbbb2014-02-20 15:54:03 -070050 /// fires after each Action
51 EventEmitter<> onAction;
Junxiao Shi727ed292014-02-19 23:26:45 -070052
53protected:
54 virtual void
Junxiao Shid938a6b2014-05-11 23:40:29 -070055 sendInterest(shared_ptr<pit::Entry> pitEntry,
56 shared_ptr<Face> outFace,
57 bool wantNewNonce = false);
Junxiao Shi727ed292014-02-19 23:26:45 -070058
59 virtual void
Junxiao Shi09498f02014-02-26 19:41:08 -070060 rejectPendingInterest(shared_ptr<pit::Entry> pitEntry);
Junxiao Shi727ed292014-02-19 23:26:45 -070061
62public:
63 typedef boost::tuple<shared_ptr<pit::Entry>, shared_ptr<Face> > SendInterestArgs;
64 std::vector<SendInterestArgs> m_sendInterestHistory;
65
Junxiao Shi09498f02014-02-26 19:41:08 -070066 typedef boost::tuple<shared_ptr<pit::Entry> > RejectPendingInterestArgs;
67 std::vector<RejectPendingInterestArgs> m_rejectPendingInterestHistory;
Junxiao Shi727ed292014-02-19 23:26:45 -070068};
69
70
71template<typename S>
72inline void
73StrategyTester<S>::sendInterest(shared_ptr<pit::Entry> pitEntry,
Junxiao Shid938a6b2014-05-11 23:40:29 -070074 shared_ptr<Face> outFace,
75 bool wantNewNonce)
Junxiao Shi727ed292014-02-19 23:26:45 -070076{
77 m_sendInterestHistory.push_back(SendInterestArgs(pitEntry, outFace));
Junxiao Shi0b5fbbb2014-02-20 15:54:03 -070078 pitEntry->insertOrUpdateOutRecord(outFace, pitEntry->getInterest());
79 onAction();
Junxiao Shi727ed292014-02-19 23:26:45 -070080}
81
82template<typename S>
83inline void
Junxiao Shi09498f02014-02-26 19:41:08 -070084StrategyTester<S>::rejectPendingInterest(shared_ptr<pit::Entry> pitEntry)
Junxiao Shi727ed292014-02-19 23:26:45 -070085{
Junxiao Shi09498f02014-02-26 19:41:08 -070086 m_rejectPendingInterestHistory.push_back(RejectPendingInterestArgs(pitEntry));
Junxiao Shi0b5fbbb2014-02-20 15:54:03 -070087 onAction();
Junxiao Shi727ed292014-02-19 23:26:45 -070088}
89
Junxiao Shid9ee45c2014-02-27 15:38:11 -070090} // namespace tests
Junxiao Shi727ed292014-02-19 23:26:45 -070091} // namespace nfd
92
Alexander Afanasyev613e2a92014-04-15 13:36:58 -070093#endif // NFD_TESTS_NFD_FW_STRATEGY_TESTER_HPP