test(modules): add module suffix

Author: tjbell

tests/test_module1.py

@@ -1,52 +1,45 @@
 import pytest
-import sys, os, inspect, re
-
-from bs4 import BeautifulSoup
+import os 
+import inspect
 
+from pprint import pprint
+from .utils import *
 from jobs import app
 
-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))
-    return rules
 
 @pytest.mark.app_import_flask
-def test_app_import_flask():
+def test_app_import_flask_module1():
     assert 'Flask' in dir(app), 'Have you imported the `Flask` class from `flask`'
     assert inspect.isclass(app.Flask), '`Flask` is not a class.'
     assert 'render_template' in dir(app), '`render_template` has not been imported.'
-    assert callable(app.render_template), '`render_template` is not a function.'
+    assert inspect.isfunction(app.render_template), '`render_template` is not a function.'
 
 @pytest.mark.app_create_flask_app
-def test_app_create_flask_app():
+def test_app_create_flask_app_module1():
     assert 'app' in dir(app), 'Have you created an instance of the `Flask` class called `app`?'
     assert isinstance(app.app, app.Flask), '`app` is not an instance of the `Flask` class.'
 
 @pytest.mark.templates_folder
-def test_templates_folder():
+def test_templates_folder_module1():
     assert os.path.isdir('jobs/templates'), 'The `templates` folder has not been created.'
 
 @pytest.mark.index_template
-def test_index_template():
-    assert os.path.isfile('jobs/templates/index.html'), 'The `index.html` template does not exist in the `templates` folder.'
-    index = BeautifulSoup(open(os.getcwd() + '/jobs/templates/index.html'), 'html.parser')
-    assert index.find('h1'), 'The `index.html` template does not contain an `<h1>`.'
-    assert index.find('h1').text == 'Jobs', "The `<h1>` in the `index.html` template does not contain the contents 'Jobs'."
+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'."
 
 @pytest.mark.app_index_route_function
-def test_app_index_route_function():
-    assert os.path.isfile('jobs/templates/index.html'), 'The `index.html` template does not exist in the `templates` folder.'
-    assert 'jobs' in dir(app)
+def test_app_index_route_function_module1():
+    assert 'app' in dir(app), 'Have you created an instance of the `Flask` class called `app`?'
+    assert 'jobs' in dir(app), 'Have you created the `jobs` function?'
     with app.app.app_context():
-        assert re.findall(r"return\s*render_template\s*\(\s*(?:'|\")index\.html(?:'|\")\s*\)", inspect.getsource(app.jobs))
+        assert 'render_template:index.html' in get_functions(app.jobs)
 
 @pytest.mark.app_route_decoractors
-def test_app_route_decoractors():
-    assert os.path.isfile('jobs/templates/index.html'), 'The `index.html` template does not exist in the `templates` folder.'
-    assert 'jobs' in dir(app)
+def test_app_route_decoractors_module1():
+    assert 'app' in dir(app), 'Have you created an instance of the `Flask` class called `app`?'
+    assert template_exists('index'), 'The `index.html` template does not exist in the `templates` folder.'
+    assert 'jobs' in dir(app), 'Have you created the `jobs` function?'
 
     rules = list_routes(app.app)
 

tests/test_module2.py

@@ -4,21 +4,21 @@
 from jobs import app
 
 @pytest.mark.layout_template
-def test_layout_template():
+def test_layout_template_module2():
     pass
 
 @pytest.mark.add_bulma_css_framework
-def test_add_bulma_css_framework():
+def test_add_bulma_css_framework_module2():
     pass
 
 @pytest.mark.add_custom_css
-def test_add_custom_css():
+def test_add_custom_css_module2():
     pass
 
 @pytest.mark.add_fontawesome
-def test_add_fontawesome():
+def test_add_fontawesome_module2():
     pass
 
 @pytest.mark.extend_base_template
-def test_extend_base_template():
+def test_extend_base_template_module2():
     pass

