unit test for float bandwidth

StorjOld · Mar 14, 2016 · 0e997e5 · 0e997e5
1 parent fc9b012
commit 0e997e5
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/tests/test_App.py b/tests/test_App.py
@@ -245,20 +245,20 @@ def test_farmer_set_bandwidth(self):
         # not found
         btc_addr = self.gen_wallet()
         nodeid = address2nodeid(btc_addr)
-        rv = self.app.get('/api/bandwidth/{0}/1/2'.format(nodeid))
+        rv = self.app.get('/api/bandwidth/{0}/1.1/2.2'.format(nodeid))
         self.assertEqual(rv.status_code, 404)
 
         # register farmer
         self.app.get('/api/register/{0}/{1}'.format(nodeid, btc_addr))
 
         # correct
-        rv = self.app.get('/api/bandwidth/{0}/5/6'.format(nodeid))
+        rv = self.app.get('/api/bandwidth/{0}/5.5/6.6'.format(nodeid))
         self.assertEqual(rv.status_code, 200)
         # rv = self.app.get('/api/online/json'.format(nodeid))
         # self.assertTrue(b"5" in rv.data)
 
         # invalid btc address
-        rv = self.app.get('/api/bandwidth/{0}/1/2'.format(self.bad_addr))
+        rv = self.app.get('/api/bandwidth/{0}/1.1/2.2'.format(self.bad_addr))
         self.assertEqual(rv.status_code, 400)
 
 

diff --git a/tests/test_Farmer.py b/tests/test_Farmer.py
@@ -107,14 +107,14 @@ def test_bandwidth(self):
         # set height and check function output
         self.assertEqual(farmer.bandwidth_upload, 0)
         self.assertEqual(farmer.bandwidth_download, 0)
-        farmer.set_bandwidth(5, 6)
-        self.assertEqual(farmer.bandwidth_upload, 5)
-        self.assertEqual(farmer.bandwidth_download, 6)
+        farmer.set_bandwidth(5.5, 6.6)
+        self.assertEqual(farmer.bandwidth_upload, 5.5)
+        self.assertEqual(farmer.bandwidth_download, 6.6)
 
         # check the db object as well
         farmer2 = farmer.lookup()
-        self.assertEqual(farmer2.bandwidth_upload, 5)
-        self.assertEqual(farmer2.bandwidth_download, 6)
+        self.assertEqual(farmer2.bandwidth_upload, 5.5)
+        self.assertEqual(farmer2.bandwidth_download, 6.6)
 
     def test_height(self):
         btc_addr = self.btctxstore.get_address(self.btctxstore.get_key(