Removed unneeded parameters after update to api.

Updated the tests to take the removal into account.

Tidied up unused imports

app/main/views/notifications.py

@@ -1,6 +1,5 @@
 # -*- coding: utf-8 -*-
 import base64
-from collections import ItemsView
 from datetime import datetime
 
 from flask import (
@@ -116,7 +115,6 @@ def view_letter_notification_as_preview(service_id, notification_id, filetype):
         page=request.args.get('page')
     )
 
-    # return base64.b64decode(preview['content']), preview['status'], ItemsView(dict(preview['headers']))
     return base64.b64decode(preview['content'])
 
 

tests/app/main/views/test_notifications.py

@@ -167,9 +167,7 @@ def test_should_show_image_of_letter_notification(
     mocked_api_client = mocker.patch(
         'app.notify_client.notification_api_client.NotificationApiClient.get',
         return_value={
-            'content': base64.b64encode(b'foo').decode('utf-8'),
-            'status': 200, 
-            'headers': [{'test': 'test', 'test1': 'test1'}]
+            'content': base64.b64encode(b'foo').decode('utf-8')
         }
     )