kea/src/lib/cache/tests/cache_test_sectioncount.h

// Copyright (C) 2010  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.

// $Id$
#include <vector>
#include <dns/tests/unittest_util.h>
#include <dns/buffer.h>
#include <dns/message.h>

using namespace isc;
using namespace isc::dns;

namespace {

/// \brief Counts the number of rrsets in the given section
///
/// \param msg The message to count in
/// \param section The section to count
///
/// \return The number of RRsets in the given section
int
sectionRRsetCount(Message& msg, Message::Section section) {
    int count = 0;
    for (RRsetIterator rrset_iter = msg.beginSection(section);
         rrset_iter != msg.endSection(section);
         ++rrset_iter) {
        ++count;
    }

    return count;
}

}   // namespace