automatically convert Hash success bodies to JSON

abstractive · Dec 16, 2019 · 4abf267 · 4abf267
1 parent 47f752d
commit 4abf267
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 1 deletion.
diff --git a/CHANGES.md b/CHANGES.md
@@ -1,3 +1,6 @@
+# v201912160309-6Mg3 0.0.1.31
+* Allow passing Hash to Artillery::Shot#success which automatically converts to JSON
+
 # v201912160106-6Mg3 0.0.1.30
 * Revamped Artillery::Shell::Request, especially Payload type and handling
 * Setup proper target for artillery-mountpoint to begin using out of bazooka mode

diff --git a/shard.yml b/shard.yml
@@ -1,5 +1,5 @@
 name: artillery
-version: 0.0.1.30
+version: 0.0.1.31
 
 description: Serverfree object-oriented microservice architecture.
 
@@ -11,6 +11,7 @@ crystal: 0.31.1
 executables:
   - artillery
   - bazooka
+  - artillery-mountpoint
 
 license: MIT
 

diff --git a/src/artillery/shot.cr b/src/artillery/shot.cr
@@ -50,6 +50,21 @@ module Artillery
       @response.body = body unless body.nil?
     end
 
+    alias ResponseHash = Hash(String, Char |
+                                      String |
+                                      Bool |
+                                      Float64 |
+                                      Int64 |
+                                      Int32 |
+                                      Int16 |
+                                      Array(ResponseHash) |
+                                      Hash(String, ResponseHash))
+
+    def success(body : ResponseHash)
+      @response.status = 200
+      @response.body = body.to_json
+    end
+
     def redirect(@redirect)
     end