From 0fc1944ad4eabacdedec19d22c9516072a5c699d Mon Sep 17 00:00:00 2001
From: yumiguan <41277086+yumiguan@users.noreply.github.com>
Date: Thu, 25 May 2023 21:33:42 +0800
Subject: [PATCH] feature/core-keep-origin-request-data (#757)

* Core support keep origin request data; Inspector add keep_origin_data switch

* update version

* Update code

* remove unused func

* recover config

* add case
---
 e2e_tests/test_core.py                        |   7 +
 e2e_tests/test_flow_editor.py                 |   9 +
 frontend/src/api/inspector.js                 |  14 --
 frontend/src/store/inspector.js               |   8 +
 frontend/src/store/settings.js                |  20 ++
 frontend/src/views/inspector/ButtonBar.vue    | 114 ++++++-----
 lyrebird/mock/handlers/handler_context.py     |   7 +
 .../handlers/http_data_helper/__init__.py     |  25 +++
 .../content_encoding/__init__.py              |  14 +-
 .../content_encoding/default.py               |   4 +
 .../http_data_helper/content_encoding/gzip.py |   5 +
 .../http_data_helper/content_type/__init__.py |  13 +-
 .../http_data_helper/content_type/default.py  |   5 +
 .../http_data_helper/content_type/form.py     |   6 +-
 .../content_type/javascript.py                |   4 +
 .../http_data_helper/content_type/json.py     |   4 +
 .../http_data_helper/content_type/text.py     |   4 +
 lyrebird/version.py                           |   2 +-
 tests/test_mock_server.py                     | 181 +++++++++++++++++-
 19 files changed, 383 insertions(+), 63 deletions(-)

diff --git a/e2e_tests/test_core.py b/e2e_tests/test_core.py
index cf8805360..5783afac6 100644
--- a/e2e_tests/test_core.py
+++ b/e2e_tests/test_core.py
@@ -32,6 +32,13 @@ def test_json(lyrebird, mock_server):
     assert r.text == hashlib.md5(mock_server.api_post.encode() + data.encode()).hexdigest()
 
 
+def test_javascript(lyrebird, mock_server):
+    data = "(function(){\n    var name = 'Lyrebird';\n  })();"
+    headers = {"Content-Type": "application/javascript"}
+    r = requests.post(url=lyrebird.uri_mock + mock_server.api_post, data=data, headers=headers)
+    assert r.text == hashlib.md5(mock_server.api_post.encode() + data.encode()).hexdigest()
+
+
 def test_text(lyrebird, mock_server):
     data = "asdasdasd"
     headers = {"Content-Type": "text/plain"}
diff --git a/e2e_tests/test_flow_editor.py b/e2e_tests/test_flow_editor.py
index af7875735..7e4159ece 100644
--- a/e2e_tests/test_flow_editor.py
+++ b/e2e_tests/test_flow_editor.py
@@ -38,6 +38,15 @@ def test_flow_editor_json(lyrebird_with_args, mock_server):
     assert r.text == hashlib.md5(mock_server.api_post.encode() + data.encode()).hexdigest()
 
 
+def test_flow_editor_js(lyrebird_with_args, mock_server):
+    lyrebird_with_args.start(checker_path=flow_editor_path)
+
+    data = "console.log('hello world')"
+    headers = {"Content-Type": "application/javascript"}
+    r = requests.post(url=lyrebird_with_args.uri_mock + mock_server.api_post, data=data, headers=headers)
+    assert r.text == hashlib.md5(mock_server.api_post.encode() + data.encode()).hexdigest()
+
+
 def test_flow_editor_text(lyrebird_with_args, mock_server):
     lyrebird_with_args.start(checker_path=flow_editor_path)
     data = "asdasdasd"
diff --git a/frontend/src/api/inspector.js b/frontend/src/api/inspector.js
index 8045a7aaa..52f8644c3 100644
--- a/frontend/src/api/inspector.js
+++ b/frontend/src/api/inspector.js
@@ -1,19 +1,5 @@
 import axios from 'axios'
 
-export const getDiffModeStatus = () => {
-  return axios({
-    url: '/api/diffmode'
-  })
-}
-
-export const setDiffModeStatus = (status) => {
-  return axios({
-    url: '/api/diffmode',
-    data: { status },
-    method: 'PUT'
-  })
-}
-
 export const getFlowFilters = () => {
   return axios({
     url: '/api/flowfilter'
diff --git a/frontend/src/store/inspector.js b/frontend/src/store/inspector.js
index 75f8c8e74..b514dd630 100644
--- a/frontend/src/store/inspector.js
+++ b/frontend/src/store/inspector.js
@@ -11,6 +11,8 @@ export default {
     focusedFlowDetail: null,
     originFlowList: [],
     recordMode: '',
+    diffMode: 'normal',
+    isRequestKeepOriginData: false,
     flowFilters: [],
     selectedFlowFilter: {}
   },
@@ -54,6 +56,12 @@ export default {
     setRecordMode (state, recordMode) {
       state.recordMode = recordMode
     },
+    setDiffMode (state, diffMode) {
+      state.diffMode = diffMode
+    },
+    setIsRequestKeepOriginData (state, isRequestKeepOriginData) {
+      state.isRequestKeepOriginData = isRequestKeepOriginData
+    },
     setFlowFilters (state, flowFilters) {
       state.flowFilters = flowFilters
     },
diff --git a/frontend/src/store/settings.js b/frontend/src/store/settings.js
index 2d6ed6276..e2b42f3e4 100644
--- a/frontend/src/store/settings.js
+++ b/frontend/src/store/settings.js
@@ -2,8 +2,10 @@ import * as api from '@/api'
 import { bus } from '@/eventbus'
 
 var configCommitMap = [
+  {'name': 'mock.mode', 'commit': 'setDiffMode'},
   {'name': 'mock.data.showLabel', 'commit': 'setIsLabelDisplay'},
   {'name': 'mock.data.tree.closeReload', 'commit': 'setIsCloseReloadWhenEnter'},
+  {'name': 'mock.request.keep_origin_data', 'commit': 'setIsRequestKeepOriginData'},
   {'name': 'mock.data.tree.undeletableId', 'commit': 'concatTreeUndeletableId'},
   {'name': 'mock.data.detail.stickyTopKey', 'commit': 'concatStickyTopKey'},
   {'name': 'mock.data.detail.undeletableKey', 'commit': 'concatUndeletableKey'},
@@ -73,6 +75,24 @@ export default {
         .catch(error => {
           bus.$emit('msg.error', `Update config failed ${error.data.message}`)
         })
+    },
+    commitAndupdateConfigByKey({ dispatch, commit }, { command, val }) {
+      let updateConfig = {}
+      for (const config of configCommitMap) {
+        if (config.commit == command) {
+          updateConfig[config.name] = val
+          commit(command, val)
+          break
+        }
+      }
+      api.updateConfigByKey(updateConfig)
+        .then(_ => {
+          dispatch('loadConfig')
+          bus.$emit('msg.success', `Update config success!`)
+        })
+        .catch(error => {
+          bus.$emit('msg.error', `Update config failed ${error.data.message}`)
+        })
     }
   }
 }
diff --git a/frontend/src/views/inspector/ButtonBar.vue b/frontend/src/views/inspector/ButtonBar.vue
index 813c044dc..63a60c656 100644
--- a/frontend/src/views/inspector/ButtonBar.vue
+++ b/frontend/src/views/inspector/ButtonBar.vue
@@ -29,30 +29,8 @@
         <span>Clear</span>
       </v-tooltip>
 
-      <v-divider vertical class="button-bar-divider border"/>
-      <b style="padding-right:5px">Diff Mode</b>
-      <v-switch
-        small
-        dense
-        inset
-        v-model="diffMode"
-        color="primary"
-        class="button-bar-diff-mode"
-        @change="changeDiffMode"
-      />
 
-      <v-tooltip bottom>
-        <template v-slot:activator="{ on, attrs }">
-          <v-btn plain icon small v-bind="attrs" v-on="on">
-            <v-icon small size="18px" color="content">mdi-help-circle-outline</v-icon>
-          </v-btn>
-        </template>
-        <span>Get the proxy response while the request is mocked</span>
-      </v-tooltip>
-
-      <v-divider vertical class="button-bar-divider border"/>
-
-      <b style="padding-right:5px">Mock Group</b>
+      <b class="pr-1 pl-3">Mock Group</b>
 
       <v-chip
         label small outlined
@@ -124,6 +102,54 @@
           @click:clear="clearInspectorSearch"
         />
       </div>
+
+
+      <v-menu
+        left
+        bottom
+        offset-y
+        offset-overflow
+        :close-on-content-click="false"
+        style="position: absolute;"
+      >
+
+        <template v-slot:activator="{ on, attrs }">
+          <v-btn
+            icon
+            v-bind="attrs"
+            v-on="on"
+            class="ml-1"
+            title="Settings"
+          >
+            <v-icon size="18px" color="content">mdi-cog-outline</v-icon>
+          </v-btn>
+        </template>
+
+        <v-list dense>
+          <v-list-item>
+            <v-list-item-action>
+              <v-switch v-model="diffMode"/>
+            </v-list-item-action>
+
+            <v-list-item-content>
+              <v-list-item-title>Diff Mode</v-list-item-title>
+              <v-list-item-subtitle>Get the proxy response while the request is mocked</v-list-item-subtitle>
+            </v-list-item-content>
+          </v-list-item>
+
+          <v-list-item>
+            <v-list-item-action>
+              <v-switch v-model="isRequestKeepOriginData"/>
+            </v-list-item-action>
+
+            <v-list-item-content>
+              <v-list-item-title>Origin request body</v-list-item-title>
+              <v-list-item-subtitle>Keep origin request body</v-list-item-subtitle>
+            </v-list-item-content>
+          </v-list-item>
+        </v-list>
+
+      </v-menu>
     </div>
 
     <MockDataSelector ref="searchModal" :showRoot="false">
@@ -158,7 +184,6 @@
 <script>
 import MockDataSelector from '@/components/SearchModal.vue'
 import Icon from 'vue-svg-icon/Icon.vue'
-import { getDiffModeStatus, setDiffModeStatus } from '@/api'
 
 export default {
   name: 'buttonBar',
@@ -168,15 +193,31 @@ export default {
   },
   data () {
     return {
-      diffMode: false
     }
   },
-  mounted () {
-    this.loadDiffModeStatus()
-  },
   computed: {
-    isRecordMode () {
-      return this.$store.state.inspector.recordMode === 'record'
+    diffMode: {
+      get () {
+        return this.$store.state.inspector.diffMode === 'multiple'
+      },
+      set (val) {
+        const mode = val ? 'multiple' : 'normal'
+        this.$store.commit('setDiffMode', mode)
+        this.$store.dispatch('updateConfigByKey', {
+          'mock.mode': mode
+        })
+      }
+    },
+    isRequestKeepOriginData: {
+      get () {
+        return this.$store.state.inspector.isRequestKeepOriginData
+      },
+      set (val) {
+        this.$store.dispatch('commitAndupdateConfigByKey', {
+          'command': 'setIsRequestKeepOriginData',
+          val
+        })
+      }
     },
     isEmptySelectedFlow () {
       return this.$store.state.inspector.selectedFlows.length === 0
@@ -225,19 +266,6 @@ export default {
     showMockDataSelector () {
       this.$refs.searchModal.toggal()
     },
-    loadDiffModeStatus () {
-      getDiffModeStatus()
-        .then(response => {
-          this.diffMode = response.data.diffmode === 'multiple'
-        })
-        .catch(error => {
-          this.$bus.$emit('msg.error', 'Load diff mode status failed: ' + error.data.message)
-        })
-    },
-    changeDiffMode (payload) {
-      const mode = payload ? 'multiple' : 'normal'
-      setDiffModeStatus(mode)
-    },
     changeFlowFilter () {
       this.$store.dispatch('loadFlowList')
     },
diff --git a/lyrebird/mock/handlers/handler_context.py b/lyrebird/mock/handlers/handler_context.py
index 53ca8610f..2573dabeb 100644
--- a/lyrebird/mock/handlers/handler_context.py
+++ b/lyrebird/mock/handlers/handler_context.py
@@ -98,6 +98,13 @@ def _parse_request(self):
         self.flow['client_address'] = self.client_address
 
         self.flow['request'] = _request
+
+        if self.request.method in ['POST', 'PUT'] and application.config.get('mock.request.keep_origin_data'):
+            origin_data = DataHelper.origin2string(self.request)
+            self.flow['origin_request'] = {
+                'data': origin_data
+            }
+
         context.application.cache.add(self.flow)
 
         logger.debug(f'[On client request] {self.flow["request"]["url"]}')
diff --git a/lyrebird/mock/handlers/http_data_helper/__init__.py b/lyrebird/mock/handlers/http_data_helper/__init__.py
index 7bdccc703..570fd8e3d 100644
--- a/lyrebird/mock/handlers/http_data_helper/__init__.py
+++ b/lyrebird/mock/handlers/http_data_helper/__init__.py
@@ -59,3 +59,28 @@ def flow2origin(flow_obj, output=None, chain=None):
             output.data = _data
         else:
             return _data
+
+    @staticmethod
+    def origin2string(origin_obj, output=None):
+        if not origin_obj:
+            return
+
+        _data = origin_obj.data
+        if not _data:
+            return
+
+        # Read raw headers, support the request from extra mock 9999 port
+        if 'Proxy-Raw-Headers' in origin_obj.headers:
+            _origin_headers = json.loads(origin_obj.headers['Proxy-Raw-Headers'])
+            raw_headers = CaseInsensitiveDict(_origin_headers)
+        else:
+            raw_headers = origin_obj.headers
+
+        for headers_key, func in origin2flow_handlers.items():
+            headers_val = raw_headers.get(headers_key, '')
+            _data = func.origin2string(headers_val, _data)
+
+        if output:
+            output['data'] = _data
+        else:
+            return _data
diff --git a/lyrebird/mock/handlers/http_data_helper/content_encoding/__init__.py b/lyrebird/mock/handlers/http_data_helper/content_encoding/__init__.py
index d9ce902b3..df66bb191 100644
--- a/lyrebird/mock/handlers/http_data_helper/content_encoding/__init__.py
+++ b/lyrebird/mock/handlers/http_data_helper/content_encoding/__init__.py
@@ -22,7 +22,8 @@ def origin2flow(content_encoding, request_data, chain=None):
         _data = func.origin2flow(request_data)
         logger.warning(f'Convert Content-Encoding: {content_encoding} data origin2flow failed! {e}')
     finally:
-        chain.append(func)
+        if chain and isinstance(chain, list):
+            chain.append(func)
 
     return _data
 
@@ -34,3 +35,14 @@ def flow2origin(content_encoding, flow_data):
         _data = DefaultHandler.flow2origin(flow_data)
         logger.warning(f'Convert Content-Encoding: {content_encoding} data flow2origin failed! {e}')
     return _data
+
+def origin2string(content_encoding, request_data):
+    func = _get_matched_action(content_encoding)
+    try:
+        _data = func.origin2string(request_data)
+    except Exception as e:
+        func = DefaultHandler
+        _data = func.origin2string(request_data)
+        logger.warning(f'Convert Content-Encoding: {content_encoding} data origin2flow failed! {e}')
+
+    return _data
diff --git a/lyrebird/mock/handlers/http_data_helper/content_encoding/default.py b/lyrebird/mock/handlers/http_data_helper/content_encoding/default.py
index 527e2acfc..2dc43d751 100644
--- a/lyrebird/mock/handlers/http_data_helper/content_encoding/default.py
+++ b/lyrebird/mock/handlers/http_data_helper/content_encoding/default.py
@@ -7,3 +7,7 @@ def origin2flow(request_data):
     @staticmethod
     def flow2origin(flow_data):
         return flow_data
+
+    @staticmethod
+    def origin2string(request_data):
+        return request_data
diff --git a/lyrebird/mock/handlers/http_data_helper/content_encoding/gzip.py b/lyrebird/mock/handlers/http_data_helper/content_encoding/gzip.py
index 4e4e904ac..dfcf24cc0 100644
--- a/lyrebird/mock/handlers/http_data_helper/content_encoding/gzip.py
+++ b/lyrebird/mock/handlers/http_data_helper/content_encoding/gzip.py
@@ -12,3 +12,8 @@ def origin2flow(request_data):
     def flow2origin(flow_data):
         _data = gzip.compress(flow_data)
         return _data
+
+    @staticmethod
+    def origin2string(request_data):
+        _data = gzip.decompress(request_data)
+        return _data
diff --git a/lyrebird/mock/handlers/http_data_helper/content_type/__init__.py b/lyrebird/mock/handlers/http_data_helper/content_type/__init__.py
index cd2e4b71d..6b8e4d58d 100644
--- a/lyrebird/mock/handlers/http_data_helper/content_type/__init__.py
+++ b/lyrebird/mock/handlers/http_data_helper/content_type/__init__.py
@@ -29,7 +29,8 @@ def origin2flow(content_type, request_data, chain=None):
         _data = func.origin2flow(request_data)
         logger.warning(f'Convert Content-Type: {content_type} data origin2flow failed! {e}')
     finally:
-        chain.append(func)
+        if chain and isinstance(chain, list):
+            chain.append(func)
 
     return _data
 
@@ -41,3 +42,13 @@ def flow2origin(content_type, flow_data):
         _data = DefaultHandler.flow2origin(flow_data)
         logger.warning(f'Convert Content-Type: {content_type} data flow2origin failed! {e}')
     return _data
+
+def origin2string(content_type, request_data):
+    func = _get_matched_action(content_type)
+    try:
+        _data = func.origin2string(request_data)
+    except Exception as e:
+        func = DefaultHandler
+        _data = func.origin2string(request_data)
+        logger.warning(f'Convert Content-Type: {content_type} data origin2flow failed! {e}')
+    return _data
diff --git a/lyrebird/mock/handlers/http_data_helper/content_type/default.py b/lyrebird/mock/handlers/http_data_helper/content_type/default.py
index 8b61a0471..2f82c7a84 100644
--- a/lyrebird/mock/handlers/http_data_helper/content_type/default.py
+++ b/lyrebird/mock/handlers/http_data_helper/content_type/default.py
@@ -12,3 +12,8 @@ def origin2flow(request_data):
     def flow2origin(flow_data):
         _data = binascii.a2b_base64(flow_data)
         return _data
+
+    @staticmethod
+    def origin2string(request_data):
+        _data = binascii.b2a_base64(request_data).decode('utf-8')
+        return _data
diff --git a/lyrebird/mock/handlers/http_data_helper/content_type/form.py b/lyrebird/mock/handlers/http_data_helper/content_type/form.py
index d85a1e795..dcbe8d944 100644
--- a/lyrebird/mock/handlers/http_data_helper/content_type/form.py
+++ b/lyrebird/mock/handlers/http_data_helper/content_type/form.py
@@ -5,7 +5,7 @@ class FormHandler:
 
     @staticmethod
     def origin2flow(request_data):
-        params = parse_qs(request_data.decode('utf-8'))
+        params = parse_qs(request_data.decode('utf-8'), keep_blank_values=True)
         _data = {k:v[0] for k,v in params.items()}
         return _data
 
@@ -13,3 +13,7 @@ def origin2flow(request_data):
     def flow2origin(flow_data):
         _data = urlencode(flow_data).encode()
         return _data
+
+    @staticmethod
+    def origin2string(request_data):
+        return request_data.decode('utf-8')
diff --git a/lyrebird/mock/handlers/http_data_helper/content_type/javascript.py b/lyrebird/mock/handlers/http_data_helper/content_type/javascript.py
index c3a9cd80f..8d2544fb0 100644
--- a/lyrebird/mock/handlers/http_data_helper/content_type/javascript.py
+++ b/lyrebird/mock/handlers/http_data_helper/content_type/javascript.py
@@ -9,3 +9,7 @@ def origin2flow(request_data):
     def flow2origin(flow_data):
         _data = flow_data.encode()
         return _data
+
+    @staticmethod
+    def origin2string(request_data):
+        return request_data.decode('utf-8')
diff --git a/lyrebird/mock/handlers/http_data_helper/content_type/json.py b/lyrebird/mock/handlers/http_data_helper/content_type/json.py
index 7cf046914..6f1c8da5e 100644
--- a/lyrebird/mock/handlers/http_data_helper/content_type/json.py
+++ b/lyrebird/mock/handlers/http_data_helper/content_type/json.py
@@ -12,3 +12,7 @@ def origin2flow(request_data):
     def flow2origin(flow_data):
         _data = json.dumps(flow_data, ensure_ascii=False).encode()
         return _data
+
+    @staticmethod
+    def origin2string(request_data):
+        return request_data.decode('utf-8')
diff --git a/lyrebird/mock/handlers/http_data_helper/content_type/text.py b/lyrebird/mock/handlers/http_data_helper/content_type/text.py
index 0bff92df9..8aaf6e61b 100644
--- a/lyrebird/mock/handlers/http_data_helper/content_type/text.py
+++ b/lyrebird/mock/handlers/http_data_helper/content_type/text.py
@@ -9,3 +9,7 @@ def origin2flow(request_data):
     def flow2origin(flow_data):
         _data = flow_data.encode()
         return _data
+
+    @staticmethod
+    def origin2string(request_data):
+        return request_data.decode('utf-8')
diff --git a/lyrebird/version.py b/lyrebird/version.py
index 345fa6016..37613129e 100644
--- a/lyrebird/version.py
+++ b/lyrebird/version.py
@@ -1,3 +1,3 @@
-IVERSION = (2, 13, 0)
+IVERSION = (2, 14, 0)
 VERSION = ".".join(str(i) for i in IVERSION)
 LYREBIRD = "Lyrebird " + VERSION
diff --git a/tests/test_mock_server.py b/tests/test_mock_server.py
index e5120f9f4..d5047a52e 100644
--- a/tests/test_mock_server.py
+++ b/tests/test_mock_server.py
@@ -5,9 +5,12 @@
 from lyrebird.event import EventServer
 from lyrebird import application
 from lyrebird import reporter
-from typing import NamedTuple
+
+import json
 import pytest
-from urllib.parse import quote
+import binascii
+from typing import NamedTuple
+from urllib.parse import quote, urlencode
 
 
 conf = {
@@ -45,6 +48,11 @@ def clear():
     context.application.cache._cache.clear()
 
 
+@pytest.fixture
+def keep_origin_data():
+    application.config['mock.request.keep_origin_data'] = True
+
+
 def test_mock_api(client):
     resp = client.get('/mock/http://www.bing.com')
     assert 200 <= resp.status_code <= 400
@@ -63,3 +71,172 @@ def test_mock_api_with_query(client, clear):
     assert len(cache_list._cache) == 1
     assert cache_list._cache[0]['request']['query']['url'] == url
     assert cache_list._cache[0]['request']['url'] == f'?url={url}'
+
+
+def test_mock_content_type_json(client, clear):
+    url = '/mock/http://www.bing.com'
+    origin_json = {
+        'name': 'Lyrebird'
+    }
+    origin_body = json.dumps(origin_json)
+    headers = {'Content-Type': 'application/json'}
+
+    client.post(url, headers=headers, data=origin_body)
+
+    cache_list = context.application.cache
+    assert len(cache_list._cache) == 1
+
+    flow = cache_list._cache[0]
+    flow_body = flow['request']['data']
+
+    assert origin_json == flow_body
+
+
+def test_mock_content_type_javascript(client, clear):
+    url = '/mock/http://www.bing.com'
+    origin_js = 'console.log("Hello, world!");'
+    headers = {'Content-Type': 'application/javascript'}
+
+    client.post(url, headers=headers, data=origin_js)
+
+    cache_list = context.application.cache
+    assert len(cache_list._cache) == 1
+
+    flow = cache_list._cache[0]
+    flow_body = flow['request']['data']
+
+    assert origin_js == flow_body
+
+
+def test_mock_content_type_text(client, clear):
+    url = '/mock/http://www.bing.com'
+    origin_text = 'Hello, world!'
+    headers = {'Content-Type': 'text/plain'}
+
+    client.post(url, headers=headers, data=origin_text)
+
+    cache_list = context.application.cache
+    assert len(cache_list._cache) == 1
+
+    flow = cache_list._cache[0]
+    flow_body = flow['request']['data']
+
+    assert origin_text == flow_body
+
+
+def test_mock_content_type_form(client, clear):
+    url = '/mock/http://www.bing.com'
+    origin_data = 'name=Lyrebird&com=mt&blank='
+    headers = {'Content-Type': 'application/x-www-form-urlencoded'}
+
+    client.post(url, headers=headers, data=origin_data)
+
+    cache_list = context.application.cache
+    assert len(cache_list._cache) == 1
+
+    flow = cache_list._cache[0]
+    flow_body = flow['request']['data']
+
+    assert origin_data == urlencode(flow_body)
+
+
+def test_mock_content_type_default(client, clear):
+    url = '/mock/http://www.bing.com'
+    origin_data = b'Lyrebird'
+    headers = {'Content-Type': 'image/png'}
+
+    client.post(url, headers=headers, data=origin_data)
+
+    cache_list = context.application.cache
+    assert len(cache_list._cache) == 1
+
+    flow = cache_list._cache[0]
+    flow_body = flow['request']['data']
+
+    flow_body_a2b = binascii.a2b_base64(flow_body.encode())
+
+    assert origin_data == flow_body_a2b
+
+
+def test_mock_content_type_json_keep_origin_request(client, clear, keep_origin_data):
+    url = '/mock/http://www.bing.com'
+    origin_json = {
+        'name': 'Lyrebird'
+    }
+    origin_body = json.dumps(origin_json)
+    headers = {'Content-Type': 'application/json'}
+
+    client.post(url, headers=headers, data=origin_body)
+
+    cache_list = context.application.cache
+    assert len(cache_list._cache) == 1
+
+    flow = cache_list._cache[0]
+    flow_body = flow['origin_request']['data']
+
+    assert origin_body == flow_body
+
+
+def test_mock_content_type_javascript_keep_origin_request(client, clear, keep_origin_data):
+    url = '/mock/http://www.bing.com'
+    origin_js = 'console.log("Hello, world!");'
+    headers = {'Content-Type': 'application/javascript'}
+
+    client.post(url, headers=headers, data=origin_js)
+
+    cache_list = context.application.cache
+    assert len(cache_list._cache) == 1
+
+    flow = cache_list._cache[0]
+    flow_body = flow['origin_request']['data']
+
+    assert origin_js == flow_body
+
+
+def test_mock_content_type_text_keep_origin_request(client, clear, keep_origin_data):
+    url = '/mock/http://www.bing.com'
+    origin_text = 'Hello, world!'
+    headers = {'Content-Type': 'text/plain'}
+
+    client.post(url, headers=headers, data=origin_text)
+
+    cache_list = context.application.cache
+    assert len(cache_list._cache) == 1
+
+    flow = cache_list._cache[0]
+    flow_body = flow['origin_request']['data']
+
+    assert origin_text == flow_body
+
+
+def test_mock_content_type_form_keep_origin_request(client, clear, keep_origin_data):
+    url = '/mock/http://www.bing.com'
+    origin_data = 'name=Lyrebird&com=mt&blank'
+    headers = {'Content-Type': 'application/x-www-form-urlencoded'}
+
+    client.post(url, headers=headers, data=origin_data)
+
+    cache_list = context.application.cache
+    assert len(cache_list._cache) == 1
+
+    flow = cache_list._cache[0]
+    flow_body = flow['origin_request']['data']
+
+    assert origin_data == flow_body
+
+def test_mock_content_type_default_keep_origin_request(client, clear, keep_origin_data):
+    url = '/mock/http://www.bing.com'
+    origin_data = b'Lyrebird'
+    headers = {'Content-Type': 'image/png'}
+
+    client.post(url, headers=headers, data=origin_data)
+
+    cache_list = context.application.cache
+    assert len(cache_list._cache) == 1
+
+    flow = cache_list._cache[0]
+    flow_body = flow['origin_request']['data']
+
+    flow_body_a2b = binascii.a2b_base64(flow_body.encode())
+
+    assert origin_data == flow_body_a2b