test(modules): module2 complete

Author: tjbell

tasks.md

@@ -151,7 +151,7 @@ In the body of the `get_db` function use the built_in `getattr()` function to ge
 
 ## 3.5 - Global Database Connection
 
-@pytest.mark.app_get_db_connection Still in the `get_db` function, test if `db` is `None` if it is, set `db` and `g._database` to `sqlite3.connect(DATABASE)` using multiple assignment. 
+@pytest.mark.app_get_db_connection Still in the `get_db` function, test if `db` is `None` if it is, set `db` and `g._database` to `sqlite3.connect(DATABASE)` using multiple assignment.
 
 ## 3.6 - sqlite3 Row Factory
 

tests/test_module1.py

@@ -26,7 +26,8 @@ def test_templates_folder_module1():
 @pytest.mark.index_template
 def test_index_template_module1():
     assert template_exists('index'), 'The `index.html` template does not exist in the `templates` folder.'
-    assert tag_contains('index', 'h1', 'Jobs'), "The `<h1>` in the `index.html` template does not contain the contents 'Jobs'."
+    assert template_find('index', 'h1', limit=1), "The `<h1>` in the `index.html` template does not contain the contents 'Jobs'."
+    assert template_find('index', 'h1', limit=1).text == 'Jobs', "The `<h1>` in the `index.html` template does not contain the contents 'Jobs'."
 
 @pytest.mark.app_index_route_function
 def test_app_index_route_function_module1():

tests/test_module2.py

@@ -1,24 +1,37 @@
 import pytest
 import sys
 
+from jinja2 import Environment, PackageLoader, meta
 from jobs import app
+from .utils import *
+
+values = template_values('layout', 'url_for')
 
 @pytest.mark.layout_template
 def test_layout_template_module2():
-    pass
+    assert template_exists('layout'), 'The `layout.html` template does not exist in the `templates` folder.'
 
 @pytest.mark.add_bulma_css_framework
 def test_add_bulma_css_framework_module2():
-    pass
+    assert template_exists('layout'), 'The `layout.html` template does not exist in the `templates` folder.'
+    assert 'static:filename:css/bulma.min.css' in values, 'Looks like `bulma.min.css` is not linked in `layout.html`.'
 
 @pytest.mark.add_custom_css
 def test_add_custom_css_module2():
-    pass
+    assert template_exists('layout'), 'The `layout.html` template does not exist in the `templates` folder.'
+    assert 'static:filename:css/app.css' in values, 'Looks like `app.css` is not linked in `layout.html`.'
 
 @pytest.mark.add_fontawesome
 def test_add_fontawesome_module2():
-    pass
+    assert template_exists('layout'), 'The `layout.html` template does not exist in the `templates` folder.'
+    attr = {
+        'href': 'https://use.fontawesome.com/releases/v5.2.0/css/all.css',
+        'rel': 'stylesheet'
+    }
+    assert template_doc('layout').find('link', attr), 'Looks like FontAwesome is not linked in `layout.html`.'
 
 @pytest.mark.extend_base_template
 def test_extend_base_template_module2():
-    pass
+    assert template_exists('index'), 'The `index.html` template does not exist in the `templates` folder.'
+    assert template_exists('layout'), 'The `layout.html` template does not exist in the `templates` folder.'
+    assert 'layout.html' in template_extends('index'), 'The `index.html` template does not extend `layout.html`.'

tests/test_module3.py

@@ -2,26 +2,31 @@
 import sys
 
 from jobs import app
+from .utils import *
+
 
 @pytest.mark.app_import_sqlite
 def test_app_import_sqlite_module3():
-    pass
+    assert 'sqlite3' in dir(app), 'Have you imported `sqlite`'
 
 @pytest.mark.app_import_g
 def test_app_import_g_module3():
-    pass
+    assert 'g' in dir(app), 'Have you imported the `g` class from `flask`'
 
 @pytest.mark.app_db_path
 def test_app_db_path_module3():
