From 42eded903af1a2af71700767952e6b6ce6e048a1 Mon Sep 17 00:00:00 2001
From: Ilya Khaprov <dead.trickster@gmail.com>
Date: Fri, 15 Apr 2016 11:04:00 -0700
Subject: [PATCH] tests for pushgateway-error

---
 t/pushgateway/pushgateway.lisp | 21 +++++++++++++++++++--
 t/pushgateway/test-server.lisp | 25 ++++++++++++++++++-------
 2 files changed, 37 insertions(+), 9 deletions(-)

diff --git a/t/pushgateway/pushgateway.lisp b/t/pushgateway/pushgateway.lisp
index eff49d2..70b6d49 100644
--- a/t/pushgateway/pushgateway.lisp
+++ b/t/pushgateway/pushgateway.lisp
@@ -34,6 +34,16 @@
       (is (third req) nil)
       (is (fourth req) nil))))
 
+(defun test-error (address)
+  (subtest "ERROR"
+    (handler-case
+        (prom.pushgateway:delete "error" :gateway address)
+      (prom.pushgateway:pushgateway-error (e)
+        (ok "Pushgateway error is throwed")
+        (is (slot-value e 'prom.pushgateway::status-code) 404)
+        (is (slot-value e 'prom.pushgateway::body) "Not Found!")
+        (is-type (slot-value e 'prom.pushgateway::headers) 'list)))))
+
 (subtest "PUSHGATEWAY"
   (subtest "Errors & Validatoins"
     (is-error-report (prom.pushgateway:delete :qwe) prom:invalid-value-error "Label value :QWE is invalid. Reason: job name is not a string")
@@ -44,7 +54,13 @@
     (is-error-report (prom.pushgateway:delete "qwe" :grouping-key '(:qwe "qwe")) prom:invalid-label-name-error "Label name :QWE is invalid. Reason: label name is not a string")
     (is-error-report (prom.pushgateway:delete "qwe" :grouping-key '("q/we" "qwe")) prom:invalid-label-name-error "Label name \"q/we\" is invalid. Reason: label name doesn't match regex [a-zA-Z_][a-zA-Z0-9_]*")
     (is-error-report (prom.pushgateway:delete "qwe" :grouping-key '("qwe" :qwe)) prom:invalid-label-value-error "Label value :QWE is invalid. Reason: label value is not a string")
-    (is-error-report (prom.pushgateway:delete "qwe" :grouping-key '("qwe" "q/we")) prom:invalid-label-value-error "Label value \"q/we\" is invalid. Reason: label value contains / or %2f"))
+    (is-error-report (prom.pushgateway:delete "qwe" :grouping-key '("qwe" "q/we")) prom:invalid-label-value-error "Label value \"q/we\" is invalid. Reason: label value contains / or %2f")
+
+    (subtest "Pushgateway Error"
+      (error-class-exists prom.pushgateway:pushgateway-error)
+
+      (error-report-test prom.pushgateway:pushgateway-error ((:status-code 123 :headers 12 :body 321)
+                                                             "Error talking to pushgateway. Response Code: 123, headers: 12, body: 321"))))
 
   (subtest "API"
     (let ((metrics-acceptor)
@@ -56,7 +72,8 @@
              (sleep 1)
              (test-replace pushgateway-address metrics-acceptor)
              (test-push pushgateway-address metrics-acceptor)
-             (test-delete pushgateway-address metrics-acceptor))
+             (test-delete pushgateway-address metrics-acceptor)
+             (test-error pushgateway-address))
         (when metrics-acceptor
           (tbnl:stop metrics-acceptor :soft t))))))
 
diff --git a/t/pushgateway/test-server.lisp b/t/pushgateway/test-server.lisp
index e1be326..5ef9a9f 100644
--- a/t/pushgateway/test-server.lisp
+++ b/t/pushgateway/test-server.lisp
@@ -10,10 +10,21 @@
   (declare (ignore log-level format-string format-arguments)))
 
 (defmethod tbnl:acceptor-dispatch-request ((acceptor my-acceptor) request)
-  (push (list (tbnl:request-method request)
-              (tbnl:request-uri request)
-              (tbnl:header-in :content-type request)
-              (tbnl:raw-post-data :request request :force-text t))
-        (my-acceptor-requests acceptor))
-  (setf (tbnl:return-code*) tbnl:+http-accepted+)
-  (tbnl:abort-request-handler))
+  (if (search "error" (tbnl:request-uri request))
+      (progn
+        (setf (tbnl:return-code*) tbnl:+http-not-found+)
+        "Error!")
+      (progn
+        (push (list (tbnl:request-method request)
+                    (tbnl:request-uri request)
+                    (tbnl:header-in :content-type request)
+                    (tbnl:raw-post-data :request request :force-text t))
+              (my-acceptor-requests acceptor))
+        (setf (tbnl:return-code*) tbnl:+http-accepted+)
+        (tbnl:abort-request-handler))))
+
+(defmethod tbnl::acceptor-status-message ((acceptor my-acceptor) http-status-code &rest args &key &allow-other-keys)
+  (case http-status-code
+    (404
+     "Not Found!")
+    (t (apply 'tbnl::make-cooked-message http-status-code args))))