From 4982b0ffbbb2115956f57061ed541096df47f64a Mon Sep 17 00:00:00 2001
From: Alex Zai <azai91@gmail.com>
Date: Sun, 22 Nov 2015 13:42:21 -0800
Subject: [PATCH] (feat) update test

---
 Makefile                |  6 +++---
 tests/venmo_api_test.py | 21 ++++++++++++++++-----
 2 files changed, 19 insertions(+), 8 deletions(-)

diff --git a/Makefile b/Makefile
index bee7f57..006d09f 100644
--- a/Makefile
+++ b/Makefile
@@ -1,5 +1,5 @@
 test:
-	# python -m tests.util_test
-	# python -m tests.venmo_api_test
-	# python -m tests.server_test
+	python -m tests.util_test
+	python -m tests.venmo_api_test
+	python -m tests.server_test
 	python -m tests.venmo_test
diff --git a/tests/venmo_api_test.py b/tests/venmo_api_test.py
index e58dfd4..55a89af 100644
--- a/tests/venmo_api_test.py
+++ b/tests/venmo_api_test.py
@@ -10,6 +10,8 @@
 from src.config import FRIENDS_URL, LOGIN, LOGOUT, CLEAR_CACHE, INVALID
 import json
 import logging
+import os
+import sys
 # from src.venmo_api import Venmo as Venmo_backup
 
 CachedData = {}
@@ -95,36 +97,42 @@ def test_generate_payload(self):
             'user_id' : friend['id'],
             'amount' : '[amount]',
             'note' : '[note]',
+            'display_name' : friend['display_name']
             })
 
         self.assertEqual(Venmo.generate_payload(['1'], friend),{
             'user_id' : friend['id'],
             'amount' : '1.00',
-            'note' : '[note]'
+            'note' : '[note]',
+            'display_name' : friend['display_name']
             })
 
         self.assertEqual(Venmo.generate_payload(['-'], friend),{
             'user_id' : friend['id'],
             'amount' : '[amount]',
-            'note' : '[note]'
+            'note' : '[note]',
+            'display_name' : friend['display_name']
             })
 
         self.assertEqual(Venmo.generate_payload(['1',''], friend),{
             'user_id' : friend['id'],
             'amount' : '1.00',
-            'note' : '[note]'
+            'note' : '[note]',
+            'display_name' : friend['display_name']
             })
 
         self.assertEqual(Venmo.generate_payload(['1','t'], friend),{
             'user_id' : friend['id'],
             'amount' : '1.00',
-            'note' : 't'
+            'note' : 't',
+            'display_name' : friend['display_name']
             })
 
         self.assertEqual(Venmo.generate_payload(['1','test test'], friend),{
             'user_id' : friend['id'],
             'amount' : '1.00',
-            'note' : 'test test'
+            'note' : 'test test',
+            'display_name' : friend['display_name']
             })
 
     def test_format_title(self):
@@ -197,6 +205,9 @@ def test_show_formatting(self):
 
 
     def setUp(self):
+        f = open(os.devnull, 'w')
+        sys.stdout = f
+
         CachedData.clear()
         StoredData.clear()
         Passwords.clear()