Merge pull request #31 from seomoz/add-crawl-delay-to-str

Add Crawl-Delay to string representation of Agent.
seomoz · Nov 17, 2017 · 8abadd8 · 8abadd8
2 parents 7046f17 + dc98f75
commit 8abadd8
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/src/agent.cpp b/src/agent.cpp
@@ -1,4 +1,5 @@
 #include <algorithm>
+#include <iomanip>
 #include <sstream>
 
 #include "url.h"
@@ -89,6 +90,10 @@ namespace Rep
     std::string Agent::str() const
     {
         std::stringstream out;
+        if (delay_ > 0)
+        {
+            out << "Crawl-Delay: " << std::setprecision(3) << delay_ << ' ';
+        }
         out << '[';
         const auto& d = directives();
         auto begin = d.begin();

diff --git a/test/test-agent.cpp b/test/test-agent.cpp
@@ -116,3 +116,12 @@ TEST(AgentTest, Str)
     agent.allow("/bar");
     EXPECT_EQ("[Directive(Disallow: /foo), Directive(Allow: /bar)]", agent.str());
 }
+
+TEST(AgentTest, StrWithDelay)
+{
+    Rep::Agent agent("a.com");
+    agent.delay(1.0);
+    agent.disallow("/foo");
+    agent.allow("/bar");
+    EXPECT_EQ("Crawl-Delay: 1 [Directive(Disallow: /foo), Directive(Allow: /bar)]", agent.str());
+}