adding more tests

Author: esecules

src/groupsplit.py

@@ -46,9 +46,9 @@ def split(total, num_people):
     return base, extra
 
 class Splitwise:
-    def __init__(self, options, args):
-        if os.path.isfile(options.api_client):
-            with open(options.api_client, 'rb') as oauth_pkl:
+    def __init__(self, api_client='oauth_client.pkl'):
+        if os.path.isfile(api_client):
+            with open(api_client, 'rb') as oauth_pkl:
                 self.client = pickle.load(oauth_pkl)
         else:
             self.get_client()
@@ -129,6 +129,13 @@ def post_expense(self, uri):
             sys.stdout.write(".")
             sys.stdout.flush()
 
+    def delete_expense(self, expense_id):
+        return self.api_call("https://secure.splitwise.com/api/v3.0/delete_expense/%s" % expense_id, 'POST')
+
+    def get_expenses(self, limit=0):
+        resp = self.api_call("https://secure.splitwise.com/api/v3.0/get_expenses?limit=%s" % limit, 'GET')
+        return resp['expenses']
+            
 class CsvSettings():
     def __init__(self, rows):
         print "These are the first two rows of your csv"
@@ -244,7 +251,7 @@ def main():
     parser.add_option('', '--api-client', default='oauth_client.pkl', dest='api_client', help='supply different splitwise api client (for testing mostly)')
     options, args = parser.parse_args()
     logger.setLevel(log_levels[options.verbosity])
-    splitwise = Splitwise(options, args)
+    splitwise = Splitwise(options.api_client)
     split_gen = SplitGenerator(options, args, splitwise)
     print "Uploading splits"
     for uri in split_gen:

test/testSplit.py

@@ -3,9 +3,13 @@
 import subprocess
 from money import Money
 sys.path.append("../src")
-from groupsplit import split
+from groupsplit import split, Splitwise
 
 class UtilsTests(unittest.TestCase):
+    def __init__(self, *args, **kwargs):
+        super(UtilsTests, self).__init__(*args, **kwargs)
+        self.api = Splitwise()
+
     def test_splits(self):
         cases = [
             {"amount": "1.00", "ppl": 2, "expect": ("0.50","0.00")},
@@ -18,13 +22,25 @@ def test_splits(self):
             expect = (Money(case['expect'][0], "CAD"), Money(case['expect'][1], "CAD"))
             self.assertEqual(expect, split(Money(case['amount'], "CAD"), case['ppl']))
 
+    def test_get_id(self):
+        assertGreater(int(self.api.get_id()), 0)
+
+    def test_get_groups(self):
+        self.assertGreater(len(self.api.get_groups()), 0)
+
 class SystemTests(unittest.TestCase):
+    def __del__(self):
+        api = Splitwise()
+        for expense in api.get_expenses():
+            api.delete_expense(expense['id'])
+
     def test_group_of_2(self):
         proc = subprocess.Popen(['python', '../src/groupsplit.py', 'transactions.csv', 'group_of_2',
                                  '--csv-settings=csv_settings.pkl', '--api-client=oauth_client.pkl',
                                  '-y'], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
         stdout, stderr = proc.communicate()
         self.assertEqual(stderr, '')
+
     def test_group_of_3(self):
         proc = subprocess.Popen(['python', '../src/groupsplit.py', 'transactions.csv', 'group_of_3',
                                  '--csv-settings=csv_settings.pkl', '--api-client=oauth_client.pkl',