完成mock

bin/mock

@@ -9,3 +9,5 @@ import mocking
 workDir = os.getcwd()
 mock = mocking.Mocking(workDir)
 mock.convert()
+if len(sys.argv) > 1 and sys.argv[1] == 'run':
+    os.system('newman run ' + mock.getOutFileName())

scripts/postman/mocking.py

@@ -33,7 +33,7 @@ class Mocking(object):
         self._out()
 
     def _out(self):
-        wf = open(self._workDir + self._env['name'] + '.postman.json', 'w')
+        wf = open(self.getOutFileName(), 'w')
         wf.write(json.dumps(self._projObj, indent = 4))
         wf.close()
 
@@ -75,6 +75,11 @@ class Mocking(object):
                         'type': 'text/javascript'
                     }
                 })
+            for key, val in self._getArgs(item['url']).items():
+                itemObj['request']['url']['query'].append({
+                    'key': key,
+                    'value': val[0],
+                })
             for param in item['params']:
                 itemObj['request']['url']['query'].append({
                     'key': param[0],
@@ -104,3 +109,11 @@ class Mocking(object):
     def _getPath(self, url):
         parsed_tuple = urlparse.urlparse(url)
         return [path for path in parsed_tuple.path.split('/') if path != '']
+
+    def _getArgs(self, url):
+        parsed_tuple = urlparse.urlparse(url)
+        args = urlparse.parse_qs(parsed_tuple.query)
+        return args
+
+    def getOutFileName(self):
+        return self._workDir + self._env['name'] + '.postman.json'