124 lines
3.7 KiB
C++
124 lines
3.7 KiB
C++
/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
|
|
/*
|
|
* Copyright (c) 2009 University of Washington
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License version 2 as
|
|
* published by the Free Software Foundation;
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program; if not, write to the Free Software
|
|
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
*/
|
|
|
|
#include "test.h"
|
|
#include "traced-callback.h"
|
|
|
|
using namespace ns3;
|
|
|
|
class BasicTracedCallbackTestCase : public TestCase
|
|
{
|
|
public:
|
|
BasicTracedCallbackTestCase ();
|
|
virtual ~BasicTracedCallbackTestCase () {}
|
|
|
|
private:
|
|
virtual void DoRun (void);
|
|
|
|
void CbOne (uint8_t a, double b);
|
|
void CbTwo (uint8_t a, double b);
|
|
|
|
bool m_one;
|
|
bool m_two;
|
|
};
|
|
|
|
BasicTracedCallbackTestCase::BasicTracedCallbackTestCase ()
|
|
: TestCase ("Check basic TracedCallback operation")
|
|
{
|
|
}
|
|
|
|
void
|
|
BasicTracedCallbackTestCase::CbOne (uint8_t a, double b)
|
|
{
|
|
m_one = true;
|
|
}
|
|
|
|
void
|
|
BasicTracedCallbackTestCase::CbTwo (uint8_t a, double b)
|
|
{
|
|
m_two = true;
|
|
}
|
|
|
|
void
|
|
BasicTracedCallbackTestCase::DoRun (void)
|
|
{
|
|
//
|
|
// Create a traced callback and connect it up to our target methods. All that
|
|
// these methods do is to set corresponding member variables m_one and m_two.
|
|
//
|
|
TracedCallback<uint8_t, double> trace;
|
|
|
|
//
|
|
// Connect both callbacks to their respective test methods. If we hit the
|
|
// trace, both callbacks should be called and the two variables should be set
|
|
// to true.
|
|
//
|
|
trace.ConnectWithoutContext (MakeCallback (&BasicTracedCallbackTestCase::CbOne, this));
|
|
trace.ConnectWithoutContext (MakeCallback (&BasicTracedCallbackTestCase::CbTwo, this));
|
|
m_one = false;
|
|
m_two = false;
|
|
trace (1, 2);
|
|
NS_TEST_ASSERT_MSG_EQ (m_one, true, "Callback CbOne not called");
|
|
NS_TEST_ASSERT_MSG_EQ (m_two, true, "Callback CbTwo not called");
|
|
|
|
//
|
|
// If we now disconnect callback one then only callback two should be called.
|
|
//
|
|
trace.DisconnectWithoutContext (MakeCallback (&BasicTracedCallbackTestCase::CbOne, this));
|
|
m_one = false;
|
|
m_two = false;
|
|
trace (1, 2);
|
|
NS_TEST_ASSERT_MSG_EQ (m_one, false, "Callback CbOne unexpectedly called");
|
|
NS_TEST_ASSERT_MSG_EQ (m_two, true, "Callback CbTwo not called");
|
|
|
|
//
|
|
// If we now disconnect callback two then neither callback should be called.
|
|
//
|
|
trace.DisconnectWithoutContext (MakeCallback (&BasicTracedCallbackTestCase::CbTwo, this));
|
|
m_one = false;
|
|
m_two = false;
|
|
trace (1, 2);
|
|
NS_TEST_ASSERT_MSG_EQ (m_one, false, "Callback CbOne unexpectedly called");
|
|
NS_TEST_ASSERT_MSG_EQ (m_two, false, "Callback CbTwo unexpectedly called");
|
|
|
|
//
|
|
// If we connect them back up, then both callbacks should be called.
|
|
//
|
|
trace.ConnectWithoutContext (MakeCallback (&BasicTracedCallbackTestCase::CbOne, this));
|
|
trace.ConnectWithoutContext (MakeCallback (&BasicTracedCallbackTestCase::CbTwo, this));
|
|
m_one = false;
|
|
m_two = false;
|
|
trace (1, 2);
|
|
NS_TEST_ASSERT_MSG_EQ (m_one, true, "Callback CbOne not called");
|
|
NS_TEST_ASSERT_MSG_EQ (m_two, true, "Callback CbTwo not called");
|
|
}
|
|
|
|
class TracedCallbackTestSuite : public TestSuite
|
|
{
|
|
public:
|
|
TracedCallbackTestSuite ();
|
|
};
|
|
|
|
TracedCallbackTestSuite::TracedCallbackTestSuite ()
|
|
: TestSuite ("traced-callback", UNIT)
|
|
{
|
|
AddTestCase (new BasicTracedCallbackTestCase);
|
|
}
|
|
|
|
static TracedCallbackTestSuite tracedCallbackTestSuite;
|