Refactor codebase to use solution of #25

Author: EliuX

cli.py

@@ -6,9 +6,11 @@
 from flask_script import Manager
 
 from time_tracker_api import create_app
-from time_tracker_api.api import api
 
 app = create_app()
+
+from time_tracker_api.api import api
+
 cli_manager = Manager(app)
 
 
@@ -44,6 +46,23 @@ def gen_postman_collection(filename='timetracker-api-postman-collection.json',
     save_data(parsed_json, filename)
 
 
+@cli_manager.command
+def seed():
+    from time_tracker_api.database import seeder as seed
+    seed()
+
+
+@cli_manager.command
+def re_create_db():
+    print('This command will drop all tables and seed again the database')
+    confirm_answer = input('Do you confirm (Y) you want to remove all your data?\n')
+    if confirm_answer.upper() == 'Y':
+        from time_tracker_api.database import seeder
+        seeder.fresh()
+    else:
+        print('\nThe process was cancelled!')
+
+
 def save_data(data: str, filename: str) -> None:
     """ Save text content to a file """
     if filename:

requirements/dev.txt

@@ -12,7 +12,4 @@ pytest-mock==2.0.0
 Faker==4.0.2
 
 # Coverage
-coverage==4.5.1
-
-# SQL database (MS SQL)
-flask_sqlalchemy==2.4.1
+coverage==4.5.1

requirements/prod.txt

@@ -13,7 +13,13 @@ Jinja2==2.11.1
 gunicorn==20.0.4
 
 #Swagger support for Restful API
-flask-restplus==0.13.0
+flask-restplus==0.12.1
 
 #CLI support
-Flask-Script==2.0.6
+Flask-Script==2.0.6
+
+# SQL database (MS SQL)
+flask_sqlalchemy==2.4.1
+
+# Handling requests
+requests==2.23.0

tests/conftest.py

@@ -24,14 +24,14 @@ def client(app: Flask) -> FlaskClient:
 @pytest.fixture(scope="module")
 def sql_repository(app: Flask):
     with app.app_context():
-        from .resources import TestModel
+        from .resources import SQLPersonModel
+        from time_tracker_api.database import seeder
         from time_tracker_api.sql_repository import db
 
-        db.create_all()
-        print("Models for test created!")
+        seeder.fresh()
 
         from time_tracker_api.sql_repository import SQLRepository
-        yield SQLRepository(TestModel)
+        yield SQLRepository(SQLPersonModel)
 
-        print("Models for test removed!")
         db.drop_all()
+        print("Models for test removed!")

tests/projects/projects_namespace_test.py

@@ -1,11 +1,17 @@
 from flask import json
+from flask.testing import FlaskClient
+from pytest_mock import MockFixture
 
 
-def test_list_should_return_nothing(client):
-    """Should return an empty array"""
+def test_list_all_elements(client: FlaskClient, mocker: MockFixture):
+    """Should return all elements in a list"""
+    from time_tracker_api.projects.projects_namespace import project_dao
+    repository_find_all_mock = mocker.patch.object(project_dao.repository, 'find_all', return_value=[])
+
     response = client.get("/projects", follow_redirects=True)
 
     assert 200 == response.status_code
 
     json_data = json.loads(response.data)
     assert [] == json_data
+    repository_find_all_mock.assert_called_once()

tests/resources.py

@@ -1,9 +1,11 @@
-from time_tracker_api.sql_repository import db
+from time_tracker_api.database import AuditedModel
+from time_tracker_api.sql_repository import db, SQLAuditedModel
 
 
-class TestModel(db.Model):
+class SQLPersonModel(db.Model, SQLAuditedModel):
+    __tablename__ = 'tests'
     id = db.Column(db.Integer, primary_key=True)
-    name = db.Column(db.String(80), unique=True, nullable=False)
+    name = db.Column(db.String(80), unique=False, nullable=False)
     email = db.Column(db.String(120), unique=True, nullable=False)
     age = db.Column(db.Integer, nullable=False)
 

tests/smoke_test.py

@@ -1,4 +1,3 @@
-
 def test_app_exists(app):
     """Does app exists"""
     assert app is not None

tests/sql_repository_test.py

@@ -8,16 +8,19 @@
 
 def test_create(sql_repository):
     """Should create a new Entry"""
-    from .resources import TestModel
     global sample_element
-    sample_element = TestModel(name=fake.name(),
-                               email=fake.safe_email(),
-                               age=fake.pyint(min_value=10, max_value=80))
+    sample_element = dict(name=fake.name(),
+                          email=fake.safe_email(),
+                          age=fake.pyint(min_value=10, max_value=80))
 
     result = sql_repository.create(sample_element)
 
     assert result is not None
     assert result.id is not None
+    assert result.created_at is not None
+    assert result.created_by is not None
+    assert result.updated_at is None
+    assert result.updated_by is None
 
     existing_elements_registry.append(result)
 
@@ -43,6 +46,9 @@ def test_update(sql_repository):
     assert updated_element.id == existing_element.id
     assert updated_element.name == "Jon Snow"
     assert updated_element.age == 34
+    assert updated_element.updated_at is not None
+    assert updated_element.updated_at > updated_element.created_at
+    assert updated_element.updated_by is not None
 
 
 def test_find_all(sql_repository):
@@ -54,10 +60,10 @@ def test_find_all(sql_repository):
 
 def test_find_all_that_contains_property_with_string(sql_repository):
     """Find all elements that have a property that partially contains a string (case-insensitive)"""
-    from .resources import TestModel
-    new_element = TestModel(name='Ramsay Snow',
-                            email=fake.safe_email(),
-                            age=fake.pyint(min_value=10, max_value=80))
+    fake_name = fake.name()
+    new_element = dict(name="%s Snow" % fake_name,
+                       email=fake.safe_email(),
+                       age=fake.pyint(min_value=10, max_value=80))
     sql_repository.create(new_element)
     existing_elements_registry.append(new_element)
 
@@ -67,8 +73,8 @@ def test_find_all_that_contains_property_with_string(sql_repository):
     search_jon_result = sql_repository.find_all_contain_str('name', 'Jon')
     assert len(search_jon_result) == 1
 
-    search_ram_result = sql_repository.find_all_contain_str('name', 'RAM')
-    assert search_ram_result[0] is new_element
+    search_ram_result = sql_repository.find_all_contain_str('name', fake_name)
+    assert search_ram_result[0].name == new_element['name']
 
 
 def test_delete_existing_element(sql_repository):