def test_expense_flow(self):
receipt = os.path.join(os.path.dirname(os.path.realpath(__file__)),
"receipt.jpg")
expense = Expense()
expense.setDescription("End2EndTest")
expense.setCost('10')
expense.setGroupId(19571167)
expense.setSplitEqually()
expense.setReceipt(receipt)
# create expense
expense, error = self.sObj.createExpense(expense)
self.assertIsNone(error)
self.assertIsNotNone(expense.getId())
# delete expense
success, error = self.sObj.deleteExpense(expense.getId())
self.assertIsNone(error)
self.assertTrue(success)
def test_createExpense_split_equally_with_receipt_success(
self, mockOpen, mockMakeRequest):
mockFile = MagicMock()
mockOpen.return_value = mockFile
mockMakeRequest.return_value = '{"expenses":[{"id":1010906976,"group_id":19433671,"friendship_id":null,"expense_bundle_id":null,"description":"Testing","repeats":false,"repeat_interval":"never","email_reminder":false,"email_reminder_in_advance":-1,"next_repeat":null,"details":null,"comments_count":0,"payment":false,"creation_method":null,"transaction_method":"offline","transaction_confirmed":false,"transaction_id":null,"cost":"10.0","currency_code":"SGD","repayments":[{"from":643871,"to":79774,"amount":"5.0"}],"date":"2020-06-24T06:57:29Z","created_at":"2020-06-24T06:57:29Z","created_by":{"id":79774,"first_name":"Naman","last_name":"Aggarwal","picture":{"medium":"https://splitwise.s3.amazonaws.com/uploads/user/avatar/79774/medium_mypic.jpg"},"custom_picture":true},"updated_at":"2020-06-24T06:57:29Z","updated_by":null,"deleted_at":null,"deleted_by":null,"category":{"id":18,"name":"General"},"receipt":{"large":null,"original":null},"users":[{"user":{"id":79774,"first_name":"Naman","last_name":"Aggarwal","picture":{"medium":"https://splitwise.s3.amazonaws.com/uploads/user/avatar/79774/medium_mypic.jpg"}},"user_id":79774,"paid_share":"10.0","owed_share":"5.0","net_balance":"5.0"},{"user":{"id":643871,"first_name":"Shantanu","last_name":"Alshi","picture":{"medium":"https://s3.amazonaws.com/splitwise/uploads/user/default_avatars/avatar-blue19-100px.png"}},"user_id":643871,"paid_share":"0.0","owed_share":"5.0","net_balance":"-5.0"}]}],"errors":{}}' # noqa: E501
expense = Expense()
expense.setCost('10')
expense.setDescription("Testing")
expense.setGroupId(19433671)
expense.setSplitEqually()
expense.setReceipt("temp.jpg")
expenseRes, errors = self.sObj.createExpense(expense)
mockMakeRequest.assert_called_with(
"https://secure.splitwise.com/api/v3.0/create_expense",
"POST", {
'cost': '10',
'description': 'Testing',
'group_id': 19433671,
'split_equally': True,
'receiptPath': 'temp.jpg'
},
files={'receipt': mockFile})
self.assertIsNone(errors)
self.assertEqual(expenseRes.getId(), 1010906976)
self.assertEqual(expenseRes.getGroupId(), 19433671)
self.assertEqual(expenseRes.getFriendshipId(), None)
self.assertEqual(expenseRes.getExpenseBundleId(), None)
self.assertEqual(expenseRes.getDescription(), "Testing")
self.assertEqual(expenseRes.getRepeatInterval(), "never")
self.assertEqual(expenseRes.getEmailReminderInAdvance(), -1)
self.assertEqual(expenseRes.getNextRepeat(), None)
self.assertEqual(expenseRes.getDetails(), None)
self.assertEqual(expenseRes.getCommentsCount(), 0)
self.assertEqual(expenseRes.getCreationMethod(), None)
self.assertEqual(expenseRes.getTransactionMethod(), "offline")
self.assertEqual(expenseRes.getTransactionId(), None)
self.assertEqual(expenseRes.getCost(), "10.0")
self.assertEqual(expenseRes.getCurrencyCode(), "SGD")
self.assertEqual(len(expenseRes.getRepayments()), 1)
self.assertEqual(expenseRes.getRepayments()[0].getFromUser(), 643871)
self.assertEqual(expenseRes.getRepayments()[0].getToUser(), 79774)
self.assertEqual(expenseRes.getRepayments()[0].getAmount(), "5.0")
self.assertEqual(expenseRes.getDate(), "2020-06-24T06:57:29Z")
self.assertEqual(expenseRes.getCreatedAt(), "2020-06-24T06:57:29Z")
self.assertEqual(expenseRes.getCreatedBy().getId(), 79774)
self.assertEqual(expenseRes.getCreatedBy().getFirstName(), "Naman")
self.assertEqual(expenseRes.getCreatedBy().getLastName(), "Aggarwal")
self.assertEqual(
expenseRes.getCreatedBy().getPicture().getMedium(),
"https://splitwise.s3.amazonaws.com/uploads/user/avatar/79774/medium_mypic.jpg"
)
self.assertEqual(expenseRes.getUpdatedAt(), "2020-06-24T06:57:29Z")
self.assertEqual(expenseRes.getUpdatedBy(), None)
self.assertEqual(expenseRes.getDeletedAt(), None)
self.assertEqual(expenseRes.getDeletedBy(), None)
self.assertEqual(expenseRes.getCategory().getId(), 18)
self.assertEqual(expenseRes.getCategory().getName(), "General")
self.assertEqual(expenseRes.getReceipt().getLarge(), None)
self.assertEqual(expenseRes.getReceipt().getOriginal(), None)
self.assertEqual(len(expenseRes.getUsers()), 2)
self.assertEqual(expenseRes.getUsers()[0].getId(), 79774)
self.assertEqual(expenseRes.getUsers()[0].getFirstName(), "Naman")
self.assertEqual(expenseRes.getUsers()[0].getLastName(), "Aggarwal")
self.assertEqual(
expenseRes.getUsers()[0].getPicture().getMedium(),
"https://splitwise.s3.amazonaws.com/uploads/user/avatar/79774/medium_mypic.jpg"
)
self.assertEqual(expenseRes.getUsers()[0].getPaidShare(), "10.0")
self.assertEqual(expenseRes.getUsers()[0].getOwedShare(), "5.0")
self.assertEqual(expenseRes.getUsers()[0].getNetBalance(), "5.0")
self.assertEqual(expenseRes.getUsers()[1].getId(), 643871)
self.assertEqual(expenseRes.getUsers()[1].getFirstName(), "Shantanu")
self.assertEqual(expenseRes.getUsers()[1].getLastName(), "Alshi")
self.assertEqual(
expenseRes.getUsers()[1].getPicture().getMedium(),
"https://s3.amazonaws.com/splitwise/uploads/user/default_avatars/avatar-blue19-100px.png"
)
self.assertEqual(expenseRes.getUsers()[1].getPaidShare(), "0.0")
self.assertEqual(expenseRes.getUsers()[1].getOwedShare(), "5.0")
self.assertEqual(expenseRes.getUsers()[1].getNetBalance(), "-5.0")
mockOpen.assert_called_with("temp.jpg", "rb")