tests/test_module3.py

@@ -4,53 +4,53 @@
 from jobs import app
 
 @pytest.mark.app_import_sqlite
-def test_app_import_sqlite():
+def test_app_import_sqlite_module3():
     pass
 
 @pytest.mark.app_import_g
-def test_app_import_g():
+def test_app_import_g_module3():
     pass
 
 @pytest.mark.app_db_path
-def test_app_db_path():
+def test_app_db_path_module3():
     pass
 
 @pytest.mark.app_get_db_get_attribute
-def test_app_get_db_get_attribute():
+def test_app_get_db_get_attribute_module3():
     pass
 
 @pytest.mark.app_get_db_connection
-def test_app_get_db_connection():
+def test_app_get_db_connection_module3():
     pass
 
 @pytest.mark.app_get_db_row_factory
-def test_app_get_db_row_factory():
+def test_app_get_db_row_factory_module3():
     pass
 
 @pytest.mark.app_query_db
-def test_app_query_db():
+def test_app_query_db_module3():
     pass
 
 @pytest.mark.app_query_db_parameters
-def test_app_query_db_parameters():
+def test_app_query_db_parameters_module3():
     pass
 
 @pytest.mark.app_query_db_execute
-def test_app_query_db_execute():
+def test_app_query_db_execute_module3():
     pass
 
 @pytest.mark.app_query_db_fetchall
-def test_app_query_db_fetchall():
+def test_app_query_db_fetchall_module3():
     pass
 
 @pytest.mark.app_query_db_one
-def test_app_query_db_one():
+def test_app_query_db_one_module3():
     pass
 
 @pytest.mark.app_close_connection
-def test_app_close_connection():
+def test_app_close_connection_module3():
     pass
 
 @pytest.mark.app_close_connection_decorator
-def test_app_close_connection_decorator():
+def test_app_close_connection_decorator_module3():
     pass

tests/test_module4.py

@@ -4,49 +4,49 @@
 from jobs import app
 
 @pytest.mark.template_macros
-def test_template_macros():
+def test_template_macros_module4():
     pass
 
 @pytest.mark.show_job_macro_definition
-def test_show_job_macro_definition():
+def test_show_job_macro_definition_module4():
     pass
 
 @pytest.mark.show_job_macro_html
-def test_show_job_macro_html():
+def test_show_job_macro_html_module4():
     pass
 
 @pytest.mark.show_job_macro_header
-def test_show_job_macro_header():
+def test_show_job_macro_header_module4():
     pass
 
 @pytest.mark.show_job_macro_body
-def test_show_job_macro_body():
+def test_show_job_macro_body_module4():
     pass
 
 @pytest.mark.show_jobs_macro_definition
-def test_show_jobs_macro_definition():
+def test_show_jobs_macro_definition_module4():
     pass
 
 @pytest.mark.show_jobs_macro_for_loop
-def test_show_jobs_macro_for_loop():
+def test_show_jobs_macro_for_loop_module4():
     pass
 
 @pytest.mark.show_jobs_macro_for_loop_body
-def test_show_jobs_macro_for_loop_body():
+def test_show_jobs_macro_for_loop_body_module4():
     pass
 
 @pytest.mark.import_macros
-def test_import_macros():
+def test_import_macros_module4():
     pass
 
 @pytest.mark.index_template
-def test_index_template():
+def test_index_template_module4():
     pass
 
 @pytest.mark.display_all_jobs
-def test_display_all_jobs():
+def test_display_all_jobs_module4():
     pass
 
 @pytest.mark.app_jobs_route_jobs
-def test_app_jobs_route_jobs():
+def test_app_jobs_route_jobs_module4():
     pass

tests/test_module5.py

@@ -4,25 +4,25 @@
 from jobs import app
 
 @pytest.mark.app_job_template
-def test_app_job_template():
+def test_app_job_template_module5():
     pass
 
 @pytest.mark.app_job_route
