Add error handling for the GM API wrapper

Author: lonerz

gm_api.py

@@ -14,6 +14,10 @@ def post(route, body):
     )
     json = resp.json()
 
-    # deal with error handling later
+    print(json)
+
+    # handle errors
+    if json.get('status') != '200':
+      return {'error': json.get('reason')}
 
     return json

test_gm_api.py

@@ -2,12 +2,22 @@
 import requests
 from gm_api import GM_Api as GM_Api
 
-def test_post(mocker):
+def test_post_success(mocker):
   mock = mocker.patch('requests.post')
-  def json():
-    return 'this is a mock'
-  mock.return_value.json = json
+  def success_json():
+    return {'status': '200', 'data': 'this is a mock'}
+  mock.return_value.json = success_json
 
   resp = GM_Api.post('/getVehicleInfoService', {'id': '1234'})
   mock.assert_called_with('http://gmapi.azurewebsites.net/getVehicleInfoService', json={'id': '1234', 'responseType': 'JSON'})
-  assert(resp == 'this is a mock')
+  assert(resp == {'status': '200', 'data': 'this is a mock'})
+
+def test_post_fail(mocker):
+  mock = mocker.patch('requests.post')
+  def fail_json():
+    return {'status': '404', 'reason': 'Vehicle id: 1000 not found.'}
+  mock.return_value.json = fail_json 
+
+  resp = GM_Api.post('/getVehicleInfoService', {'id': '1000'})
+  mock.assert_called_with('http://gmapi.azurewebsites.net/getVehicleInfoService', json={'id': '1000', 'responseType': 'JSON'})
+  assert(resp == {'error': 'Vehicle id: 1000 not found.'})