test: update response key for whitelist entries and add validation test for missing fields

This commit is contained in:
GitHub Actions
2026-04-16 01:15:53 +00:00
parent 973efd6412
commit f0fdf9b752
2 changed files with 31 additions and 2 deletions

View File

@@ -54,7 +54,7 @@ func TestListWhitelists_Empty(t *testing.T) {
assert.Equal(t, http.StatusOK, w.Code) assert.Equal(t, http.StatusOK, w.Code)
var resp map[string]interface{} var resp map[string]interface{}
require.NoError(t, json.Unmarshal(w.Body.Bytes(), &resp)) require.NoError(t, json.Unmarshal(w.Body.Bytes(), &resp))
entries, ok := resp["entries"].([]interface{}) entries, ok := resp["whitelist"].([]interface{})
assert.True(t, ok) assert.True(t, ok)
assert.Empty(t, entries) assert.Empty(t, entries)
} }
@@ -154,6 +154,22 @@ func TestListWhitelists_AfterAdd(t *testing.T) {
assert.Equal(t, http.StatusOK, w.Code) assert.Equal(t, http.StatusOK, w.Code)
var resp map[string]interface{} var resp map[string]interface{}
require.NoError(t, json.Unmarshal(w.Body.Bytes(), &resp)) require.NoError(t, json.Unmarshal(w.Body.Bytes(), &resp))
entries := resp["entries"].([]interface{}) entries := resp["whitelist"].([]interface{})
assert.Len(t, entries, 1) assert.Len(t, entries, 1)
} }
func TestAddWhitelist_400_MissingField(t *testing.T) {
t.Parallel()
_, r, _ := setupWhitelistHandler(t)
body := `{}`
w := httptest.NewRecorder()
req := httptest.NewRequest(http.MethodPost, "/api/v1/admin/crowdsec/whitelist", bytes.NewBufferString(body))
req.Header.Set("Content-Type", "application/json")
r.ServeHTTP(w, req)
assert.Equal(t, http.StatusBadRequest, w.Code)
var resp map[string]interface{}
require.NoError(t, json.Unmarshal(w.Body.Bytes(), &resp))
assert.Equal(t, "ip_or_cidr is required", resp["error"])
}

View File

@@ -164,3 +164,16 @@ func TestCrowdSecWhitelistService_List_AfterAdd(t *testing.T) {
require.NoError(t, err) require.NoError(t, err)
assert.Len(t, entries, 3) assert.Len(t, entries, 3)
} }
func TestAdd_ValidIPv6_Success(t *testing.T) {
t.Parallel()
svc := services.NewCrowdSecWhitelistService(openWhitelistTestDB(t), "")
entry, err := svc.Add(context.Background(), "2001:db8::1", "ipv6 test")
require.NoError(t, err)
assert.Equal(t, "2001:db8::1", entry.IPOrCIDR)
entries, err := svc.List(context.Background())
require.NoError(t, err)
assert.Len(t, entries, 1)
assert.Equal(t, "2001:db8::1", entries[0].IPOrCIDR)
}