def test_fetch_text_file_with_correct_data(self):
# Arrange
file_name = "testdata\\test_data_2_rows.txt"
line1 = {
'gender': 'F',
'age': '21',
'sales': '001',
'bmi': 'Normal',
'salary': '12',
'birthday': '01/01/1996',
'valid': '1'
}
line2 = {
'gender': 'M',
'age': '45',
'sales': '999',
'bmi': 'Underweight',
'salary': '725',
'birthday': '31/12/1971',
'valid': '1'
}
expected_result = {'A001': line1, 'Q001': line2}
file = FileReader.fetch_text_file(file_name)
# Act
result = FileReader.fetch_text_contents(file, "", ",")
# Assert
self.assertTrue(result == expected_result)
def test_fetch_text_file_not_found_returns_nothing(self):
# Arrange
file_name = "testdata\\file_does_not_exist.txt"
expected_result = None
file = FileReader.fetch_text_file(file_name)
# Act
result = FileReader.fetch_text_contents(file, "", ",")
# Assert
self.assertTrue(result == expected_result)
