Fix regex to support query parameters

axiomhq · Jul 14, 2021 · aa3efe3 · aa3efe3
1 parent ab5af69
commit aa3efe3
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/axiom/client.go b/axiom/client.go
@@ -27,7 +27,7 @@ var (
 	ErrUnprivilegedToken = errors.New("using ingest token for non-ingest operation")
 )
 
-var validIngestTokenPathRe = regexp.MustCompile("^/api/v1/(datasets/.+/ingest|tokens/ingest/validate)$")
+var validIngestTokenPathRe = regexp.MustCompile("^/api/v1/(datasets/.+/ingest|tokens/ingest/validate)")
 
 // Error is the generic error response returned on non 2xx HTTP status codes.
 // Either one of the two fields is populated. However, calling the Error()

diff --git a/axiom/client_test.go b/axiom/client_test.go
@@ -277,6 +277,10 @@ func TestIngestPathRegex(t *testing.T) {
 			input: "/api/v1/datasets/test/ingest",
 			match: true,
 		},
+		{
+			input: "/api/v1/datasets/test/ingest?timestamp-field=time",
+			match: true,
+		},
 		{
 			input: "/api/v1/tokens/ingest/validate",
 			match: true,