Skip to content

Commit 2014a7e

Browse files
committed
(PDB-5216) Test stream-query-result query id logging too
It's the other query path.
1 parent 7d89c9a commit 2014a7e

File tree

1 file changed

+15
-1
lines changed

1 file changed

+15
-1
lines changed

test/puppetlabs/puppetdb/http/query_logging_test.clj

Lines changed: 15 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,13 @@
11
(ns puppetlabs.puppetdb.http.query-logging-test
22
(:require
3+
[clojure.java.io :as io]
34
[clojure.string :as str]
45
[clojure.test :refer :all]
6+
[clojure.walk :refer [keywordize-keys]]
57
[puppetlabs.kitchensink.core :as kitchensink]
8+
[puppetlabs.puppetdb.cheshire :as json]
69
[puppetlabs.puppetdb.cli.services :as svcs]
10+
[puppetlabs.puppetdb.testutils.catalogs :refer [replace-catalog]]
711
[puppetlabs.puppetdb.testutils.db :refer [with-test-db *db*]]
812
[puppetlabs.puppetdb.testutils.http
913
:refer [call-with-http-app query-response with-http-app*]]
@@ -41,17 +45,27 @@
4145
(defn prep-logs [logs]
4246
(->> @logs (map :message) keep-only-pdbquery-logs))
4347

48+
(def catalog-1
49+
(-> "puppetlabs/puppetdb/cli/export/tiny-catalog.json"
50+
io/resource slurp json/parse-string keywordize-keys))
51+
4452
(deftest queries-are-logged-when-log-queries-is-true
4553
(tk-log/with-log-level "puppetlabs.puppetdb.query-eng" :debug
4654
(with-test-db
55+
(replace-catalog catalog-1)
4756
(with-http-app* #(assoc % :log-queries true)
4857
(doseq [[query ast-exp sql-exp]
58+
;; produce-streaming-body
4959
[[["/v4" ["from" "nodes"]]
5060
"\"from\" \"nodes\""
5161
"latest_report_noop_pending"]
5262
[["/v4" ["from" "facts"]]
5363
"\"from\" \"facts\""
54-
"(jsonb_each((stable||volatile)))"]]]
64+
"(jsonb_each((stable||volatile)))"]
65+
;; stream-query-result
66+
[["/v4/catalogs/myhost.localdomain" []]
67+
"\"from\" \"catalogs\""
68+
"row_to_json(edge_data)"]]]
5569
(with-logged-event-maps events
5670
(is (= 200 (:status (apply query-response :get query))))
5771

0 commit comments

Comments
 (0)