def _handle_uploaded_file(file_name, file, manager):
base_name, extension = os.path.splitext(file_name)
if extension == '.csv':
csv_player = CsvPlayer(dbm=manager,
submission_handler=SubmissionHandler(manager),
parser=CsvParser())
response = csv_player.accept(file)
elif extension == '.xls':
xls_player = XlsPlayer(dbm=manager,
submission_handler=SubmissionHandler(manager),
parser=XlsParser())
response = xls_player.accept(file)
else:
raise InvalidFileFormatException()
return response
def setUp(self):
self.dbm = Mock(spec=DatabaseManager)
loc_tree = Mock()
loc_tree.get_hierarchy_path.return_value = None
self.submission_handler_mock = Mock(spec=SubmissionHandler)
self.parser = CsvParser()
self.csv_data = """
FORM_CODE,ID,BEDS,DIRECTOR,MEDS
CLF1,CL001,10,Dr. A,201
CLF1,CL002,11,Dr. B,202
CLF2,CL003,12,Dr. C,203
CLF1,CL004,13,Dr. D,204
CLF1,CL005,14,Dr. E,205
"""
self._mock_short_code_generator()
self.player = CsvPlayer(self.dbm, self.submission_handler_mock,
self.parser, loc_tree)
