From b9fa88961ddc00c658a7d79011369f0c27e279bf Mon Sep 17 00:00:00 2001
From: Gabe <7622243+decentralgabe@users.noreply.github.com>
Date: Fri, 12 Apr 2024 10:08:20 -0700
Subject: [PATCH] add test for spam filter (#181)

* add test for spam filter

* remove test
---
 impl/pkg/server/pkarr.go      |  2 +-
 impl/pkg/server/pkarr_test.go | 15 +++++++++++++++
 impl/pkg/service/pkarr.go     |  4 ++--
 3 files changed, 18 insertions(+), 3 deletions(-)

diff --git a/impl/pkg/server/pkarr.go b/impl/pkg/server/pkarr.go
index a3ea8323..0b932985 100644
--- a/impl/pkg/server/pkarr.go
+++ b/impl/pkg/server/pkarr.go
@@ -59,7 +59,7 @@ func (r *PkarrRouter) GetRecord(c *gin.Context) {
 	resp, err := r.service.GetPkarr(c, *id)
 	if err != nil {
 		// TODO(gabe): provide a more maintainable way to handle custom errors
-		if strings.Contains("spam", err.Error()) {
+		if strings.Contains(err.Error(), "spam") {
 			LoggingRespondErrMsg(c, fmt.Sprintf("too many requests for bad key %s", *id), http.StatusTooManyRequests)
 			return
 		}
diff --git a/impl/pkg/server/pkarr_test.go b/impl/pkg/server/pkarr_test.go
index f87bd5c6..fe43655c 100644
--- a/impl/pkg/server/pkarr_test.go
+++ b/impl/pkg/server/pkarr_test.go
@@ -146,6 +146,21 @@ func TestPkarrRouter(t *testing.T) {
 		pkarrRouter.GetRecord(c)
 		assert.Equal(t, http.StatusNotFound, w.Result().StatusCode, "unexpected %s", w.Result().Status)
 	})
+
+	t.Run("test get not found spam", func(t *testing.T) {
+		w := httptest.NewRecorder()
+		suffix := "cz13drbfxy3ih6xun4mw3cyiexrtfcs9gyp46o4469e93y36zhsy"
+		req := httptest.NewRequest(http.MethodGet, fmt.Sprintf("%s/%s", testServerURL, suffix), nil)
+		c := newRequestContextWithParams(w, req, map[string]string{IDParam: suffix})
+		pkarrRouter.GetRecord(c)
+		assert.Equal(t, http.StatusNotFound, w.Result().StatusCode, "unexpected %s", w.Result().Status)
+
+		w = httptest.NewRecorder()
+		req = httptest.NewRequest(http.MethodGet, fmt.Sprintf("%s/%s", testServerURL, suffix), nil)
+		c = newRequestContextWithParams(w, req, map[string]string{IDParam: suffix})
+		pkarrRouter.GetRecord(c)
+		assert.Equal(t, http.StatusTooManyRequests, w.Result().StatusCode, "unexpected %s", w.Result().Status)
+	})
 }
 
 func testPkarrService(t *testing.T) service.PkarrService {
diff --git a/impl/pkg/service/pkarr.go b/impl/pkg/service/pkarr.go
index 30c9368a..1d677d21 100644
--- a/impl/pkg/service/pkarr.go
+++ b/impl/pkg/service/pkarr.go
@@ -53,8 +53,8 @@ func NewPkarrService(cfg *config.Config, db storage.Storage, d *dht.DHT) (*Pkarr
 	}
 
 	// create a new cache for bad gets to prevent spamming the DHT
-	cacheConfig.LifeWindow = 120 * time.Second
-	cacheConfig.CleanWindow = 60 * time.Second
+	cacheConfig.LifeWindow = 60 * time.Second
+	cacheConfig.CleanWindow = 30 * time.Second
 	badGetCache, err := bigcache.New(context.Background(), cacheConfig)
 	if err != nil {
 		return nil, ssiutil.LoggingErrorMsg(err, "failed to instantiate badGetCache")