def test_get_file_empty_field(self): """ObjectDownloadView.get_file() raises FileNotFound if field does not exist.""" view = setup_view(views.ObjectDownloadView(file_field='other_field'), 'fake request') view.object = mock.Mock() view.object.other_field = None with self.assertRaises(exceptions.FileNotFound): view.get_file()
def test_get_file_wrong_field(self): """ObjectDownloadView.get_file() raises AttributeError if field does not exist. ``AttributeError`` is expected because this is a configuration error, i.e. it is related to Python code. """ view = setup_view(views.ObjectDownloadView(file_field='other_field'), 'fake request') view.object = mock.Mock(spec=['file']) with self.assertRaises(AttributeError): view.get_file()
def test_get_file_ok(self): "ObjectDownloadView.get_file() returns ``file`` field by default." view = setup_view(views.ObjectDownloadView(), 'fake request') view.object = mock.Mock(spec=['file']) view.get_file()