Fix report_result

1 files changed, 6 insertions, 6 deletions

diff --git a/clientlib.py b/clientlib.py
index c6f1b04..b9a348a 100644
--- a/clientlib.py
+++ b/clientlib.py
@@ -74,7 +74,11 @@ class IRBClient(BaseClient):
         params = {'treeid':treeid, 'testid':testid, 'result': result}
         if detail_url:
             params['detail_url'] = detail_url
-        self.send_request('/add_result', auth=True, req_params=params)
+        r = self.send_request('/add_result', auth=True, req_params=params)
+        if 'id' in r:
+            return r['id']
+        else:
+            pass # FIXME raise an exception or something
 
     def report_result(self, testid, tree_data, result, add_tree=False):
         '''Convenience method that will look up the tree using tree_data and
@@ -99,8 +103,4 @@ class IRBClient(BaseClient):
             return False
         else:
             tree = treelist.pop()
-        r = self.add_result(treeid=tree['id'], testid=testid, result=result)
-        if 'id' in r:
-            return r['id']
-        else:
-            pass # FIXME raise an exception or something
+        return self.add_result(treeid=tree['id'], testid=testid, result=result)