def test_read_write_list_with_append(self):
for i in range(5):
sync_write_list_to_csv(['A', 'B'], [{'A': '20', 'B': 40}], CSV_TEST_FILE, 'a')
l = read_csv_as_list(CSV_TEST_FILE)
self.assertEqual(len(l), 5)
os.remove(CSV_TEST_FILE) # clean up
def test_add_item(self):
common.TMP_FOLDER_PATH = './'
common.Timeline.add_item('Sachin', 'its friday', 'Woooooo', 'test.gif', 'fa-beer', 'warning')
l = read_csv_as_list(TIMELINE_FILE)
self.assertEqual(l[0]['name'], 'Sachin')
self.assertEqual(l[0]['title'], 'its friday')
self.assertEqual(l[0]['content'], 'Woooooo')
self.assertEqual(l[0]['img'], '../../../uploads/sachin/test.gif')
self.assertEqual(l[0]['icon'], 'fa-beer')
self.assertEqual(l[0]['iconBackground'], 'warning')
os.remove(TIMELINE_FILE) # clean up
def test_read_write_list(self):
sync_write_list_to_csv(
['A', 'B', 'C'],
[
{'A': '20', 'B': 30, 'C': {'complex': 'iam'}},
{'A': '20', 'B': 30, 'C': ['complex', 'you', 'are']}
], CSV_TEST_FILE, 'w')
l = read_csv_as_list(CSV_TEST_FILE)
self.assertEqual(len(l), 2)
self.assertEqual(l[0]['A'], '20')
self.assertEqual(l[0]['B'], '30')
self.assertEqual(ast.literal_eval(l[0]['C'])['complex'], 'iam')
self.assertEqual(ast.literal_eval(l[1]['C'])[1], 'you')
os.remove(CSV_TEST_FILE) # clean up
__author__ = 'sachinpatney'
import json
import sys
sys.path.append('/var/www/git/redalert/tasks')
from common import read_csv_as_list
print(json.dumps(read_csv_as_list(sys.argv[1])))
