Hack to Make This Works with Zapier

StorjOld · Aug 13, 2015 · 433679a · 433679a
1 parent 1228517
commit 433679a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/dataserv/app.py b/dataserv/app.py
@@ -4,6 +4,7 @@
 import sys
 import os.path
 import datetime
+from random import randint
 from flask import make_response, jsonify
 sys.path.append(os.path.join(os.path.dirname(__file__), '..'))
 
@@ -107,7 +108,7 @@ def total():
     # return in TB the number
     app.config["BYTE_SIZE"] = 1024*1024*128
     result = total_shards * (app.config["BYTE_SIZE"] / (1024*1024*1024*1024))  # bytes / 1 TB
-    json_data = {'total_TB': round(result, 2)}
+    json_data = { 'id': randint(0,9999999) ,'total_TB': round(result, 2) }
 
     return jsonify(json_data)
 

diff --git a/tests/test_App.py b/tests/test_App.py
@@ -154,6 +154,6 @@ def test_farmer_total_bytes(self):
 
         # check total bytes
         rv = self.app.get('/api/total')
-        json_data =  b'{\n  "total_TB": 1.22\n}'
+        json_data =  b'"total_TB": 1.22\n}'
 
-        self.assertEqual(json_data, rv.data)
+        self.assertTrue(json_data in  rv.data)