123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791 |
- // Copyright (C) 2013-2014 Internet Systems Consortium, Inc. ("ISC")
- //
- // Permission to use, copy, modify, and/or distribute this software for any
- // purpose with or without fee is hereby granted, provided that the above
- // copyright notice and this permission notice appear in all copies.
- //
- // THE SOFTWARE IS PROVIDED "AS IS" AND ISC DISCLAIMS ALL WARRANTIES WITH
- // REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
- // AND FITNESS. IN NO EVENT SHALL ISC BE LIABLE FOR ANY SPECIAL, DIRECT,
- // INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
- // LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE
- // OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
- // PERFORMANCE OF THIS SOFTWARE.
- #include <d2/d2_cfg_mgr.h>
- #include <d2/d2_cfg_mgr.h>
- #include <d2/nc_add.h>
- #include <dns/messagerenderer.h>
- #include <nc_test_utils.h>
- #include <boost/function.hpp>
- #include <boost/bind.hpp>
- #include <gtest/gtest.h>
- using namespace std;
- using namespace isc;
- using namespace isc::d2;
- namespace {
- /// @brief Test class derived from NameAddTransaction to provide visiblity
- // to protected methods.
- class NameAddStub : public NameAddTransaction {
- public:
- NameAddStub(IOServicePtr& io_service,
- dhcp_ddns::NameChangeRequestPtr& ncr,
- DdnsDomainPtr& forward_domain,
- DdnsDomainPtr& reverse_domain)
- : NameAddTransaction(io_service, ncr, forward_domain, reverse_domain),
- simulate_send_exception_(false),
- simulate_build_request_exception_(false) {
- }
- virtual ~NameAddStub() {
- }
- /// @brief Simulates sending update requests to the DNS server
- ///
- /// This method simulates the initiation of an asynchronous send of
- /// a DNS update request. It overrides the actual sendUpdate method in
- /// the base class, thus avoiding an actual send, yet still increments
- /// the update attempt count and posts a next event of NOP_EVT.
- ///
- /// It will also simulate an exception-based failure of sendUpdate, if
- /// the simulate_send_exception_ flag is true.
- ///
- /// @param comment Parameter is unused, but present in base class method.
- /// @param use_tsig_ Parameter is unused, but present in base class method.
- ///
- virtual void sendUpdate(const std::string& /*comment*/,
- bool /* use_tsig_ = false */) {
- if (simulate_send_exception_) {
- // Make the flag a one-shot by resetting it.
- simulate_send_exception_ = false;
- // Transition to failed.
- transition(PROCESS_TRANS_FAILED_ST, UPDATE_FAILED_EVT);
- return;
- }
- // Update send attempt count and post a NOP_EVT.
- setUpdateAttempts(getUpdateAttempts() + 1);
- postNextEvent(StateModel::NOP_EVT);
- }
- /// @brief Prepares the initial D2UpdateMessage
- ///
- /// This method overrides the NameChangeTransactio implementation to
- /// provide the ability to simulate an exception throw in the build
- /// request logic.
- /// If the one-shot flag, simulate_build_request_exception_ is true,
- /// this method will throw an exception, otherwise it will invoke the
- /// base class method, providing normal functionality.
- ///
- /// For parameter description see the NameChangeTransaction implementation.
- virtual D2UpdateMessagePtr prepNewRequest(DdnsDomainPtr domain) {
- if (simulate_build_request_exception_) {
- simulate_build_request_exception_ = false;
- isc_throw (NameAddTransactionError,
- "Simulated build requests exception");
- }
- return (NameChangeTransaction::prepNewRequest(domain));
- }
- /// @brief Simulates receiving a response
- ///
- /// This method simulates the completion of a DNSClient send. This allows
- /// the state handler logic devoted to dealing with IO completion to be
- /// fully exercised without requiring any actual IO. The two primary
- /// pieces of information gleaned from IO completion are the DNSClient
- /// status which indicates whether or not the IO exchange was successful
- /// and the rcode, which indicates the server's reaction to the request.
- ///
- /// This method updates the transaction's DNS status value to that of the
- /// given parameter, and then constructs and DNS update response message
- /// with the given rcode value. To complete the simulation it then posts
- /// a next event of IO_COMPLETED_EVT.
- ///
- /// @param status simulated DNSClient status
- /// @param rcode simulated server response code
- void fakeResponse(const DNSClient::Status& status,
- const dns::Rcode& rcode) {
- // Set the DNS update status. This is normally set in
- // DNSClient IO completion handler.
- setDnsUpdateStatus(status);
- // Construct an empty message with the given Rcode.
- D2UpdateMessagePtr msg(new D2UpdateMessage(D2UpdateMessage::OUTBOUND));
- msg->setRcode(rcode);
- // Set the update response to the message.
- setDnsUpdateResponse(msg);
- // Post the IO completion event.
- postNextEvent(NameChangeTransaction::IO_COMPLETED_EVT);
- }
- /// @brief Selects the first forward server.
- /// Some state handlers require a server to have been selected.
- /// This selects a server without going through the state
- /// transition(s) to do so.
- bool selectFwdServer() {
- if (getForwardDomain()) {
- initServerSelection(getForwardDomain());
- selectNextServer();
- return (getCurrentServer());
- }
- return (false);
- }
- /// @brief Selects the first reverse server.
- /// Some state handlers require a server to have been selected.
- /// This selects a server without going through the state
- /// transition(s) to do so.
- bool selectRevServer() {
- if (getReverseDomain()) {
- initServerSelection(getReverseDomain());
- selectNextServer();
- return (getCurrentServer());
- }
- return (false);
- }
- /// @brief One-shot flag which will simulate sendUpdate failure if true.
- bool simulate_send_exception_;
- /// @brief One-shot flag which will simulate an exception when sendUpdate
- /// failure if true.
- bool simulate_build_request_exception_;
- using StateModel::postNextEvent;
- using StateModel::setState;
- using StateModel::initDictionaries;
- using NameAddTransaction::defineEvents;
- using NameAddTransaction::verifyEvents;
- using NameAddTransaction::defineStates;
- using NameAddTransaction::verifyStates;
- using NameAddTransaction::readyHandler;
- using NameAddTransaction::selectingFwdServerHandler;
- using NameAddTransaction::getCurrentServer;
- using NameAddTransaction::addingFwdAddrsHandler;
- using NameAddTransaction::setDnsUpdateStatus;
- using NameAddTransaction::replacingFwdAddrsHandler;
- using NameAddTransaction::selectingRevServerHandler;
- using NameAddTransaction::replacingRevPtrsHandler;
- using NameAddTransaction::processAddOkHandler;
- using NameAddTransaction::processAddFailedHandler;
- using NameAddTransaction::buildAddFwdAddressRequest;
- using NameAddTransaction::buildReplaceFwdAddressRequest;
- using NameAddTransaction::buildReplaceRevPtrsRequest;
- };
- typedef boost::shared_ptr<NameAddStub> NameAddStubPtr;
- /// @brief Test fixture for testing NameAddTransaction
- ///
- /// Note this class uses NameAddStub class to exercise non-public
- /// aspects of NameAddTransaction.
- class NameAddTransactionTest : public TransactionTest {
- public:
- NameAddTransactionTest() {
- }
- virtual ~NameAddTransactionTest() {
- }
- /// @brief Creates a transaction which requests an IPv4 DNS update.
- ///
- /// The transaction is constructed around a predefined (i.e. "canned")
- /// IPv4 NameChangeRequest. The request has both forward and reverse DNS
- /// changes requested. Based upon the change mask, the transaction
- /// will have either the forward, reverse, or both domains populated.
- ///
- /// @param change_mask determines which change directions are requested
- NameAddStubPtr makeTransaction4(int change_mask = FWD_AND_REV_CHG) {
- // Creates IPv4 remove request, forward, and reverse domains.
- setupForIPv4Transaction(dhcp_ddns::CHG_ADD, change_mask);
- // Now create the test transaction as would occur in update manager.
- return (NameAddStubPtr(new NameAddStub(io_service_, ncr_,
- forward_domain_,
- reverse_domain_)));
- }
- /// @brief Creates a transaction which requests an IPv6 DNS update.
- ///
- /// The transaction is constructed around a predefined (i.e. "canned")
- /// IPv6 NameChangeRequest. The request has both forward and reverse DNS
- /// changes requested. Based upon the change mask, the transaction
- /// will have either the forward, reverse, or both domains populated.
- ///
- /// @param change_mask determines which change directions are requested
- NameAddStubPtr makeTransaction6(int change_mask = FWD_AND_REV_CHG) {
- // Creates IPv6 remove request, forward, and reverse domains.
- setupForIPv6Transaction(dhcp_ddns::CHG_ADD, change_mask);
- // Now create the test transaction as would occur in update manager.
- return (NameAddStubPtr(new NameAddStub(io_service_, ncr_,
- forward_domain_,
- reverse_domain_)));
- }
- /// @brief Create a test transaction at a known point in the state model.
- ///
- /// Method prepares a new test transaction and sets its state and next
- /// event values to those given. This makes the transaction appear to
- /// be at that point in the state model without having to transition it
- /// through prerequisite states. It also provides the ability to set
- /// which change directions are requested: forward change only, reverse
- /// change only, or both.
- ///
- /// @param state value to set as the current state
- /// @param event value to post as the next event
- /// @param change_mask determines which change directions are requested
- /// @param family selects between an IPv4 (AF_INET) and IPv6 (AF_INET6)
- /// transaction.
- NameAddStubPtr prepHandlerTest(unsigned int state, unsigned int event,
- unsigned int change_mask = FWD_AND_REV_CHG,
- short family = AF_INET) {
- NameAddStubPtr name_add = (family == AF_INET ?
- makeTransaction4(change_mask) :
- makeTransaction4(change_mask));
- name_add->initDictionaries();
- name_add->postNextEvent(event);
- name_add->setState(state);
- return (name_add);
- }
- };
- /// @brief Tests NameAddTransaction construction.
- /// This test verifies that:
- /// 1. Construction with invalid type of request
- /// 2. Valid construction functions properly
- TEST(NameAddTransaction, construction) {
- IOServicePtr io_service(new isc::asiolink::IOService());
- const char* msg_str =
- "{"
- " \"change_type\" : 1 , "
- " \"forward_change\" : true , "
- " \"reverse_change\" : true , "
- " \"fqdn\" : \"example.com.\" , "
- " \"ip_address\" : \"192.168.2.1\" , "
- " \"dhcid\" : \"0102030405060708\" , "
- " \"lease_expires_on\" : \"20130121132405\" , "
- " \"lease_length\" : 1300 "
- "}";
- dhcp_ddns::NameChangeRequestPtr ncr;
- DnsServerInfoStoragePtr servers;
- DdnsDomainPtr forward_domain;
- DdnsDomainPtr reverse_domain;
- DdnsDomainPtr empty_domain;
- ASSERT_NO_THROW(ncr = dhcp_ddns::NameChangeRequest::fromJSON(msg_str));
- ASSERT_NO_THROW(forward_domain.reset(new DdnsDomain("*", "", servers)));
- ASSERT_NO_THROW(reverse_domain.reset(new DdnsDomain("*", "", servers)));
- // Verify that construction with wrong change type fails.
- EXPECT_THROW(NameAddTransaction(io_service, ncr,
- forward_domain, reverse_domain),
- NameAddTransactionError);
- // Verify that a valid construction attempt works.
- ncr->setChangeType(isc::dhcp_ddns::CHG_ADD);
- EXPECT_NO_THROW(NameAddTransaction(io_service, ncr,
- forward_domain, reverse_domain));
- }
- /// @brief Tests event and state dictionary construction and verification.
- TEST_F(NameAddTransactionTest, dictionaryCheck) {
- NameAddStubPtr name_add;
- ASSERT_NO_THROW(name_add = makeTransaction4());
- // Verify that the event and state dictionary validation fails prior
- // dictionary construction.
- ASSERT_THROW(name_add->verifyEvents(), StateModelError);
- ASSERT_THROW(name_add->verifyStates(), StateModelError);
- // Construct both dictionaries.
- ASSERT_NO_THROW(name_add->defineEvents());
- ASSERT_NO_THROW(name_add->defineStates());
- // Verify both event and state dictionaries now pass validation.
- ASSERT_NO_THROW(name_add->verifyEvents());
- ASSERT_NO_THROW(name_add->verifyStates());
- }
- /// @brief Tests construction of a DNS update request for adding a forward
- /// dns entry.
- TEST_F(NameAddTransactionTest, buildForwardAdd) {
- // Create a IPv4 forward add transaction.
- // Verify the request builds without error.
- // and then verify the request contents.
- NameAddStubPtr name_add;
- ASSERT_NO_THROW(name_add = makeTransaction4());
- ASSERT_NO_THROW(name_add->buildAddFwdAddressRequest());
- checkAddFwdAddressRequest(*name_add);
- // Create a IPv6 forward add transaction.
- // Verify the request builds without error.
- // and then verify the request contents.
- ASSERT_NO_THROW(name_add = makeTransaction6());
- ASSERT_NO_THROW(name_add->buildAddFwdAddressRequest());
- checkAddFwdAddressRequest(*name_add);
- }
- /// @brief Tests construction of a DNS update request for replacing a forward
- /// dns entry.
- TEST_F(NameAddTransactionTest, buildReplaceFwdAddressRequest) {
- // Create a IPv4 forward replace transaction.
- // Verify the request builds without error.
- // and then verify the request contents.
- NameAddStubPtr name_add;
- ASSERT_NO_THROW(name_add = makeTransaction4());
- ASSERT_NO_THROW(name_add->buildReplaceFwdAddressRequest());
- checkReplaceFwdAddressRequest(*name_add);
- // Create a IPv6 forward replace transaction.
- // Verify the request builds without error.
- // and then verify the request contents.
- ASSERT_NO_THROW(name_add = makeTransaction6());
- ASSERT_NO_THROW(name_add->buildReplaceFwdAddressRequest());
- checkReplaceFwdAddressRequest(*name_add);
- }
- /// @brief Tests the construction of a DNS update request for replacing a
- /// reverse dns entry.
- TEST_F(NameAddTransactionTest, buildReplaceRevPtrsRequest) {
- // Create a IPv4 reverse replace transaction.
- // Verify the request builds without error.
- // and then verify the request contents.
- NameAddStubPtr name_add;
- ASSERT_NO_THROW(name_add = makeTransaction4());
- ASSERT_NO_THROW(name_add->buildReplaceRevPtrsRequest());
- checkReplaceRevPtrsRequest(*name_add);
- // Create a IPv6 reverse replace transaction.
- // Verify the request builds without error.
- // and then verify the request contents.
- ASSERT_NO_THROW(name_add = makeTransaction6());
- ASSERT_NO_THROW(name_add->buildReplaceRevPtrsRequest());
- checkReplaceRevPtrsRequest(*name_add);
- }
- // Tests the readyHandler functionality.
- // It verifies behavior for the following scenarios:
- //
- // 1. Posted event is START_EVT and request includes only a forward change
- // 2. Posted event is START_EVT and request includes both a forward and a
- // reverse change
- // 3. Posted event is START_EVT and request includes only a reverse change
- // 4. Posted event is invalid
- //
- TEST_F(NameAddTransactionTest, readyHandler) {
- NameAddStubPtr name_add;
- // Create a transaction which includes only a forward change.
- ASSERT_NO_THROW(name_add =
- prepHandlerTest(NameChangeTransaction::READY_ST,
- StateModel::START_EVT, FORWARD_CHG));
- // Run readyHandler.
- EXPECT_NO_THROW(name_add->readyHandler());
- // Verify that a request requiring only a forward change, transitions to
- // selecting a forward server.
- EXPECT_EQ(NameChangeTransaction::SELECTING_FWD_SERVER_ST,
- name_add->getCurrState());
- EXPECT_EQ(NameChangeTransaction::SELECT_SERVER_EVT,
- name_add->getNextEvent());
- // Create a transaction which includes both a forward and a reverse change.
- ASSERT_NO_THROW(name_add =
- prepHandlerTest(NameChangeTransaction::READY_ST,
- StateModel::START_EVT, FWD_AND_REV_CHG));
- // Run readyHandler.
- EXPECT_NO_THROW(name_add->readyHandler());
- // Verify that a request requiring both forward and reverse, starts with
- // the forward change by transitioning to selecting a forward server.
- EXPECT_EQ(NameChangeTransaction::SELECTING_FWD_SERVER_ST,
- name_add->getCurrState());
- EXPECT_EQ(NameChangeTransaction::SELECT_SERVER_EVT,
- name_add->getNextEvent());
- // Create and prep a reverse only transaction.
- ASSERT_NO_THROW(name_add =
- prepHandlerTest(NameChangeTransaction::READY_ST,
- StateModel::START_EVT, REVERSE_CHG));
- // Run readyHandler.
- EXPECT_NO_THROW(name_add->readyHandler());
- // Verify that a request requiring only a reverse change, transitions to
- // selecting a reverse server.
- EXPECT_EQ(NameChangeTransaction::SELECTING_REV_SERVER_ST,
- name_add->getCurrState());
- EXPECT_EQ(NameChangeTransaction::SELECT_SERVER_EVT,
- name_add->getNextEvent());
- // Create and prep transaction, poised to run the handler but with an
- // invalid event.
- ASSERT_NO_THROW(name_add =
- prepHandlerTest(NameChangeTransaction::READY_ST,
- StateModel::NOP_EVT));
- // Running the readyHandler should throw.
- EXPECT_THROW(name_add->readyHandler(), NameAddTransactionError);
- }
- // Tests the selectingFwdServerHandler functionality.
- // It verifies behavior for the following scenarios:
- //
- // 1. Posted event is SELECT_SERVER_EVT
- // 2. Posted event is SERVER_IO_ERROR_EVT
- // 3. Posted event is invalid
- //
- TEST_F(NameAddTransactionTest, selectingFwdServerHandler) {
- NameAddStubPtr name_add;
- // Create and prep a transaction, poised to run the handler.
- ASSERT_NO_THROW(name_add =
- prepHandlerTest(NameChangeTransaction::
- SELECTING_FWD_SERVER_ST,
- NameChangeTransaction::SELECT_SERVER_EVT));
- // Call selectingFwdServerHandler enough times to select all of the
- // servers in it's current domain. The first time, it will be with
- // next event of SELECT_SERVER_EVT. Thereafter it will be with a next
- // event of SERVER_IO_ERROR_EVT.
- int num_servers = name_add->getForwardDomain()->getServers()->size();
- for (int i = 0; i < num_servers; ++i) {
- // Run selectingFwdServerHandler.
- ASSERT_NO_THROW(name_add->selectingFwdServerHandler())
- << " num_servers: " << num_servers
- << " selections: " << i;
- // Verify that a server was selected.
- ASSERT_TRUE(name_add->getCurrentServer())
- << " num_servers: " << num_servers << " selections: " << i;
- // Verify that we transitioned correctly.
- ASSERT_EQ(NameAddTransaction::ADDING_FWD_ADDRS_ST,
- name_add->getCurrState())
- << " num_servers: " << num_servers << " selections: " << i;
- ASSERT_EQ(NameChangeTransaction::SERVER_SELECTED_EVT,
- name_add->getNextEvent())
- << " num_servers: " << num_servers << " selections: " << i;
- // Post a server IO error event. This simulates an IO error occuring
- // and a need to select the new server.
- ASSERT_NO_THROW(name_add->postNextEvent(NameChangeTransaction::
- SERVER_IO_ERROR_EVT))
- << " num_servers: " << num_servers
- << " selections: " << i;
- }
- // We should have exhausted the list of servers. Processing another
- // SERVER_IO_ERROR_EVT should transition us to failure.
- EXPECT_NO_THROW(name_add->selectingFwdServerHandler());
- EXPECT_EQ(NameChangeTransaction::PROCESS_TRANS_FAILED_ST,
- name_add->getCurrState());
- EXPECT_EQ(NameChangeTransaction::NO_MORE_SERVERS_EVT,
- name_add->getNextEvent());
- // Create and prep transaction, poised to run the handler but with an
- // invalid event.
- ASSERT_NO_THROW(name_add =
- prepHandlerTest(NameChangeTransaction::
- SELECTING_FWD_SERVER_ST,
- StateModel::NOP_EVT));
- // Running the handler should throw.
- EXPECT_THROW(name_add->selectingFwdServerHandler(),
- NameAddTransactionError);
- }
- // ************************ addingFwdAddrHandler Tests *****************
- // Tests that addingFwdAddrsHandler rejects invalid events.
- TEST_F(NameAddTransactionTest, addingFwdAddrsHandler_InvalidEvent) {
- NameAddStubPtr name_add;
- // Create and prep a transaction, poised to run the handler but with
- // an invalid event.
- ASSERT_NO_THROW(name_add =
- prepHandlerTest(NameAddTransaction::ADDING_FWD_ADDRS_ST,
- NameChangeTransaction::
- StateModel::NOP_EVT));
- // Running the handler should throw.
- EXPECT_THROW(name_add->addingFwdAddrsHandler(),
- NameAddTransactionError);
- }
- // Tests addingFwdAddrsHandler with the following scenario:
- //
- // The request includes only a forward change.
- // Initial posted event is SERVER_SELECTED_EVT.
- // The update request is sent without error.
- // A server response is received which indicates successful update
- //
- TEST_F(NameAddTransactionTest, addingFwdAddrsHandler_FwdOnlyAddOK) {
- NameAddStubPtr name_add;
- // Create and prep a transaction, poised to run the handler.
- ASSERT_NO_THROW(name_add =
- prepHandlerTest(NameAddTransaction::ADDING_FWD_ADDRS_ST,
- NameChangeTransaction::
- SERVER_SELECTED_EVT, FORWARD_CHG));
- // Should not be an update message yet.
- D2UpdateMessagePtr update_msg = name_add->getDnsUpdateRequest();
- ASSERT_FALSE(update_msg);
- // At this point completion flags should be false.
- EXPECT_FALSE(name_add->getForwardChangeCompleted());
- EXPECT_FALSE(name_add->getReverseChangeCompleted());
- // Run addingFwdAddrsHandler to construct and send the request.
- EXPECT_NO_THROW(name_add->addingFwdAddrsHandler());
- // Verify that an update message was constructed properly.
- checkAddFwdAddressRequest(*name_add);
- // Verify that we are still in this state and next event is NOP_EVT.
- // This indicates we "sent" the message and are waiting for IO completion.
- EXPECT_EQ(NameAddTransaction::ADDING_FWD_ADDRS_ST,
- name_add->getCurrState());
- EXPECT_EQ(NameChangeTransaction::NOP_EVT,
- name_add->getNextEvent());
- // Simulate receiving a successful update response.
- name_add->fakeResponse(DNSClient::SUCCESS, dns::Rcode::NOERROR());
- // Run addingFwdAddrsHandler again to process the response.
- EXPECT_NO_THROW(name_add->addingFwdAddrsHandler());
- // Forward completion should be true, reverse should be false.
- EXPECT_TRUE(name_add->getForwardChangeCompleted());
- EXPECT_FALSE(name_add->getReverseChangeCompleted());
- // Since it is a forward only change, we should be done.
- // Verify that we transitioned correctly.
- EXPECT_EQ(NameChangeTransaction::PROCESS_TRANS_OK_ST,
- name_add->getCurrState());
- EXPECT_EQ(NameChangeTransaction::UPDATE_OK_EVT,
- name_add->getNextEvent());
- }
- // Tests addingFwdAddrsHandler with the following scenario:
- //
- // The request includes a forward and reverse change.
- // Initial posted event is SERVER_SELECTED_EVT.
- // The update request is sent without error.
- // A server response is received which indicates successful update.
- //
- TEST_F(NameAddTransactionTest, addingFwdAddrsHandler_fwdAndRevAddOK) {
- NameAddStubPtr name_add;
- // Create and prep a transaction, poised to run the handler.
- ASSERT_NO_THROW(name_add =
- prepHandlerTest(NameAddTransaction::ADDING_FWD_ADDRS_ST,
- NameChangeTransaction::
- SERVER_SELECTED_EVT, FWD_AND_REV_CHG));
- // Run addingFwdAddrsHandler to construct and send the request.
- EXPECT_NO_THROW(name_add->addingFwdAddrsHandler());
- // Simulate receiving a successful update response.
- name_add->fakeResponse(DNSClient::SUCCESS, dns::Rcode::NOERROR());
- // Run addingFwdAddrsHandler again to process the response.
- EXPECT_NO_THROW(name_add->addingFwdAddrsHandler());
- // Forward change completion should be true, reverse flag should be false.
- EXPECT_TRUE(name_add->getForwardChangeCompleted());
- EXPECT_FALSE(name_add->getReverseChangeCompleted());
- // Since the request also includes a reverse change we should
- // be poised to start it. Verify that we transitioned correctly.
- EXPECT_EQ(NameChangeTransaction::SELECTING_REV_SERVER_ST,
- name_add->getCurrState());
- EXPECT_EQ(NameChangeTransaction::SELECT_SERVER_EVT,
- name_add->getNextEvent());
- }
- // Tests addingFwdAddrsHandler with the following scenario:
- //
- // The request includes a forward and reverse change.
- // Initial posted event is SERVER_SELECTED_EVT.
- // The update request is sent without error.
- // A server response is received which indicates the FQDN is in use.
- //
- TEST_F(NameAddTransactionTest, addingFwdAddrsHandler_FqdnInUse) {
- NameAddStubPtr name_add;
- // Create and prep a transaction, poised to run the handler.
- ASSERT_NO_THROW(name_add =
- prepHandlerTest(NameAddTransaction::ADDING_FWD_ADDRS_ST,
- NameChangeTransaction::
- SERVER_SELECTED_EVT));
- // Run addingFwdAddrsHandler to construct and send the request.
- EXPECT_NO_THROW(name_add->addingFwdAddrsHandler());
- // Simulate receiving a FQDN in use response.
- name_add->fakeResponse(DNSClient::SUCCESS, dns::Rcode::YXDOMAIN());
- // Run addingFwdAddrsHandler again to process the response.
- EXPECT_NO_THROW(name_add->addingFwdAddrsHandler());
- // Completion flags should still be false.
- EXPECT_FALSE(name_add->getForwardChangeCompleted());
- EXPECT_FALSE(name_add->getReverseChangeCompleted());
- // Since the FQDN is in use, per the RFC we must attempt to replace it.
- // Verify that we transitioned correctly.
- EXPECT_EQ(NameAddTransaction::REPLACING_FWD_ADDRS_ST,
- name_add->getCurrState());
- EXPECT_EQ(NameAddTransaction::FQDN_IN_USE_EVT,
- name_add->getNextEvent());
- }
- // Tests addingFwdAddrsHandler with the following scenario:
- //
- // The request includes a forward and reverse change.
- // Initial posted event is SERVER_SELECTED_EVT.
- // The update request is sent without error.
- // A server response is received which indicates the update was rejected.
- //
- TEST_F(NameAddTransactionTest, addingFwdAddrsHandler_OtherRcode) {
- NameAddStubPtr name_add;
- // Create and prep a transaction, poised to run the handler.
- ASSERT_NO_THROW(name_add =
- prepHandlerTest(NameAddTransaction::ADDING_FWD_ADDRS_ST,
- NameChangeTransaction::
- SERVER_SELECTED_EVT));
- // Select a server to satisfy log statements.
- ASSERT_TRUE(name_add->selectFwdServer());
- // Run addingFwdAddrsHandler to construct and send the request.
- EXPECT_NO_THROW(name_add->addingFwdAddrsHandler());
- // Simulate receiving server rejection response. Per RFC, anything other
- // than no error or FQDN in use is failure. Arbitrarily choosing refused.
- name_add->fakeResponse(DNSClient::SUCCESS, dns::Rcode::REFUSED());
- // Run addingFwdAddrsHandler again to process the response.
- EXPECT_NO_THROW(name_add->addingFwdAddrsHandler());
- // Completion flags should still be false.
- EXPECT_FALSE(name_add->getForwardChangeCompleted());
- EXPECT_FALSE(name_add->getReverseChangeCompleted());
- // We should have failed the transaction. Verify that we transitioned
- // correctly.
- EXPECT_EQ(NameChangeTransaction::PROCESS_TRANS_FAILED_ST,
- name_add->getCurrState());
- EXPECT_EQ(NameChangeTransaction::UPDATE_FAILED_EVT,
- name_add->getNextEvent());
- }
- // Tests addingFwdAddrsHandler with the following scenario:
- //
- // The request includes a forward and reverse change.
- // Initial posted event is SERVER_SELECTED_EVT.
- // The update request send times out MAX_UPDATE_TRIES_PER_SERVER times.
- //
- TEST_F(NameAddTransactionTest, addingFwdAddrsHandler_Timeout) {
- NameAddStubPtr name_add;
- // Create and prep a transaction, poised to run the handler.
- // The log message issued when this test succeeds, displays the
- // selected server, so we need to select a server before running this
- // test.
- ASSERT_NO_THROW(name_add =
- prepHandlerTest(NameAddTransaction::ADDING_FWD_ADDRS_ST,
- NameChangeTransaction::
- SERVER_SELECTED_EVT));
- // Select a server to satisfy log statements.
- ASSERT_TRUE(name_add->selectFwdServer());
- // Verify that we can make maximum number of update attempts permitted
- // and then transition to selecting a new server.
- int max_tries = NameChangeTransaction::MAX_UPDATE_TRIES_PER_SERVER;
- for (int i = 1; i <= max_tries; ++i) {
- const D2UpdateMessagePtr prev_msg = name_add->getDnsUpdateRequest();
- // Run addingFwdAddrsHandler to send the request.
- EXPECT_NO_THROW(name_add->addingFwdAddrsHandler());
- const D2UpdateMessagePtr curr_msg = name_add->getDnsUpdateRequest();
- if (i == 1) {
- // First time out we should build the message.
- EXPECT_FALSE(prev_msg);
- EXPECT_TRUE(curr_msg);
- } else {
- // Subsequent passes should reuse the request. We are only
- // looking to check that we have not replaced the pointer value
- // with a new pointer. This tests the on_entry() logic which
- // clears the request ONLY upon initial entry into the state.
- EXPECT_TRUE(prev_msg == curr_msg);
- }
- // Simulate a server IO timeout.
- name_add->setDnsUpdateStatus(DNSClient::TIMEOUT);
- name_add->postNextEvent(NameChangeTransaction::IO_COMPLETED_EVT);
- // Run addingFwdAddrsHandler again to process the response.
- EXPECT_NO_THROW(name_add->addingFwdAddrsHandler());
- // Completion flags should be false.
- EXPECT_FALSE(name_add->getForwardChangeCompleted());
- EXPECT_FALSE(name_add->getReverseChangeCompleted());
- if (i < max_tries) {
- // We should be ready to try again.
- EXPECT_EQ(NameAddTransaction::ADDING_FWD_ADDRS_ST,
- name_add->getCurrState());
- EXPECT_EQ(NameChangeTransaction::SERVER_SELECTED_EVT,
- name_add->getNextEvent());
- } else {
- // Server retries should be exhausted, time for a new server.
- EXPECT_EQ(NameAddTransaction::SELECTING_FWD_SERVER_ST,
- name_add->getCurrState());
- EXPECT_EQ(NameChangeTransaction::SERVER_IO_ERROR_EVT,
- name_add->getNextEvent());
- }
- }
- }
- // Tests addingFwdAddrsHandler with the following scenario:
- //
- // The request includes a forward and reverse change.
- // Initial posted event is SERVER_SELECTED_EVT.
- // The update request is sent but a corrupt response is received, this occurs
- // MAX_UPDATE_TRIES_PER_SERVER times.
- //
- TEST_F(NameAddTransactionTest, addingFwdAddrsHandler_InvalidResponse) {
- NameAddStubPtr name_add;
- // Create and prep a transaction, poised to run the handler.
- // The log message issued when this test succeeds, displays the
- // selected server, so we need to select a server before running this
- // test.
- ASSERT_NO_THROW(name_add =
- prepHandlerTest(NameAddTransaction::ADDING_FWD_ADDRS_ST,
- NameChangeTransaction::
- SERVER_SELECTED_EVT));
- // Select a server to satisfy log statements.
- ASSERT_TRUE(name_add->selectFwdServer());
- // Verify that we can make maximum number of update attempts permitted
- // and then transition to selecting a new server.
- int max_tries = NameChangeTransaction::MAX_UPDATE_TRIES_PER_SERVER;
- for (int i = 1; i <= max_tries; ++i) {
- // Run addingFwdAddrsHandler to construct send the request.
- EXPECT_NO_THROW(name_add->addingFwdAddrsHandler());
- // Simulate a corrupt server response.
- name_add->setDnsUpdateStatus(DNSClient::INVALID_RESPONSE);
- name_add->postNextEvent(NameChangeTransaction::IO_COMPLETED_EVT);
- // Run addingFwdAddrsHandler again to process the response.
- EXPECT_NO_THROW(name_add->addingFwdAddrsHandler());
- // Completion flags should be false.
- EXPECT_FALSE(name_add->getForwardChangeCompleted());
- EXPECT_FALSE(name_add->getReverseChangeCompleted());
- if (i < max_tries) {
- // We should be ready to try again.
- EXPECT_EQ(NameAddTransaction::ADDING_FWD_ADDRS_ST,
- name_add->getCurrState());
- EXPECT_EQ(NameChangeTransaction::SERVER_SELECTED_EVT,
- name_add->getNextEvent());
- } else {
- // Server retries should be exhausted, time for a new server.
- EXPECT_EQ(NameAddTransaction::SELECTING_FWD_SERVER_ST,
- name_add->getCurrState());
- EXPECT_EQ(NameChangeTransaction::SERVER_IO_ERROR_EVT,
- name_add->getNextEvent());
- }
- }
- }
- // ************************ replacingFwdAddrHandler Tests *****************
- // Tests that replacingFwdAddrsHandler rejects invalid events.
- TEST_F(NameAddTransactionTest, replacingFwdAddrsHandler_InvalidEvent) {
- NameAddStubPtr name_add;
- // Create and prep a transaction, poised to run the handler but with
- // an invalid event.
- ASSERT_NO_THROW(name_add =
- prepHandlerTest(NameAddTransaction::REPLACING_FWD_ADDRS_ST,
- NameChangeTransaction::
- StateModel::NOP_EVT));
- // Running the handler should throw.
- EXPECT_THROW(name_add->replacingFwdAddrsHandler(),
- NameAddTransactionError);
- }
- // Tests replacingFwdAddrsHandler with the following scenario:
- //
- // The request includes only a forward change.
- // Initial posted event is FQDN_IN_USE_EVT.
- // The update request is sent without error.
- // A server response is received which indicates successful update.
- //
- TEST_F(NameAddTransactionTest, replacingFwdAddrsHandler_FwdOnlyAddOK) {
- NameAddStubPtr name_add;
- // Create and prep a transaction, poised to run the handler.
- ASSERT_NO_THROW(name_add =
- prepHandlerTest(NameAddTransaction::REPLACING_FWD_ADDRS_ST,
- NameAddTransaction::
- FQDN_IN_USE_EVT, FORWARD_CHG));
- // Should not be an update message yet.
- D2UpdateMessagePtr update_msg = name_add->getDnsUpdateRequest();
- ASSERT_FALSE(update_msg);
- // At this point completion flags should be false.
- EXPECT_FALSE(name_add->getForwardChangeCompleted());
- EXPECT_FALSE(name_add->getReverseChangeCompleted());
- // Run replacingFwdAddrsHandler to construct and send the request.
- EXPECT_NO_THROW(name_add->replacingFwdAddrsHandler());
- // Verify that an update message was constructed properly.
- checkReplaceFwdAddressRequest(*name_add);
- // Verify that we are still in this state and next event is NOP_EVT.
- // This indicates we "sent" the message and are waiting for IO completion.
- EXPECT_EQ(NameAddTransaction::REPLACING_FWD_ADDRS_ST,
- name_add->getCurrState());
- EXPECT_EQ(NameChangeTransaction::NOP_EVT,
- name_add->getNextEvent());
- // Simulate receiving a successful update response.
- name_add->fakeResponse(DNSClient::SUCCESS, dns::Rcode::NOERROR());
- // Run replacingFwdAddrsHandler again to process the response.
- EXPECT_NO_THROW(name_add->replacingFwdAddrsHandler());
- // Forward completion should be true, reverse should be false.
- EXPECT_TRUE(name_add->getForwardChangeCompleted());
- EXPECT_FALSE(name_add->getReverseChangeCompleted());
- // Since it is a forward only change, we should be done.
- // Verify that we transitioned correctly.
- EXPECT_EQ(NameChangeTransaction::PROCESS_TRANS_OK_ST,
- name_add->getCurrState());
- EXPECT_EQ(NameChangeTransaction::UPDATE_OK_EVT,
- name_add->getNextEvent());
- }
- // Tests replacingFwdAddrsHandler with the following scenario:
- //
- // The request includes only a forward change.
- // Initial posted event is SERVER_SELECTED_EVT.
- // The update request is sent without error.
- // A server response is received which indicates successful update.
- //
- TEST_F(NameAddTransactionTest, replacingFwdAddrsHandler_FwdOnlyAddOK2) {
- NameAddStubPtr name_add;
- // Create and prep a transaction, poised to run the handler.
- ASSERT_NO_THROW(name_add =
- prepHandlerTest(NameAddTransaction::REPLACING_FWD_ADDRS_ST,
- NameChangeTransaction::
- SERVER_SELECTED_EVT, FORWARD_CHG));
- // Run replacingFwdAddrsHandler to construct and send the request.
- EXPECT_NO_THROW(name_add->replacingFwdAddrsHandler());
- // Simulate receiving a successful update response.
- name_add->fakeResponse(DNSClient::SUCCESS, dns::Rcode::NOERROR());
- // Run replacingFwdAddrsHandler again to process the response.
- EXPECT_NO_THROW(name_add->replacingFwdAddrsHandler());
- // Forward completion should be true, reverse should be false.
- EXPECT_TRUE(name_add->getForwardChangeCompleted());
- EXPECT_FALSE(name_add->getReverseChangeCompleted());
- // Since it is a forward only change, we should be done.
- // Verify that we transitioned correctly.
- EXPECT_EQ(NameChangeTransaction::PROCESS_TRANS_OK_ST,
- name_add->getCurrState());
- EXPECT_EQ(NameChangeTransaction::UPDATE_OK_EVT,
- name_add->getNextEvent());
- }
- // Tests replacingFwdAddrsHandler with the following scenario:
- //
- // The request includes a forward and reverse change.
- // Initial posted event is FQDN_IN_USE_EVT.
- // The update request is sent without error.
- // A server response is received which indicates successful update.
- //
- TEST_F(NameAddTransactionTest, replacingFwdAddrsHandler_FwdAndRevAddOK) {
- NameAddStubPtr name_add;
- // Create and prep a transaction, poised to run the handler.
- ASSERT_NO_THROW(name_add =
- prepHandlerTest(NameAddTransaction::REPLACING_FWD_ADDRS_ST,
- NameAddTransaction::
- FQDN_IN_USE_EVT, FWD_AND_REV_CHG));
- // Run replacingFwdAddrsHandler to construct and send the request.
- EXPECT_NO_THROW(name_add->replacingFwdAddrsHandler());
- // Simulate receiving a successful update response.
- name_add->fakeResponse(DNSClient::SUCCESS, dns::Rcode::NOERROR());
- // Run replacingFwdAddrsHandler again to process the response.
- EXPECT_NO_THROW(name_add->replacingFwdAddrsHandler());
- // Forward change completion should be true, reverse flag should be false.
- EXPECT_TRUE(name_add->getForwardChangeCompleted());
- EXPECT_FALSE(name_add->getReverseChangeCompleted());
- // Since the request also includes a reverse change we should
- // be poised to start it. Verify that we transitioned correctly.
- EXPECT_EQ(NameChangeTransaction::SELECTING_REV_SERVER_ST,
- name_add->getCurrState());
- EXPECT_EQ(NameChangeTransaction::SELECT_SERVER_EVT,
- name_add->getNextEvent());
- }
- // Tests replacingFwdAddrsHandler with the following scenario:
- //
- // The request includes a forward and reverse change.
- // Initial posted event is FQDN_IN_USE_EVT.
- // The update request is sent without error.
- // A server response is received which indicates the FQDN is NOT in use.
- //
- TEST_F(NameAddTransactionTest, replacingFwdAddrsHandler_FqdnNotInUse) {
- NameAddStubPtr name_add;
- // Create and prep a transaction, poised to run the handler.
- ASSERT_NO_THROW(name_add =
- prepHandlerTest(NameAddTransaction::REPLACING_FWD_ADDRS_ST,
- NameAddTransaction::
- FQDN_IN_USE_EVT, FWD_AND_REV_CHG));
- // Run replacingFwdAddrsHandler to construct and send the request.
- EXPECT_NO_THROW(name_add->replacingFwdAddrsHandler());
- // Simulate receiving a FQDN not in use response.
- name_add->fakeResponse(DNSClient::SUCCESS, dns::Rcode::NXDOMAIN());
- // Run replacingFwdAddrsHandler again to process the response.
- EXPECT_NO_THROW(name_add->replacingFwdAddrsHandler());
- // Completion flags should still be false.
- EXPECT_FALSE(name_add->getForwardChangeCompleted());
- EXPECT_FALSE(name_add->getReverseChangeCompleted());
- // Since the FQDN is no longer in use, per the RFC, try to add it.
- // Verify that we transitioned correctly.
- EXPECT_EQ(NameAddTransaction::ADDING_FWD_ADDRS_ST,
- name_add->getCurrState());
- EXPECT_EQ(NameChangeTransaction::SERVER_SELECTED_EVT,
- name_add->getNextEvent());
- }
- // Tests replacingFwdAddrsHandler with the following scenario:
- //
- // The request includes a forward and reverse change.
- // The update request is sent without error.
- // A server response is received which indicates the update was rejected.
- //
- TEST_F(NameAddTransactionTest, replacingFwdAddrsHandler_OtherRcode) {
- NameAddStubPtr name_add;
- // Create the transaction.
- ASSERT_NO_THROW(name_add =
- prepHandlerTest(NameAddTransaction::REPLACING_FWD_ADDRS_ST,
- NameAddTransaction::
- FQDN_IN_USE_EVT, FWD_AND_REV_CHG));
- // Select a server to satisfy log statements.
- ASSERT_TRUE(name_add->selectFwdServer());
- // Run replacingFwdAddrsHandler to construct and send the request.
- EXPECT_NO_THROW(name_add->replacingFwdAddrsHandler());
- // Simulate receiving server rejection response. Per RFC, anything other
- // than no error or FQDN in use is failure. Arbitrarily choosing refused.
- name_add->fakeResponse(DNSClient::SUCCESS, dns::Rcode::REFUSED());
- // Run replacingFwdAddrsHandler again to process the response.
- EXPECT_NO_THROW(name_add->replacingFwdAddrsHandler());
- // Completion flags should still be false.
- EXPECT_FALSE(name_add->getForwardChangeCompleted());
- EXPECT_FALSE(name_add->getReverseChangeCompleted());
- // We should have failed the transaction. Verifiy that we transitioned
- // correctly.
- EXPECT_EQ(NameChangeTransaction::PROCESS_TRANS_FAILED_ST,
- name_add->getCurrState());
- EXPECT_EQ(NameChangeTransaction::UPDATE_FAILED_EVT,
- name_add->getNextEvent());
- }
- // Tests replacingFwdAddrsHandler with the following scenario:
- //
- // The request includes a forward and reverse change.
- // Initial posted event is FQDN_IN_USE_EVT.
- // The update request send times out MAX_UPDATE_TRIES_PER_SERVER times.
- //
- TEST_F(NameAddTransactionTest, replacingFwdAddrsHandler_Timeout) {
- NameAddStubPtr name_add;
- // Create the transaction.
- ASSERT_NO_THROW(name_add =
- prepHandlerTest(NameAddTransaction::REPLACING_FWD_ADDRS_ST,
- NameAddTransaction::
- FQDN_IN_USE_EVT, FWD_AND_REV_CHG));
- // Select a server to satisfy log statements.
- ASSERT_TRUE(name_add->selectFwdServer());
- // Verify that we can make maximum number of update attempts permitted
- // and then transition to selecting a new server.
- int max_tries = NameChangeTransaction::MAX_UPDATE_TRIES_PER_SERVER;
- for (int i = 1; i <= max_tries; ++i) {
- const D2UpdateMessagePtr prev_msg = name_add->getDnsUpdateRequest();
- // Run replacingFwdAddrsHandler to send the request.
- EXPECT_NO_THROW(name_add->replacingFwdAddrsHandler());
- const D2UpdateMessagePtr curr_msg = name_add->getDnsUpdateRequest();
- if (i == 1) {
- // First time out we should build the message.
- EXPECT_FALSE(prev_msg);
- EXPECT_TRUE(curr_msg);
- } else {
- // Subsequent passes should reuse the request. We are only
- // looking to check that we have not replaced the pointer value
- // with a new pointer. This tests the on_entry() logic which
- // clears the request ONLY upon initial entry into the state.
- EXPECT_TRUE(prev_msg == curr_msg);
- }
- // Simulate a server IO timeout.
- name_add->setDnsUpdateStatus(DNSClient::TIMEOUT);
- name_add->postNextEvent(NameChangeTransaction::IO_COMPLETED_EVT);
- // Run replacingFwdAddrsHandler again to process the response.
- EXPECT_NO_THROW(name_add->replacingFwdAddrsHandler());
- // Completion flags should be false.
- EXPECT_FALSE(name_add->getForwardChangeCompleted());
- EXPECT_FALSE(name_add->getReverseChangeCompleted());
- if (i < max_tries) {
- // We should be ready to try again.
- EXPECT_EQ(NameAddTransaction::REPLACING_FWD_ADDRS_ST,
- name_add->getCurrState());
- EXPECT_EQ(NameChangeTransaction::SERVER_SELECTED_EVT,
- name_add->getNextEvent());
- } else {
- // Server retries should be exhausted, time for a new server.
- EXPECT_EQ(NameAddTransaction::SELECTING_FWD_SERVER_ST,
- name_add->getCurrState());
- EXPECT_EQ(NameChangeTransaction::SERVER_IO_ERROR_EVT,
- name_add->getNextEvent());
- }
- }
- }
- // Tests replacingFwdAddrsHandler with the following scenario:
- //
- // The request includes a forward and reverse change.
- // Initial posted event is FQDN_IN_USE_EVT.
- // The update request is sent but a corrupt response is received, this occurs
- // MAX_UPDATE_TRIES_PER_SERVER times.
- //
- TEST_F(NameAddTransactionTest, replacingFwdAddrsHandler_CorruptResponse) {
- NameAddStubPtr name_add;
- // Create the transaction.
- ASSERT_NO_THROW(name_add =
- prepHandlerTest(NameAddTransaction::REPLACING_FWD_ADDRS_ST,
- NameAddTransaction::
- FQDN_IN_USE_EVT, FWD_AND_REV_CHG));
- // Select a server to satisfy log statements.
- ASSERT_TRUE(name_add->selectFwdServer());
- // Verify that we can make maximum number of update attempts permitted
- // and then transition to selecting a new server.
- int max_tries = NameChangeTransaction::MAX_UPDATE_TRIES_PER_SERVER;
- for (int i = 1; i <= max_tries; ++i) {
- const D2UpdateMessagePtr prev_msg = name_add->getDnsUpdateRequest();
- // Run replacingFwdAddrsHandler to send the request.
- EXPECT_NO_THROW(name_add->replacingFwdAddrsHandler());
- const D2UpdateMessagePtr curr_msg = name_add->getDnsUpdateRequest();
- if (i == 1) {
- // First time out we should build the message.
- EXPECT_FALSE(prev_msg);
- EXPECT_TRUE(curr_msg);
- } else {
- // Subsequent passes should reuse the request. We are only
- // looking to check that we have not replaced the pointer value
- // with a new pointer. This tests the on_entry() logic which
- // clears the request ONLY upon initial entry into the state.
- EXPECT_TRUE(prev_msg == curr_msg);
- }
- // Simulate a corrupt server response.
- name_add->setDnsUpdateStatus(DNSClient::INVALID_RESPONSE);
- name_add->postNextEvent(NameChangeTransaction::IO_COMPLETED_EVT);
- // Run replacingFwdAddrsHandler again to process the response.
- EXPECT_NO_THROW(name_add->replacingFwdAddrsHandler());
- // Completion flags should be false.
- EXPECT_FALSE(name_add->getForwardChangeCompleted());
- EXPECT_FALSE(name_add->getReverseChangeCompleted());
- if (i < max_tries) {
- // We should be ready to try again.
- EXPECT_EQ(NameAddTransaction::REPLACING_FWD_ADDRS_ST,
- name_add->getCurrState());
- EXPECT_EQ(NameChangeTransaction::SERVER_SELECTED_EVT,
- name_add->getNextEvent());
- } else {
- // Server retries should be exhausted, time for a new server.
- EXPECT_EQ(NameAddTransaction::SELECTING_FWD_SERVER_ST,
- name_add->getCurrState());
- EXPECT_EQ(NameChangeTransaction::SERVER_IO_ERROR_EVT,
- name_add->getNextEvent());
- }
- }
- }
- // Tests the selectingRevServerHandler functionality.
- // It verifies behavior for the following scenarios:
- //
- // 1. Posted event is SELECT_SERVER_EVT
- // 2. Posted event is SERVER_IO_ERROR_EVT
- // 3. Posted event is invalid
- //
- TEST_F(NameAddTransactionTest, selectingRevServerHandler) {
- NameAddStubPtr name_add;
- // Create and prep a transaction, poised to run the handler.
- ASSERT_NO_THROW(name_add =
- prepHandlerTest(NameChangeTransaction::
- SELECTING_REV_SERVER_ST,
- NameChangeTransaction::SELECT_SERVER_EVT));
- // Call selectingRevServerHandler enough times to select all of the
- // servers in it's current domain. The first time, it will be with
- // next event of SELECT_SERVER_EVT. Thereafter it will be with a next
- // event of SERVER_IO_ERROR_EVT.
- int num_servers = name_add->getReverseDomain()->getServers()->size();
- for (int i = 0; i < num_servers; ++i) {
- // Run selectingRevServerHandler.
- ASSERT_NO_THROW(name_add->selectingRevServerHandler())
- << " num_servers: " << num_servers
- << " selections: " << i;
- // Verify that a server was selected.
- ASSERT_TRUE(name_add->getCurrentServer())
- << " num_servers: " << num_servers
- << " selections: " << i;
- // Verify that we transitioned correctly.
- ASSERT_EQ(NameAddTransaction::REPLACING_REV_PTRS_ST,
- name_add->getCurrState())
- << " num_servers: " << num_servers << " selections: " << i;
- ASSERT_EQ(NameChangeTransaction::SERVER_SELECTED_EVT,
- name_add->getNextEvent())
- << " num_servers: " << num_servers << " selections: " << i;
- // Post a server IO error event. This simulates an IO error occuring
- // and a need to select the new server.
- ASSERT_NO_THROW(name_add->postNextEvent(NameChangeTransaction::
- SERVER_IO_ERROR_EVT))
- << " num_servers: " << num_servers
- << " selections: " << i;
- }
- // We should have exhausted the list of servers. Processing another
- // SERVER_IO_ERROR_EVT should transition us to failure.
- EXPECT_NO_THROW(name_add->selectingRevServerHandler());
- EXPECT_EQ(NameChangeTransaction::PROCESS_TRANS_FAILED_ST,
- name_add->getCurrState());
- EXPECT_EQ(NameChangeTransaction::NO_MORE_SERVERS_EVT,
- name_add->getNextEvent());
- // Create and prep transaction, poised to run the handler but with an
- // invalid event.
- ASSERT_NO_THROW(name_add =
- prepHandlerTest(NameChangeTransaction::
- SELECTING_REV_SERVER_ST,
- StateModel::NOP_EVT));
- // Running the handler should throw.
- EXPECT_THROW(name_add->selectingRevServerHandler(),
- NameAddTransactionError);
- }
- //************************** replacingRevPtrsHandler tests *****************
- // Tests that replacingRevPtrsHandler rejects invalid events.
- TEST_F(NameAddTransactionTest, replacingRevPtrsHandler_InvalidEvent) {
- NameAddStubPtr name_add;
- // Create and prep a transaction, poised to run the handler but with
- // an invalid event.
- ASSERT_NO_THROW(name_add =
- prepHandlerTest(NameAddTransaction::REPLACING_REV_PTRS_ST,
- NameChangeTransaction::
- StateModel::NOP_EVT));
- // Running the handler should throw.
- EXPECT_THROW(name_add->replacingRevPtrsHandler(),
- NameAddTransactionError);
- }
- // Tests replacingRevPtrsHandler with the following scenario:
- //
- // The request includes only a reverse change.
- // Initial posted event is SERVER_SELECTED_EVT.
- // The update request is sent without error.
- // A server response is received which indicates successful update.
- //
- TEST_F(NameAddTransactionTest, replacingRevPtrsHandler_FwdOnlyAddOK) {
- NameAddStubPtr name_add;
- // Create and prep a transaction, poised to run the handler.
- ASSERT_NO_THROW(name_add =
- prepHandlerTest(NameAddTransaction::REPLACING_REV_PTRS_ST,
- NameAddTransaction::
- SERVER_SELECTED_EVT, REVERSE_CHG));
- // Should not be an update message yet.
- D2UpdateMessagePtr update_msg = name_add->getDnsUpdateRequest();
- ASSERT_FALSE(update_msg);
- // At this point completion flags should be false.
- EXPECT_FALSE(name_add->getForwardChangeCompleted());
- EXPECT_FALSE(name_add->getReverseChangeCompleted());
- // Run replacingRevPtrsHandler to construct and send the request.
- EXPECT_NO_THROW(name_add->replacingRevPtrsHandler());
- // Verify that an update message was constructed properly.
- checkReplaceRevPtrsRequest(*name_add);
- // Verify that we are still in this state and next event is NOP_EVT.
- // This indicates we "sent" the message and are waiting for IO completion.
- EXPECT_EQ(NameAddTransaction::REPLACING_REV_PTRS_ST,
- name_add->getCurrState());
- EXPECT_EQ(NameChangeTransaction::NOP_EVT,
- name_add->getNextEvent());
- // Simulate receiving a successful update response.
- name_add->fakeResponse(DNSClient::SUCCESS, dns::Rcode::NOERROR());
- // Run replacingRevPtrsHandler again to process the response.
- EXPECT_NO_THROW(name_add->replacingRevPtrsHandler());
- // Forward completion should be false, reverse should be true.
- EXPECT_FALSE(name_add->getForwardChangeCompleted());
- EXPECT_TRUE(name_add->getReverseChangeCompleted());
- // Since it is a reverse change, we should be done.
- // Verify that we transitioned correctly.
- EXPECT_EQ(NameChangeTransaction::PROCESS_TRANS_OK_ST,
- name_add->getCurrState());
- EXPECT_EQ(NameChangeTransaction::UPDATE_OK_EVT,
- name_add->getNextEvent());
- }
- // Tests replacingRevPtrsHandler with the following scenario:
- //
- // The request includes only a reverse change.
- // Initial posted event is SERVER_SELECTED_EVT.
- // The update request is sent without error.
- // A server response is received which indicates the update was rejected.
- //
- TEST_F(NameAddTransactionTest, replacingRevPtrsHandler_OtherRcode) {
- NameAddStubPtr name_add;
- // Create the transaction.
- ASSERT_NO_THROW(name_add =
- prepHandlerTest(NameAddTransaction::REPLACING_REV_PTRS_ST,
- NameAddTransaction::
- SERVER_SELECTED_EVT, REVERSE_CHG));
- // Select a server to satisfy log statements.
- ASSERT_TRUE(name_add->selectRevServer());
- // Run replacingRevPtrsHandler to construct and send the request.
- EXPECT_NO_THROW(name_add->replacingRevPtrsHandler());
- // Simulate receiving server rejection response. Per RFC, anything other
- // than no error is failure. Arbitrarily choosing refused.
- name_add->fakeResponse(DNSClient::SUCCESS, dns::Rcode::REFUSED());
- // Run replacingRevPtrsHandler again to process the response.
- EXPECT_NO_THROW(name_add->replacingRevPtrsHandler());
- // Completion flags should still be false.
- EXPECT_FALSE(name_add->getForwardChangeCompleted());
- EXPECT_FALSE(name_add->getReverseChangeCompleted());
- // We should have failed the transaction. Verify that we transitioned
- // correctly.
- EXPECT_EQ(NameChangeTransaction::PROCESS_TRANS_FAILED_ST,
- name_add->getCurrState());
- EXPECT_EQ(NameChangeTransaction::UPDATE_FAILED_EVT,
- name_add->getNextEvent());
- }
- // Tests replacingRevPtrsHandler with the following scenario:
- //
- // The request includes only a reverse change.
- // Initial posted event is SERVER_SELECTED_EVT.
- // The update request send times out MAX_UPDATE_TRIES_PER_SERVER times.
- //
- TEST_F(NameAddTransactionTest, replacingRevPtrsHandler_Timeout) {
- NameAddStubPtr name_add;
- // Create the transaction.
- ASSERT_NO_THROW(name_add =
- prepHandlerTest(NameAddTransaction::REPLACING_REV_PTRS_ST,
- NameAddTransaction::
- SERVER_SELECTED_EVT, REVERSE_CHG));
- // Select a server to satisfy log statements.
- ASSERT_TRUE(name_add->selectRevServer());
- // Verify that we can make maximum number of update attempts permitted
- // and then transition to selecting a new server.
- int max_tries = NameChangeTransaction::MAX_UPDATE_TRIES_PER_SERVER;
- for (int i = 1; i <= max_tries; ++i) {
- const D2UpdateMessagePtr prev_msg = name_add->getDnsUpdateRequest();
- // Run replacingRevPtrsHandler to send the request.
- EXPECT_NO_THROW(name_add->replacingRevPtrsHandler());
- const D2UpdateMessagePtr curr_msg = name_add->getDnsUpdateRequest();
- if (i == 1) {
- // First time out we should build the message.
- EXPECT_FALSE(prev_msg);
- EXPECT_TRUE(curr_msg);
- } else {
- // Subsequent passes should reuse the request. We are only
- // looking to check that we have not replaced the pointer value
- // with a new pointer. This tests the on_entry() logic which
- // clears the request ONLY upon initial entry into the state.
- EXPECT_TRUE(prev_msg == curr_msg);
- }
- // Simulate a server IO timeout.
- name_add->setDnsUpdateStatus(DNSClient::TIMEOUT);
- name_add->postNextEvent(NameChangeTransaction::IO_COMPLETED_EVT);
- // Run replacingRevPtrsHandler again to process the response.
- EXPECT_NO_THROW(name_add->replacingRevPtrsHandler());
- // Completion flags should be false.
- EXPECT_FALSE(name_add->getForwardChangeCompleted());
- EXPECT_FALSE(name_add->getReverseChangeCompleted());
- if (i < max_tries) {
- // We should be ready to try again.
- EXPECT_EQ(NameAddTransaction::REPLACING_REV_PTRS_ST,
- name_add->getCurrState());
- EXPECT_EQ(NameChangeTransaction::SERVER_SELECTED_EVT,
- name_add->getNextEvent());
- } else {
- // Server retries should be exhausted, time for a new server.
- EXPECT_EQ(NameAddTransaction::SELECTING_REV_SERVER_ST,
- name_add->getCurrState());
- EXPECT_EQ(NameChangeTransaction::SERVER_IO_ERROR_EVT,
- name_add->getNextEvent());
- }
- }
- }
- // Tests replacingRevPtrsHandler with the following scenario:
- //
- // The request includes only a reverse change.
- // Initial posted event is SERVER_SELECTED_EVT.
- // The update request is sent but a corrupt response is received, this occurs
- // MAX_UPDATE_TRIES_PER_SERVER times.
- //
- TEST_F(NameAddTransactionTest, replacingRevPtrsHandler_CorruptResponse) {
- NameAddStubPtr name_add;
- // Create the transaction.
- ASSERT_NO_THROW(name_add =
- prepHandlerTest(NameAddTransaction::REPLACING_REV_PTRS_ST,
- NameAddTransaction::
- SERVER_SELECTED_EVT, REVERSE_CHG));
- // Select a server to satisfy log statements.
- ASSERT_TRUE(name_add->selectRevServer());
- // Verify that we can make maximum number of update attempts permitted
- // and then transition to selecting a new server.
- int max_tries = NameChangeTransaction::MAX_UPDATE_TRIES_PER_SERVER;
- for (int i = 1; i <= max_tries; ++i) {
- const D2UpdateMessagePtr prev_msg = name_add->getDnsUpdateRequest();
- // Run replacingRevPtrsHandler to send the request.
- EXPECT_NO_THROW(name_add->replacingRevPtrsHandler());
- const D2UpdateMessagePtr curr_msg = name_add->getDnsUpdateRequest();
- if (i == 1) {
- // First time out we should build the message.
- EXPECT_FALSE(prev_msg);
- EXPECT_TRUE(curr_msg);
- } else {
- // Subsequent passes should reuse the request. We are only
- // looking to check that we have not replaced the pointer value
- // with a new pointer. This tests the on_entry() logic which
- // clears the request ONLY upon initial entry into the state.
- EXPECT_TRUE(prev_msg == curr_msg);
- }
- // Simulate a server corrupt response.
- name_add->setDnsUpdateStatus(DNSClient::INVALID_RESPONSE);
- name_add->postNextEvent(NameChangeTransaction::IO_COMPLETED_EVT);
- // Run replacingRevPtrsHandler again to process the response.
- EXPECT_NO_THROW(name_add->replacingRevPtrsHandler());
- // Completion flags should be false.
- EXPECT_FALSE(name_add->getForwardChangeCompleted());
- EXPECT_FALSE(name_add->getReverseChangeCompleted());
- if (i < max_tries) {
- // We should be ready to try again.
- EXPECT_EQ(NameAddTransaction::REPLACING_REV_PTRS_ST,
- name_add->getCurrState());
- EXPECT_EQ(NameChangeTransaction::SERVER_SELECTED_EVT,
- name_add->getNextEvent());
- } else {
- // Server retries should be exhausted, time for a new server.
- EXPECT_EQ(NameAddTransaction::SELECTING_REV_SERVER_ST,
- name_add->getCurrState());
- EXPECT_EQ(NameChangeTransaction::SERVER_IO_ERROR_EVT,
- name_add->getNextEvent());
- }
- }
- }
- // Tests the processAddOkHandler functionality.
- // It verifies behavior for the following scenarios:
- //
- // 1. Posted event is UPDATE_OK_EVT
- // 2. Posted event is invalid
- //
- TEST_F(NameAddTransactionTest, processAddOkHandler) {
- NameAddStubPtr name_add;
- // Create and prep a transaction, poised to run the handler.
- ASSERT_NO_THROW(name_add =
- prepHandlerTest(NameChangeTransaction::PROCESS_TRANS_OK_ST,
- NameChangeTransaction::UPDATE_OK_EVT));
- // Run processAddOkHandler.
- EXPECT_NO_THROW(name_add->processAddOkHandler());
- // Verify that a server was selected.
- EXPECT_EQ(dhcp_ddns::ST_COMPLETED, name_add->getNcrStatus());
- // Verify that the model has ended.
- EXPECT_EQ(StateModel::END_ST, name_add->getCurrState());
- EXPECT_EQ(StateModel::END_EVT, name_add->getNextEvent());
- // Create and prep transaction, poised to run the handler but with an
- // invalid event.
- ASSERT_NO_THROW(name_add =
- prepHandlerTest(NameChangeTransaction::PROCESS_TRANS_OK_ST,
- StateModel::NOP_EVT));
- // Running the handler should throw.
- EXPECT_THROW(name_add->processAddOkHandler(), NameAddTransactionError);
- }
- // Tests the processAddFailedHandler functionality.
- // It verifies behavior for the following scenarios:
- //
- // 1. Posted event is UPDATE_FAILED_EVT
- // 2. Posted event is invalid
- //
- TEST_F(NameAddTransactionTest, processAddFailedHandler) {
- NameAddStubPtr name_add;
- // Create and prep a transaction, poised to run the handler.
- ASSERT_NO_THROW(name_add =
- prepHandlerTest(NameChangeTransaction::
- PROCESS_TRANS_FAILED_ST,
- NameChangeTransaction::UPDATE_FAILED_EVT));
- // Run processAddFailedHandler.
- EXPECT_NO_THROW(name_add->processAddFailedHandler());
- // Verify that a server was selected.
- EXPECT_EQ(dhcp_ddns::ST_FAILED, name_add->getNcrStatus());
- // Verify that the model has ended. (Remember, the transaction failed NOT
- // the model. The model should have ended normally.)
- EXPECT_EQ(StateModel::END_ST, name_add->getCurrState());
- EXPECT_EQ(StateModel::END_EVT, name_add->getNextEvent());
- // Create and prep transaction, poised to run the handler but with an
- // invalid event.
- ASSERT_NO_THROW(name_add =
- prepHandlerTest(NameChangeTransaction::
- PROCESS_TRANS_FAILED_ST,
- StateModel::NOP_EVT));
- // Running the handler should throw.
- EXPECT_THROW(name_add->processAddFailedHandler(), NameAddTransactionError);
- }
- // Tests the processAddFailedHandler functionality.
- // It verifies behavior for posted event of NO_MORE_SERVERS_EVT.
- TEST_F(NameAddTransactionTest, processAddFailedHandler_NoMoreServers) {
- NameAddStubPtr name_remove;
- // Create and prep a transaction, poised to run the handler.
- ASSERT_NO_THROW(name_remove =
- prepHandlerTest(NameChangeTransaction::
- PROCESS_TRANS_FAILED_ST,
- NameChangeTransaction::
- NO_MORE_SERVERS_EVT));
- // Run processAddFailedHandler.
- EXPECT_NO_THROW(name_remove->processAddFailedHandler());
- // Verify that a server was selected.
- EXPECT_EQ(dhcp_ddns::ST_FAILED, name_remove->getNcrStatus());
- // Verify that the model has ended. (Remember, the transaction failed NOT
- // the model. The model should have ended normally.)
- EXPECT_EQ(StateModel::END_ST, name_remove->getCurrState());
- EXPECT_EQ(StateModel::END_EVT, name_remove->getNextEvent());
- }
- // Tests addingFwdAddrsHandler with the following scenario:
- //
- // The request includes only a forward change.
- // Initial posted event is SERVER_SELECTED_EVT.
- // The send update request fails due to an unexpected exception.
- //
- TEST_F(NameAddTransactionTest, addingFwdAddrsHandler_sendUpdateException) {
- NameAddStubPtr name_add;
- // Create and prep a transaction, poised to run the handler.
- ASSERT_NO_THROW(name_add =
- prepHandlerTest(NameAddTransaction::ADDING_FWD_ADDRS_ST,
- NameChangeTransaction::
- SERVER_SELECTED_EVT, FORWARD_CHG));
- name_add->simulate_send_exception_ = true;
- // Run replacingFwdAddrsHandler to construct and send the request.
- ASSERT_NO_THROW(name_add->addingFwdAddrsHandler());
- // Completion flags should be false.
- EXPECT_FALSE(name_add->getForwardChangeCompleted());
- EXPECT_FALSE(name_add->getReverseChangeCompleted());
- // Since IO exceptions should be gracefully handled, any that occur
- // are unanticipated, and deemed unrecoverable, so the transaction should
- // be transitioned to failure.
- EXPECT_EQ(NameChangeTransaction::PROCESS_TRANS_FAILED_ST,
- name_add->getCurrState());
- EXPECT_EQ(NameChangeTransaction::UPDATE_FAILED_EVT,
- name_add->getNextEvent());
- }
- // Tests replacingFwdAddrsHandler with the following scenario:
- //
- // The request includes only a forward change.
- // Initial posted event is SERVER_SELECTED_EVT.
- // The send update request fails due to an unexpected exception.
- //
- TEST_F(NameAddTransactionTest, replacingFwdAddrsHandler_SendUpdateException) {
- NameAddStubPtr name_add;
- // Create and prep a transaction, poised to run the handler.
- ASSERT_NO_THROW(name_add =
- prepHandlerTest(NameAddTransaction::REPLACING_FWD_ADDRS_ST,
- NameChangeTransaction::
- SERVER_SELECTED_EVT, FORWARD_CHG));
- name_add->simulate_send_exception_ = true;
- // Run replacingFwdAddrsHandler to construct and send the request.
- ASSERT_NO_THROW(name_add->replacingFwdAddrsHandler());
- // Completion flags should be false.
- EXPECT_FALSE(name_add->getForwardChangeCompleted());
- EXPECT_FALSE(name_add->getReverseChangeCompleted());
- // Since IO exceptions should be gracefully handled, any that occur
- // are unanticipated, and deemed unrecoverable, so the transaction should
- // be transitioned to failure.
- EXPECT_EQ(NameChangeTransaction::PROCESS_TRANS_FAILED_ST,
- name_add->getCurrState());
- EXPECT_EQ(NameChangeTransaction::UPDATE_FAILED_EVT,
- name_add->getNextEvent());
- }
- // Tests replacingRevPtrHandler with the following scenario:
- //
- // The request includes only a reverse change.
- // Initial posted event is SERVER_SELECTED_EVT.
- // The send update request fails due to an unexpected exception.
- //
- TEST_F(NameAddTransactionTest, replacingRevPtrsHandler_SendUpdateException) {
- NameAddStubPtr name_add;
- // Create and prep a transaction, poised to run the handler.
- ASSERT_NO_THROW(name_add =
- prepHandlerTest(NameAddTransaction::REPLACING_REV_PTRS_ST,
- NameChangeTransaction::
- SERVER_SELECTED_EVT, REVERSE_CHG));
- name_add->simulate_send_exception_ = true;
- // Run replacingRevPtrsHandler to construct and send the request.
- ASSERT_NO_THROW(name_add->replacingRevPtrsHandler());
- // Completion flags should be false.
- EXPECT_FALSE(name_add->getForwardChangeCompleted());
- EXPECT_FALSE(name_add->getReverseChangeCompleted());
- // Since IO exceptions should be gracefully handled, any that occur
- // are unanticipated, and deemed unrecoverable, so the transaction should
- // be transitioned to failure.
- EXPECT_EQ(NameChangeTransaction::PROCESS_TRANS_FAILED_ST,
- name_add->getCurrState());
- EXPECT_EQ(NameChangeTransaction::UPDATE_FAILED_EVT,
- name_add->getNextEvent());
- }
- // Tests addingFwdAddrsHandler with the following scenario:
- //
- // The request includes only a forward change.
- // Initial posted event is SERVER_SELECTED_EVT.
- // The request build fails due to an unexpected exception.
- //
- TEST_F(NameAddTransactionTest, addingFwdAddrsHandler_BuildRequestException) {
- NameAddStubPtr name_add;
- // Create and prep a transaction, poised to run the handler.
- ASSERT_NO_THROW(name_add =
- prepHandlerTest(NameAddTransaction::ADDING_FWD_ADDRS_ST,
- NameChangeTransaction::
- SERVER_SELECTED_EVT, FORWARD_CHG));
- // Set the one-shot exception simulation flag.
- name_add->simulate_build_request_exception_ = true;
- // Run replacingRevPtrsHandler to construct and send the request.
- // This should fail with a build request throw which should be caught
- // in the state handler.
- ASSERT_NO_THROW(name_add->addingFwdAddrsHandler());
- // Verify we did not attempt to send anything.
- EXPECT_EQ(0, name_add->getUpdateAttempts());
- // Completion flags should be false.
- EXPECT_FALSE(name_add->getForwardChangeCompleted());
- EXPECT_FALSE(name_add->getReverseChangeCompleted());
- // Since IO exceptions should be gracefully handled, any that occur
- // are unanticipated, and deemed unrecoverable, so the transaction should
- // be transitioned to failure.
- EXPECT_EQ(NameChangeTransaction::PROCESS_TRANS_FAILED_ST,
- name_add->getCurrState());
- EXPECT_EQ(NameChangeTransaction::UPDATE_FAILED_EVT,
- name_add->getNextEvent());
- }
- // Tests replacingFwdAddrsHandler with the following scenario:
- //
- // The request includes only a forward change.
- // Initial posted event is SERVER_SELECTED_EVT.
- // The request build fails due to an unexpected exception.
- //
- TEST_F(NameAddTransactionTest, replacingFwdAddrsHandler_BuildRequestException) {
- NameAddStubPtr name_add;
- // Create and prep a transaction, poised to run the handler.
- ASSERT_NO_THROW(name_add =
- prepHandlerTest(NameAddTransaction::REPLACING_FWD_ADDRS_ST,
- NameChangeTransaction::
- SERVER_SELECTED_EVT, FORWARD_CHG));
- // Set the one-shot exception simulation flag.
- name_add->simulate_build_request_exception_ = true;
- // Run replacingFwdAddrsHandler to construct and send the request.
- // This should fail with a build request throw which should be caught
- // in the state handler.
- ASSERT_NO_THROW(name_add->replacingFwdAddrsHandler());
- // Verify we did not attempt to send anything.
- EXPECT_EQ(0, name_add->getUpdateAttempts());
- // Completion flags should be false.
- EXPECT_FALSE(name_add->getForwardChangeCompleted());
- EXPECT_FALSE(name_add->getReverseChangeCompleted());
- // Since IO exceptions should be gracefully handled, any that occur
- // are unanticipated, and deemed unrecoverable, so the transaction should
- // be transitioned to failure.
- EXPECT_EQ(NameChangeTransaction::PROCESS_TRANS_FAILED_ST,
- name_add->getCurrState());
- EXPECT_EQ(NameChangeTransaction::UPDATE_FAILED_EVT,
- name_add->getNextEvent());
- }
- // Tests replacingRevPtrHandler with the following scenario:
- //
- // The request includes only a reverse change.
- // Initial posted event is SERVER_SELECTED_EVT.
- // The request build fails due to an unexpected exception.
- //
- TEST_F(NameAddTransactionTest, replacingRevPtrsHandler_BuildRequestException) {
- NameAddStubPtr name_add;
- // Create and prep a transaction, poised to run the handler.
- ASSERT_NO_THROW(name_add =
- prepHandlerTest(NameAddTransaction::REPLACING_REV_PTRS_ST,
- NameChangeTransaction::
- SERVER_SELECTED_EVT, REVERSE_CHG));
- // Set the one-shot exception simulation flag.
- name_add->simulate_build_request_exception_ = true;
- // Run replacingRevPtrsHandler to construct and send the request.
- // This should fail with a build request throw which should be caught
- // in the state handler.
- ASSERT_NO_THROW(name_add->replacingRevPtrsHandler());
- // Verify we did not attempt to send anything.
- EXPECT_EQ(0, name_add->getUpdateAttempts());
- // Completion flags should be false.
- EXPECT_FALSE(name_add->getForwardChangeCompleted());
- EXPECT_FALSE(name_add->getReverseChangeCompleted());
- // Since IO exceptions should be gracefully handled, any that occur
- // are unanticipated, and deemed unrecoverable, so the transaction should
- // be transitioned to failure.
- EXPECT_EQ(NameChangeTransaction::PROCESS_TRANS_FAILED_ST,
- name_add->getCurrState());
- EXPECT_EQ(NameChangeTransaction::UPDATE_FAILED_EVT,
- name_add->getNextEvent());
- }
- }
|