Function TestBulkCheckAllowlist

pkg/apiserver/allowlists_test.go:151–190

(t *testing.T)

}

func TestBulkCheckAllowlist(t *testing.T) {
	ctx := t.Context()
	lapi := SetupLAPITest(t, ctx)

	// create an allowlist and add one live entry
	l, err := lapi.DBClient.CreateAllowList(ctx, "test", "test", "", false)
	require.NoError(t, err)

	added, err := lapi.DBClient.AddToAllowlist(ctx, l, []*models.AllowlistItem{
		{Value: "1.2.3.4"},
	})
	require.NoError(t, err)
	assert.Equal(t, 1, added)

	// craft a bulk check payload with one matching and one non-matching target
	reqBody := `{"targets":["1.2.3.4","2.3.4.5"]}`
	w := lapi.RecordResponse(t, ctx, http.MethodPost, "/v1/allowlists/check", strings.NewReader(reqBody), passwordAuthType)
	require.Equal(t, http.StatusOK, w.Code)

	// unmarshal and verify
	resp := models.BulkCheckAllowlistResponse{}
	require.NoError(t, json.Unmarshal(w.Body.Bytes(), &resp))
	require.Len(t, resp.Results, 1)

	// expect only "1.2.3.4" in the "test" allowlist, while "2.3.4.5" should not be in the response
	var match bool

	for _, r := range resp.Results {
		switch *r.Target {
		case "1.2.3.4":
			match = true

			assert.Equal(t, []string{"1.2.3.4 from test"}, r.Allowlists)
		default:
			t.Errorf("unexpected target %v", r.Target)
		}
	}

	require.True(t, match, "did not see result for 1.2.3.4")
}

func TestBulkCheckAllowlist_BadRequest(t *testing.T) {
	ctx := t.Context()