tests: updated tests and added alembic recipe

invenio_pages/alembic/b0f93ca4a147_add_has_custom_view_to_pages.py

@@ -0,0 +1,54 @@
+#
+# This file is part of Invenio.
+# Copyright (C) 2016-2018 CERN.
+#
+# Invenio is free software; you can redistribute it and/or modify it
+# under the terms of the MIT License; see LICENSE file for more details.
+
+"""Add has_custom_view to pages."""
+
+import sqlalchemy as sa
+import sqlalchemy_utils
+from alembic import op
+from sqlalchemy.dialects import postgresql
+
+# revision identifiers, used by Alembic.
+revision = "b0f93ca4a147"
+down_revision = "145402e8523a"
+branch_labels = ()
+depends_on = "98e0a418340d"
+
+
+def upgrade():
+    """Upgrade database."""
+    # Add the new column
+    op.add_column(
+        "pages_page",
+        sa.Column(
+            "has_custom_view",
+            sa.Boolean(),
+            nullable=True,
+            server_default=sa.sql.expression.literal(False),
+            default=False,
+        ),
+    )
+    op.add_column(
+        "pages_page_version",
+        sa.Column(
+            "has_custom_view",
+            sa.Boolean(),
+            nullable=True,
+            server_default=sa.sql.expression.literal(False),
+            default=False,
+        ),
+    )
+
+    # ### end Alembic commands ###
+
+
+def downgrade():
+    """Downgrade database."""
+    # Drop the column
+    op.drop_column("pages_page", "has_custom_view")
+    op.drop_column("pages_page_version", "has_custom_view")
+    # ### end Alembic commands ###

tests/records/test_records.py

@@ -18,7 +18,10 @@
 
 def test_page_repr(module_scoped_pages_fixture, base_app):
     dog_page = Page.get_by_url("/dogs/shiba")
-    assert dog_page.__repr__() == "URL: /dogs/shiba, title: Page for doge!"
+    assert (
+        dog_page.__repr__()
+        == "URL: /dogs/shiba, title: Page for doge!, has_custom_view: False"
+    )
 
 
 def test_page_versions(module_scoped_pages_fixture, base_app, db):

tests/resources/test_resources.py

@@ -27,6 +27,7 @@ def test_page_content(module_scoped_pages_fixture, base_client):
         "content": "Generic dog.",
         "id": "1",
         "template_name": "invenio_pages/default.html",
+        "has_custom_view": False,
         "links": {"self": "https://127.0.0.1:5000/api/pages/1"},
     }
     assert json == expected_data
@@ -47,6 +48,7 @@ def test_html_content(module_scoped_pages_fixture, base_client):
         "content": "<h1>HTML aware dog.</h1>.\n" '<p class="test">paragraph<br /></p>',
         "id": "4",
         "template_name": "invenio_pages/default.html",
+        "has_custom_view": False,
         "links": {"self": "https://127.0.0.1:5000/api/pages/4"},
     }
     assert json == expected_data

tests/services/test_services.py

@@ -33,6 +33,7 @@ def test_page_read(module_scoped_pages_fixture, simple_user_identity):
         "content": "Generic dog.",
         "id": "1",
         "template_name": "invenio_pages/default.html",
+        "has_custom_view": False,
         "links": {"self": "https://127.0.0.1:5000/api/pages/1"},
     }
     assert page == expected_data
@@ -50,6 +51,7 @@ def test_page_read_by_url(module_scoped_pages_fixture, simple_user_identity):
         "content": "Generic dog.",
         "id": "1",
         "template_name": "invenio_pages/default.html",
+        "has_custom_view": False,
         "links": {"self": "https://127.0.0.1:5000/api/pages/1"},
     }
     assert page == expected_data