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) 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) 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')