def test_not_a_txt_file(self):
"""
This method tests if validation error is raised when a non-text file is
uploaded
"""
f = File('invalid')
f.content_type = 'invalid'
with self.assertRaises(ValueError):
extract_text_file(f)
def test_text_file(self):
"""
This method tests if content of the text file are extracted when
a text file is uploaded
"""
f = ContentFile('This is a sample content', name='test.txt')
f.content_type = 'text/plain'
content = extract_text_file(f)
self.assertEquals(content.strip(), 'This is a sample content')
