Merge pull request #6 from Lend-it/93_visualizar_categorias

Solve visualizar categorias #93

project/__init__.py

@@ -2,6 +2,7 @@
 from flask import Flask, jsonify
 from flask_sqlalchemy import SQLAlchemy
 from database_singleton import Singleton
+from project.api.views import category_blueprint
 from project.api.views import request_blueprint
 
 # instantiate the app
@@ -19,6 +20,7 @@ def create_app(script_info=None):
     db.init_app(app)
     migrate.init_app(app, db)
 
+    app.register_blueprint(category_blueprint)
     app.register_blueprint(request_blueprint)
 
     @app.shell_context_processor

project/api/models.py

@@ -17,6 +17,12 @@ class Category(db.Model):
     def __init__(self, name):
         self.name = name
 
+    def to_json(self):
+        return {
+            "productcategoryid": self.productcategoryid,
+            "name": self.name,
+        }
+
 
 class Request(db.Model):
     __tablename__ = "request"

project/api/views.py

@@ -3,11 +3,46 @@
 from sqlalchemy import exc
 
 from project.api.models import Request
+from project.api.models import Category
 from database_singleton import Singleton
 
+db = Singleton().database_connection()
+
+category_blueprint = Blueprint("categories", __name__)
 request_blueprint = Blueprint("requests", __name__)
 
-db = Singleton().database_connection()
+
+@category_blueprint.route("/product_category", methods=["GET"])
+def get_all_request():
+    response = {
+        "status": "success",
+        "data": {
+            "categories": [category.to_json() for category in Category.query.all()]
+        },
+    }
+    return jsonify(response), 200
+
+
+@category_blueprint.route("/product_category", methods=["POST"])
+def add_categories():
+    if not post_data:
+        return jsonify(error_response), 400
+
+    name = post_data.get("name")
+
+    try:
+        db.session.add(Category(name))
+        db.session.commit()
+
+        response = {
+            "status": "success",
+            "data": {"message": f"Category {name} was created!"},
+        }
+
+        return jsonify(response), 201
+    except exc.IntegrityError:
+        db.session.rollback()
+        return jsonify(error_response), 400
 
 
 @request_blueprint.route("/requests", methods=["POST"])

project/tests/test_product_category.py

@@ -0,0 +1,34 @@
+import json
+import unittest
+
+from project.tests.base import BaseTestCase
+from database_singleton import Singleton
+
+from project.api.models import Category
+
+db = Singleton().database_connection()
+
+
+def add_category(name):
+    category = Category(name)
+    db.session.add(category)
+    db.session.commit()
+    return category
+
+
+class TestProductCategory(BaseTestCase):
+    def test_get_all_category(self):
+        add_category("Eletrodomésticos")
+        add_category("Livros e revistas")
+        add_category("Eletrônicos")
+
+        with self.client:
+            response = self.client.get("/product_category")
+            data = json.loads(response.data.decode())
+
+            self.assertEqual(response.status_code, 200)
+            self.assertIn("success", data["status"])
+
+            self.assertIn("Eletrodomésticos", data["data"]["categories"][0]["name"])
+            self.assertIn("Livros e revistas", data["data"]["categories"][1]["name"])
+            self.assertIn("Eletrônicos", data["data"]["categories"][2]["name"])