def test_create_(self):
# filename should contain ENVRIONMENT
csv_file = datawarehouse.create_kto_feedback_csv(self.csv_tmp_dir)
self.assertEqual(
os.path.join(self.csv_tmp_dir, 'kto-feedback-PRODUCTION.csv'),
csv_file)
# header and one entry should show up in written file.
with open(csv_file, 'r') as f:
reader = csv.DictReader(f)
i = None
for i, row in enumerate(reader):
self.assertEqual(row['_signal_id'],
str(self.feedback_submitted._signal.id))
self.assertEqual(row['is_satisfied'],
str(self.feedback_submitted.is_satisfied))
self.assertEqual(row['allows_contact'],
str(self.feedback_submitted.allows_contact))
self.assertEqual(row['text_extra'],
self.feedback_submitted.text_extra)
# noqa Disabled because the Postgres format is slightly different than the Python format, so we decided to comment these checks for now
# self.assertEqual(row['created_at'], str(self.feedback_submitted.created_at))
# self.assertEqual(row['submitted_at'], str(self.feedback_submitted.submitted_at))
self.assertEqual(row['text'], self.feedback_submitted.text)
self.assertEqual(i, 0)
def test_environment_set(self):
# filename should contain ENVRIONMENT
file_name = datawarehouse.create_kto_feedback_csv(self.csv_tmp_dir)
self.assertEqual(os.path.split(file_name)[-1], 'kto-feedback-PRODUCTION.csv')
# header and one entry should show up in written file.
with open(file_name, 'r') as f:
reader = csv.reader(f)
self.assertEqual(len(list(reader)), 2)
def test_environment_empty(self):
with self.assertRaises(EnvironmentError):
datawarehouse.create_kto_feedback_csv('dummy_location')