-def test_app_job_route():
+def test_app_job_route_module5():
     pass
 
 @pytest.mark.app_job_route_decorator
-def test_app_job_route_decorator():
+def test_app_job_route_decorator_module5():
     pass
 
 @pytest.mark.app_job_route_parameter
-def test_app_job_route_parameter():
+def test_app_job_route_parameter_module5():
     pass
 
 @pytest.mark.app_job_route_data
-def test_app_job_route_data():
+def test_app_job_route_data_module5():
     pass
 
 @pytest.mark.app_job_route_pass_data
-def test_app_job_route_pass_data():
+def test_app_job_route_pass_data_module5():
     pass

tests/test_module6.py

@@ -4,41 +4,41 @@
 from jobs import app
 
 @pytest.mark.employer_template
-def test_employer_template():
+def test_employer_template_module6():
     pass
 
 @pytest.mark.employer_template_details
-def test_employer_template_details():
+def test_employer_template_details_module6():
     pass
 
 @pytest.mark.employer_template_all_jobs
-def test_employer_template_all_jobs():
+def test_employer_template_all_jobs_module6():
     pass
 
 @pytest.mark.employer_template_reviews
-def test_employer_template_reviews():
+def test_employer_template_reviews_module6():
     pass
 
 @pytest.mark.employer_template_review_stars
-def test_employer_template_review_stars():
+def test_employer_template_review_stars_module6():
     pass
 
 @pytest.mark.employer_template_review_details
-def test_employer_template_review_details():
+def test_employer_template_review_details_module6():
     pass
 
 @pytest.mark.app_employer_route
-def test_app_employer_route():
+def test_app_employer_route_module6():
     pass
 
 @pytest.mark.app_employer_route_employers
-def test_app_employer_route_employers():
+def test_app_employer_route_employers_module6():
     pass
 
 @pytest.mark.app_employer_route_jobs
-def test_app_employer_route_jobs():
+def test_app_employer_route_jobs_module6():
     pass
 
 @pytest.mark.app_employer_route_reviews
-def test_app_employer_route_reviews():
+def test_app_employer_route_reviews_module6():
     pass

tests/test_module7.py

@@ -4,17 +4,17 @@
 from jobs import app
 
 @pytest.mark.app_review_route
-def test_app_review_route():
+def test_app_review_route_module7():
     pass
 
 @pytest.mark.app_review_post_request_check
-def test_app_review_post_request_check():
+def test_app_review_post_request_check_module7():
     pass
 
 @pytest.mark.app_review_insert_review
-def test_app_review_insert_review():
+def test_app_review_insert_review_module7():
     pass
 
 @pytest.mark.review_form_cancel
-def test_review_form_cancel():
+def test_review_form_cancel_module7():
     pass

tests/utils.py

@@ -0,0 +1,58 @@
+import os
+import ast
+import inspect
+from pprint import pprint
+
+from bs4 import BeautifulSoup
+
+def get_decorators(source):
+    decorators = {}
+
+    def visit_FunctionDef(node):
+        decorators[node.name] = []
+        for n in node.decorator_list:
+            name = ''
+            if isinstance(n, ast.Call):
+                name = n.func.attr if isinstance(n.func, ast.Attribute) else n.func.id
+            else:
+                name = n.attr if isinstance(n, ast.Attribute) else n.id
+
+            args = [a.s for a in n.args] if hasattr(n, 'args') else []
+            decorators[node.name].append((name, args))
+
+    node_iter = ast.NodeVisitor()
+    node_iter.visit_FunctionDef = visit_FunctionDef
+    node_iter.visit(ast.parse(inspect.getsource(source)))
+    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]))
+
+    node_iter = ast.NodeVisitor()
+    node_iter.visit_Call = visit_Call
+    node_iter.visit(ast.parse(inspect.getsource(source)))
+
+    return functions
+
+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))
+    return rules
+
+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 tag_contains(name, tag, text):
+    return template_contains(name, tag).text if template_contains(name, tag) else False