X-Git-Url: https://git.mdrn.pl/redakcja.git/blobdiff_plain/e977f7187b10b1bc0a30794cd585c6b840568996..937a362ba8599c9571824493c13fed05223fb561:/src/documents/tests/test_publish.py

diff --git a/src/documents/tests/test_publish.py b/src/documents/tests/test_publish.py
index b59136a1..4e6592a8 100644
--- a/src/documents/tests/test_publish.py
+++ b/src/documents/tests/test_publish.py
@@ -5,7 +5,7 @@
 
 from documents.test_utils import get_fixture
 
-from mock import patch
+from unittest.mock import patch
 from django.test import TestCase
 from django.contrib.auth.models import User
 from documents.models import Book
@@ -26,7 +26,16 @@ class PublishTests(TestCase):
     def test_publish(self, api_call):
         self.book[0].head.set_publishable(True)
         self.book.publish(self.user)
-        api_call.assert_called_with(self.user, 'books/', {"book_xml": self.text1, "days": 0}, beta=False)
+        api_call.assert_called_with(
+            self.user,
+            'books/',
+            {
+                "book_xml": self.text1,
+                "days": 0,
+                "hidden": False
+            },
+            beta=False
+        )
 
     @patch('apiclient.api_call')
     def test_publish_multiple(self, api_call):
@@ -35,4 +44,13 @@ class PublishTests(TestCase):
         self.book[1].commit(get_fixture('chunk2.xml'))
         self.book[1].head.set_publishable(True)
         self.book.publish(self.user)
-        api_call.assert_called_with(self.user, 'books/', {"book_xml": get_fixture('expected.xml'), "days": 0}, beta=False)
+        api_call.assert_called_with(
+            self.user,
+            'books/',
+            {
+                "book_xml": get_fixture('expected.xml'),
+                "days": 0,
+                "hidden": False
+            },
+            beta=False
+        )