From afd381badfdbd7074954536326d921c023db1f8b Mon Sep 17 00:00:00 2001
From: Imdad Ahad <imdad.ahad@digital.cabinet-office.gov.uk>
Date: Tue, 20 Sep 2016 10:39:15 +0100
Subject: [PATCH] Update links schema to be more strict and fix import

---
 tests/app/notifications/rest/test_callbacks.py                 | 2 +-
 .../app/public_contracts/schemas/GET_notifications_return.json | 3 +++
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/tests/app/notifications/rest/test_callbacks.py b/tests/app/notifications/rest/test_callbacks.py
index eac7aa477..1d8cc85ca 100644
--- a/tests/app/notifications/rest/test_callbacks.py
+++ b/tests/app/notifications/rest/test_callbacks.py
@@ -3,7 +3,7 @@ import uuid
 from datetime import datetime
 from flask import json
 from freezegun import freeze_time
-from mock import call
+from unittest.mock import call
 
 import app.celery.tasks
 from app.dao.notifications_dao import (
diff --git a/tests/app/public_contracts/schemas/GET_notifications_return.json b/tests/app/public_contracts/schemas/GET_notifications_return.json
index 37ecf74e9..3c7c23071 100644
--- a/tests/app/public_contracts/schemas/GET_notifications_return.json
+++ b/tests/app/public_contracts/schemas/GET_notifications_return.json
@@ -20,6 +20,9 @@
         },
         "next" : {
           "type" : "string"
+        },
+        "prev" : {
+          "type" : "string"
         }
       }
     },