-    pass
+    assert 'DATABASE' in dir(app), 'Have you created a constant called `DATABASE`.'
+    assert app.DATABASE == 'db/jobs.sqlite', 'Have you created a constant called `DATABASE`.'
 
 @pytest.mark.app_get_db_get_attribute
 def test_app_get_db_get_attribute_module3():
-    pass
+    assert 'get_db' in dir(app), 'Have you defined a function named `get_db`.'
+    assert 'getattr:g:_database:None' in get_functions(app.get_db)
 
 @pytest.mark.app_get_db_connection
 def test_app_get_db_connection_module3():
-    pass
+    get_statements(app.get_db)
+    assert False
 
 @pytest.mark.app_get_db_row_factory
 def test_app_get_db_row_factory_module3():

tests/utils.py

@@ -3,8 +3,11 @@
 import inspect
 from pprint import pprint
 
+from jinja2 import nodes, Environment, PackageLoader, meta, exceptions
 from bs4 import BeautifulSoup
 
+env = Environment(loader=PackageLoader('jobs', 'templates'))
+
 def get_decorators(source):
     decorators = {}
 
@@ -23,36 +26,74 @@ def visit_FunctionDef(node):
     node_iter = ast.NodeVisitor()
     node_iter.visit_FunctionDef = visit_FunctionDef
     node_iter.visit(ast.parse(inspect.getsource(source)))
-    return decorators
 
+    return decorators
 
 def get_functions(source):
     functions = []
 
     def visit_Call(node):
         name = node.func.attr if isinstance(node.func, ast.Attribute) else node.func.id
-        functions.append(name + ':' + ':'.join([a.s for a in node.args]))
+        if name == 'getattr':
+            functions.append(name + ':' + node.args[0].id + ':' + node.args[1].s + ':' + str(node.args[2].value))
+        else:
+            functions.append(name + ':' + ':'.join([a.s for a in node.args]))
 
     node_iter = ast.NodeVisitor()
     node_iter.visit_Call = visit_Call
     node_iter.visit(ast.parse(inspect.getsource(source)))
 
     return functions
 
+
+def get_statements(source):
+    statment = []
+
+    def visit_If(node):
+        print(ast.dump(node))
+
+    node_iter = ast.NodeVisitor()
+    node_iter.visit_If = visit_If
+    node_iter.visit(ast.parse(inspect.getsource(source)))
+
+    return statment
+
 def list_routes(app):
     rules = []
+
     for rule in app.url_map.iter_rules():
         methods = ','.join(sorted(rule.methods))
         if rule.endpoint is not 'static':
-            rules.append(rule.endpoint+':'+ methods +':'+ str(rule))
+            rules.append(rule.endpoint + ':' + methods + ':' + str(rule))
+
     return rules
 
+def template_values(name, function):
+    values = []
+
+    for node in parsed_content(name).find_all(nodes.Call):
+        if node.node.name == function:
+            values.append(node.args[0].value + ':' + node.kwargs[0].key + ':' +  node.kwargs[0].value.value)
+
+    return values
+
 def template_exists(name):
     return os.path.isfile('jobs/templates/' + name + '.html')
 
-def template_contains(name, tag):
-    doc = BeautifulSoup(open(os.getcwd() + '/jobs/templates/' + name + '.html'), 'html.parser')
-    return doc.find(tag)
+def template_source(name):
+    try:
+        return env.loader.get_source(env, name + '.html')[0]
+    except exceptions.TemplateNotFound:
+        return None
+
+def template_doc(name):
+    return BeautifulSoup(template_source(name), 'html.parser')
+
+def template_find(name, tag, limit=None):
+    return BeautifulSoup(template_source(name), 'html.parser').find_all(tag, limit=limit)
+
+def parsed_content(name):
+    return env.parse(template_source(name))
 
-def tag_contains(name, tag, text):
-    return template_contains(name, tag).text if template_contains(name, tag) else False
+def template_extends(name):
+    return list(meta.find_referenced_templates(parsed_content(name)))