test: ast to dict

Author: tjbell

jobs/app.py

@@ -0,0 +1,8 @@
+from flask import g
+
+DATABASE = 'db/jobs.sqlite'
+
+def get_db():
+    db = getattr(g, '_database', None)
+    if db is None:
+        db = g._database = sqlite.connect(DATABASE)

requirements.txt

@@ -1,4 +1,5 @@
+beautifulsoup4==4.6.3
 Flask==1.0.2
 pytest==3.7.1
 pytest-json-report==0.7.0
-python-dotenv==0.9.1
+python-dotenv==0.9.1

tag_names.txt

@@ -2,11 +2,9 @@
 import os 
 import inspect
 
-from pprint import pprint
 from .utils import *
 from jobs import app
 
-
 @pytest.mark.app_import_flask
 def test_app_import_flask_module1():
     assert 'Flask' in dir(app), 'Have you imported the `Flask` class from `flask`'

tests/test_module1.py

@@ -1,7 +1,6 @@
 import pytest
 import sys
 
-from jinja2 import Environment, PackageLoader, meta
 from jobs import app
 from .utils import *
 

tests/test_module2.py

@@ -3,7 +3,7 @@
 
 from jobs import app
 from .utils import *
-
+from pprint import pprint
 
 @pytest.mark.app_import_sqlite
 def test_app_import_sqlite_module3():
@@ -21,11 +21,13 @@ def test_app_db_path_module3():
 @pytest.mark.app_get_db_get_attribute
 def test_app_get_db_get_attribute_module3():
     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)
+    assert 'getattr:g:_database:None' in get_functions(app.get_db), 'Have used the `getattr` to get the global `_database`.'
 
 @pytest.mark.app_get_db_connection
 def test_app_get_db_connection_module3():
-    get_statements(app.get_db)
+    assert 'get_db' in dir(app), 'Have you defined a function named `get_db`.'
+    assert 'db:true:None' in get_statements(app.get_db), 'Have you created an `if` statement to test if `db` is `None`.'
+    print(source_dict(app.get_db))
     assert False
 
 @pytest.mark.app_get_db_row_factory

tests/test_module3.py

@@ -1,7 +1,6 @@
 import os
 import ast
 import inspect
-from pprint import pprint
 
 from jinja2 import nodes, Environment, PackageLoader, meta, exceptions
 from bs4 import BeautifulSoup
@@ -10,7 +9,6 @@
 
 def get_decorators(source):
     decorators = {}
-
     def visit_FunctionDef(node):
         decorators[node.name] = []
         for n in node.decorator_list:
@@ -26,37 +24,54 @@ def visit_FunctionDef(node):
     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
         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]))
+            if name is not 'connect':
+                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 = []
+    statements = []
 
     def visit_If(node):
-        print(ast.dump(node))
+        statement = node.test.left.id
+        if isinstance(node.test.ops[0], ast.Is) or isinstance(node.test.ops[0], ast.Eq):
+            statement += ':true'
+        statement += (':' + str(node.test.comparators[0].value))
+        statements.append(statement)
 
     node_iter = ast.NodeVisitor()
     node_iter.visit_If = visit_If
     node_iter.visit(ast.parse(inspect.getsource(source)))
-
-    return statment
+    return statements
+
+def source_dict(source):
+    return build_dict(ast.parse(inspect.getsource(source)))
+
+def build_dict(node):
+    result = {}
+    result['node_type'] = node.__class__.__name__
+    for attr in dir(node):
+        if not attr.startswith("_") and attr != 'ctx' and attr != 'lineno' and attr != 'col_offset':
+            value = getattr(node, attr)
+            if isinstance(value, ast.AST):
+                value = build_dict(value)
+            elif isinstance(value, list):
+                value = [build_dict(n) for n in value]
+            result[attr] = value
+    return result
 
 def list_routes(app):
     rules = []
@@ -97,3 +112,5 @@ def parsed_content(name):
 
 def template_extends(name):
     return list(meta.find_referenced_templates(parsed_content(name)))
+
+