Now with all the tests.

Author: mikeckennedy

app/ch14_testing/final/.idea/dictionaries/screencaster.xml

@@ -29,10 +29,12 @@ def register_blueprints():
     from pypi_org.views import package_views
     from pypi_org.views import cms_views
     from pypi_org.views import account_views
+    from pypi_org.views import seo_view
 
     app.register_blueprint(package_views.blueprint)
     app.register_blueprint(home_views.blueprint)
     app.register_blueprint(account_views.blueprint)
+    app.register_blueprint(seo_view.blueprint)
     app.register_blueprint(cms_views.blueprint)
 
 

app/ch14_testing/final/.idea/flask-testing.iml

@@ -1,5 +1,6 @@
 from typing import List, Optional
 import sqlalchemy.orm
+from sqlalchemy.orm import Session
 
 import pypi_org.data.db_session as db_session
 from pypi_org.data.package import Package
@@ -56,3 +57,11 @@ def get_package_by_id(package_id: str) -> Optional[Package]:
         session.close()
 
     return package
+
+
+def all_packages(limit: int) -> List[Package]:
+    session: Session = db_session.create_session()
+    try:
+        return list(session.query(Package).limit(limit))
+    finally:
+        session.close()

app/ch14_testing/final/pypi_org/app.py

@@ -0,0 +1,2 @@
+User-agent: *
+Disallow:

app/ch14_testing/final/pypi_org/services/package_service.py

@@ -0,0 +1,33 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<urlset xmlns="http://www.sitemaps.org/schemas/sitemap/0.9">
+    <!-- STATIC CONSTANT URLS -->
+    <url>
+        <loc>{{ site }}</loc>
+        <lastmod>{{ last_updated_text }}</lastmod>
+        <changefreq>daily</changefreq>
+        <priority>1.0</priority>
+    </url>
+    <url>
+        <loc>{{site}}/account/login</loc>
+        <lastmod>{{ last_updated_text }}</lastmod>
+        <changefreq>yearly</changefreq>
+        <priority>0.8</priority>
+    </url>
+    <url>
+        <loc>{{site}}/account/register</loc>
+        <lastmod>${last_updated_text}</lastmod>
+        <changefreq>yearly</changefreq>
+        <priority>0.8</priority>
+    </url>
+
+
+    <!-- DYNAMIC PAGES (basically packages) -->
+    {% for p in packages %}
+    <url>
+        <loc>{{site}}/project/{{p.id}}</loc>
+        <lastmod>{{p.created_date.isoformat()}}</lastmod>
+        <changefreq>weekly</changefreq>
+        <priority>1.0</priority>
+    </url>
+    {%  endfor %}
+</urlset>

app/ch14_testing/final/pypi_org/templates/seo/robots.txt

@@ -0,0 +1,12 @@
+import flask
+
+from pypi_org.services import package_service
+from pypi_org.viewmodels.shared.viewmodelbase import ViewModelBase
+
+
+class SiteMapViewModel(ViewModelBase):
+    def __init__(self, limit: int):
+        super().__init__()
+        self.packages = package_service.all_packages(limit)
+        self.last_updated_text = "2018-07-15"
+        self.site = "{}://{}".format(flask.request.scheme, flask.request.host)

app/ch14_testing/final/pypi_org/templates/seo/sitemap.html

@@ -0,0 +1,24 @@
+import flask
+
+from pypi_org.infrastructure.view_modifiers import response
+from pypi_org.viewmodels.seo.sitemap_viewmodel import SiteMapViewModel
+
+blueprint = flask.Blueprint('seo', __name__, template_folder='templates')
+
+
+# ################### Sitemap #################################
+
+
+@blueprint.route('/sitemap.xml')
+@response(mimetype='application/xml', template_file='seo/sitemap.html')
+def sitemap():
+    vm = SiteMapViewModel(1000)
+    return vm.to_dict()
+
+
+# ################### Robots #################################
+
+@blueprint.route('/robots.txt')
+@response(mimetype='text/plain', template_file='seo/robots.txt')
+def robots():
+    return {}

app/ch14_testing/final/pypi_org/viewmodels/seo/sitemap_viewmodel.py

@@ -0,0 +1,8 @@
+# noinspection PyUnresolvedReferences
+from account_tests import *
+# noinspection PyUnresolvedReferences
+from package_tests import *
+# noinspection PyUnresolvedReferences
+from sitemap_tests import *
+# noinspection PyUnresolvedReferences
+from home_tests import *

app/ch14_testing/final/pypi_org/views/seo_view.py

@@ -0,0 +1,19 @@
+from flask import Response
+
+from tests.test_client import client, flask_app
+from pypi_org.views import home_views
+
+
+def test_int_homepage(client):
+    r: Response = client.get('/')
+    assert r.status_code == 200
+    assert b'Find, install and publish Python packages' in r.data
+
+
+def test_v_homepage_directly():
+    with flask_app.test_request_context(path='/'):
+        r: Response = home_views.index()
+
+    assert r.status_code == 200
+    # noinspection PyUnresolvedReferences
+    assert len(r.model.get('releases')) > 0