CoderKungfu#7

Verify mode is a valid option.
garie · May 19, 2017 · a91a99d · a91a99d
1 parent a39ec9c
commit a91a99d
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/lib/mountebank/stub/http_response.rb b/lib/mountebank/stub/http_response.rb
@@ -4,7 +4,7 @@ def self.create(statusCode=200, headers={}, body='', behaviors={}, mode='')
     payload[:statusCode] = statusCode
     payload[:headers] = headers unless headers.empty?
     payload[:body] = body unless body.empty?
-    payload[:_mode] = mode unless mode.empty?
+    payload[:_mode] = mode if (mode == 'text' or mode == 'binary')
 
     data = {is: payload}
     data.merge!(_behaviors: behaviors) unless behaviors.empty?

diff --git a/spec/mountebank/stub/http_response_spec.rb b/spec/mountebank/stub/http_response_spec.rb
@@ -22,6 +22,14 @@
       end
     end
 
+    context 'invalid binary mode' do
+        let(:mode) { 'random' }
+        it 'returns response object without binary mode' do
+          expect(response).to be_a Mountebank::Stub::HttpResponse
+          expect(response.to_json).to eq '{"is":{"statusCode":200,"headers":{"Content-Type":"application/json"},"body":"{\"foo\":\"bar\"}"}}'
+        end
+    end
+
     context 'with behaviors' do
         let(:behaviors) { { wait: 10000 } }
       it 'returns a response object' do