Add test for reset method

peterbencze · Jun 11, 2020 · 40ae6ae · 40ae6ae
1 parent f85dd78
commit 40ae6ae
Showing 1 changed file with 36 additions and 0 deletions.
diff --git a/src/test/java/com/github/peterbencze/serritor/internal/stats/StatsCounterTest.java b/src/test/java/com/github/peterbencze/serritor/internal/stats/StatsCounterTest.java
@@ -16,6 +16,7 @@
 
 package com.github.peterbencze.serritor.internal.stats;
 
+import org.hamcrest.Matchers;
 import org.junit.Assert;
 import org.junit.Before;
 import org.junit.Test;
@@ -133,4 +134,39 @@ public void testRecordNetworkError() {
         Assert.assertEquals(processedCrawlCandidateCountBefore + 1,
                 statsCounter.getProcessedCrawlCandidateCount());
     }
+
+    @Test
+    public void testReset() {
+        statsCounter.recordRemainingCrawlCandidate();
+        statsCounter.recordRemainingCrawlCandidate();
+        statsCounter.recordRemainingCrawlCandidate();
+        statsCounter.recordRemainingCrawlCandidate();
+        statsCounter.recordRemainingCrawlCandidate();
+        statsCounter.recordRemainingCrawlCandidate();
+        statsCounter.recordRemainingCrawlCandidate();
+        statsCounter.recordResponseSuccess();
+        statsCounter.recordPageLoadTimeout();
+        statsCounter.recordRequestRedirect();
+        statsCounter.recordNonHtmlResponse();
+        statsCounter.recordResponseError();
+        statsCounter.recordNetworkError();
+        statsCounter.recordDuplicateRequest();
+        statsCounter.recordOffsiteRequest();
+        statsCounter.recordCrawlDepthLimitExceedingRequest();
+
+        statsCounter.reset();
+
+        Assert.assertThat(statsCounter.getRemainingCrawlCandidateCount(), Matchers.is(0));
+        Assert.assertThat(statsCounter.getProcessedCrawlCandidateCount(), Matchers.is(0));
+        Assert.assertThat(statsCounter.getResponseSuccessCount(), Matchers.is(0));
+        Assert.assertThat(statsCounter.getPageLoadTimeoutCount(), Matchers.is(0));
+        Assert.assertThat(statsCounter.getRequestRedirectCount(), Matchers.is(0));
+        Assert.assertThat(statsCounter.getNonHtmlResponseCount(), Matchers.is(0));
+        Assert.assertThat(statsCounter.getResponseErrorCount(), Matchers.is(0));
+        Assert.assertThat(statsCounter.getNetworkErrorCount(), Matchers.is(0));
+        Assert.assertThat(statsCounter.getFilteredDuplicateRequestCount(), Matchers.is(0));
+        Assert.assertThat(statsCounter.getFilteredOffsiteRequestCount(), Matchers.is(0));
+        Assert.assertThat(statsCounter.getFilteredCrawlDepthLimitExceedingRequestCount(),
+                Matchers.is(0));
+    }
 }