def _dump_to_csv(self, course_id, csv_dir):
"""
Create CSV files for submission/assessment data in a directory.
Args:
course_id (unicode): The ID of the course to dump data from.
csv_dir (unicode): The absolute path to the directory in which to create CSV files.
Returns:
None
"""
output_streams = {
name: open(os.path.join(csv_dir, rel_path), 'w')
for name, rel_path in six.iteritems(self.OUTPUT_CSV_PATHS)
}
csv_writer = CsvWriter(output_streams, self._progress_callback)
csv_writer.write_to_csv(course_id)
def test_write_to_csv(self, data):
# Create in-memory buffers for the CSV file data
output_streams = self._output_streams(data['expected_csv'].keys())
# Load the database fixture
# We use the database fixture to ensure that this test will
# catch backwards-compatibility issues even if the Django model
# implementation or API calls change.
self._load_fixture(data['fixture'])
# Write the data to CSV
writer = CsvWriter(output_streams)
writer.write_to_csv(data['course_id'])
# Check that the CSV matches what we expected
for output_name, expected_csv in data['expected_csv'].iteritems():
output_buffer = output_streams[output_name]
output_buffer.seek(0)
actual_csv = csv.reader(output_buffer)
for expected_row in expected_csv:
try:
actual_row = actual_csv.next()
except StopIteration:
actual_row = None
self.assertEqual(
actual_row, expected_row,
msg="Output name: {}".format(output_name)
)
# Check for extra rows
try:
extra_row = actual_csv.next()
except StopIteration:
extra_row = None
if extra_row is not None:
self.fail(u"CSV contains extra row: {}".format(extra_row))