|
@@ -0,0 +1,71 @@
|
|
|
+// 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 <config.h>
|
|
|
+#include <string>
|
|
|
+#include <gtest/gtest.h>
|
|
|
+#include "../message_cache.h"
|
|
|
+#include "../rrset_cache.h"
|
|
|
+#include "../recursor_cache.h"
|
|
|
+#include <dns/tests/unittest_util.h>
|
|
|
+#include <dns/buffer.h>
|
|
|
+
|
|
|
+using namespace isc::cache;
|
|
|
+using namespace isc;
|
|
|
+using namespace isc::dns;
|
|
|
+using namespace std;
|
|
|
+
|
|
|
+namespace {
|
|
|
+
|
|
|
+class MessageCacheTest: public testing::Test {
|
|
|
+public:
|
|
|
+ MessageCacheTest(): message_parse(Message::PARSE),
|
|
|
+ message_render(Message::RENDER)
|
|
|
+ {
|
|
|
+ uint16_t class_ = 1; // class IN
|
|
|
+ rrset_cache_.reset(new RRsetCache(RRSET_CACHE_DEFAULT_SIZE, class_));
|
|
|
+ message_cache_.reset(new MessageCache(rrset_cache_,
|
|
|
+ MESSAGE_CACHE_DEFAULT_SIZE, class_ ));
|
|
|
+ }
|
|
|
+
|
|
|
+ void messageFromFile(Message& message, const char* datafile);
|
|
|
+
|
|
|
+protected:
|
|
|
+ MessageCachePtr message_cache_;
|
|
|
+ RRsetCachePtr rrset_cache_;
|
|
|
+ Message message_parse;
|
|
|
+ Message message_render;
|
|
|
+};
|
|
|
+
|
|
|
+void
|
|
|
+MessageCacheTest::messageFromFile(Message& message, const char* datafile) {
|
|
|
+ std::vector<unsigned char> data;
|
|
|
+ UnitTestUtil::readWireData(datafile, data);
|
|
|
+
|
|
|
+ InputBuffer buffer(&data[0], data.size());
|
|
|
+ message.fromWire(buffer);
|
|
|
+}
|
|
|
+
|
|
|
+TEST_F(MessageCacheTest, testUpdate) {
|
|
|
+ messageFromFile(message_parse, "message_fromWire1");
|
|
|
+ EXPECT_TRUE(message_cache_->update(message_parse));
|
|
|
+ Name qname("test.example.com.");
|
|
|
+ RRType qtype(1);
|
|
|
+
|
|
|
+ EXPECT_TRUE(message_cache_->lookup(qname, qtype, message_render));
|
|
|
+}
|
|
|
+
|
|
|
+} // namespace
|
|
|
+
|