Merge pull request #18 from Kalgoc/gastos

Gastos

Author: anibalcontreras

expenses/__init__.py

@@ -0,0 +1,25 @@
+from django.contrib import admin
+from .models import Expense
+
+
+@admin.register(Expense)
+class ExpenseAdmin(admin.ModelAdmin):
+    list_display = (
+        "username",
+        "amount",
+        "user_expense_type_name",
+        "category_name",
+        "bankcard_id",
+        "created_at",
+        "updated_at",
+    )
+
+    def user_expense_type_name(self, obj):
+        return obj.user_expense_type.name
+
+    user_expense_type_name.short_description = "User Expense Type"
+
+    def category_name(self, obj):
+        return obj.category.name
+
+    category_name.short_description = "Category"

expenses/admin.py

@@ -0,0 +1,6 @@
+from django.apps import AppConfig
+
+
+class ExpensesConfig(AppConfig):
+    default_auto_field = "django.db.models.BigAutoField"
+    name = "expenses"

expenses/apps.py

@@ -0,0 +1,35 @@
+# Generated by Django 5.0.6 on 2024-06-17 13:41
+
+import django.db.models.deletion
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    initial = True
+
+    dependencies = [
+        ("categories", "0001_initial"),
+        ("user_expense_type", "0001_initial"),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name="Expense",
+            fields=[
+                ("id", models.AutoField(primary_key=True, serialize=False)),
+                ("username", models.UUIDField()),
+                ("bankcard_id", models.IntegerField()),
+                ("amount", models.IntegerField()),
+                ("created_at", models.DateTimeField(auto_now_add=True)),
+                ("updated_at", models.DateTimeField(auto_now=True)),
+                ("category", models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to="categories.category")),
+                (
+                    "user_expense_type",
+                    models.ForeignKey(
+                        on_delete=django.db.models.deletion.CASCADE, to="user_expense_type.userexpensetype"
+                    ),
+                ),
+            ],
+        ),
+    ]

expenses/migrations/0001_initial.py

@@ -0,0 +1,15 @@
+from django.db import models
+from user_expense_type.models import UserExpenseType
+from categories.models import Category
+
+
+class Expense(models.Model):
+    id = models.AutoField(primary_key=True)
+    username = models.UUIDField()
+    user_expense_type = models.ForeignKey(UserExpenseType, on_delete=models.CASCADE)
+    category = models.ForeignKey(Category, on_delete=models.CASCADE)
+    # cambiar cuando exista el modelo de bankcard
+    bankcard_id = models.IntegerField()
+    amount = models.IntegerField()
+    created_at = models.DateTimeField(auto_now_add=True)
+    updated_at = models.DateTimeField(auto_now=True)

expenses/migrations/__init__.py

@@ -0,0 +1,22 @@
+from rest_framework import serializers
+from .models import Expense
+from user_expense_type.models import UserExpenseType
+from categories.models import Category
+
+
+class ExpenseSerializer(serializers.Serializer):
+    id = serializers.IntegerField(read_only=True)
+
+    username = serializers.UUIDField()
+    user_expense_type = serializers.PrimaryKeyRelatedField(queryset=UserExpenseType.objects.all())
+    category = serializers.PrimaryKeyRelatedField(queryset=Category.objects.all())
+    bankcard_id = serializers.IntegerField()
+    amount = serializers.IntegerField()
+
+    def create(self, validated_data):
+        return Expense.objects.create(**validated_data)
+
+    def update(self, instance, validated_data):
+        instance.amount = validated_data.get("amount", instance.amount)
+        instance.save()
+        return instance

expenses/models.py

@@ -0,0 +1,135 @@
+from django.test import TestCase
+from unittest.mock import patch
+from rest_framework.test import APIClient
+from rest_framework import status
+from authentication.services.cognito_service import CognitoService
+from rest_framework.response import Response
+from .views import ExpenseViewSet, ExpenseGroupedByTypeAndCategoryViewSet
+
+
+class ExpenseViewSetTestCase(TestCase):
+    def setUp(self):
+        self.client = APIClient()
+        self.view = ExpenseViewSet()
+        self.expense = {
+            "user_expense_type": 1,
+            "category": 1,
+            "bankcard_id": 1,
+            "amount": 100,
+        }
+
+    @patch.object(ExpenseViewSet, "create")
+    @patch.object(CognitoService, "login_user")
+    def test_create(self, mock_login_user, mock_create):
+        mock_login_user.return_value = {
+            "AuthenticationResult": {"AccessToken": "mock_access_token", "IdToken": "mock_id_token"}
+        }
+        mock_create.return_value = Response(status=status.HTTP_201_CREATED, data=self.expense)
+
+        request = self.client.post("/expenses/", self.expense)
+        request.headers["Authorization"] = "Bearer mock_access_token"
+
+        response = self.client.post("/expenses/", self.expense)
+
+        self.assertEqual(response.status_code, status.HTTP_201_CREATED)
+        self.assertEqual(response.data["amount"], 100)
+
+    @patch.object(ExpenseViewSet, "list")
+    @patch.object(CognitoService, "login_user")
+    def test_list(self, mock_login_user, mock_list):
+        mock_login_user.return_value = {
+            "AuthenticationResult": {"AccessToken": "mock_access_token", "IdToken": "mock_id_token"}
+        }
+        mock_list.return_value = Response(status=status.HTTP_200_OK, data=self.expense)
+
+        request = self.client.get("/expenses/")
+        request.headers["Authorization"] = "Bearer mock_access_token"
+
+        response = self.client.get("/expenses/")
+
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        self.assertEqual(response.data["amount"], 100)
+
+    @patch.object(ExpenseViewSet, "retrieve")
+    @patch.object(CognitoService, "login_user")
+    def test_retrieve(self, mock_login_user, mock_retrieve):
+        mock_login_user.return_value = {
+            "AuthenticationResult": {"AccessToken": "mock_access_token", "IdToken": "mock_id_token"}
+        }
+        mock_retrieve.return_value = Response(status=status.HTTP_200_OK, data=self.expense)
+
+        request = self.client.get("/expenses/1/")
+        request.headers["Authorization"] = "Bearer mock_access_token"
+
+        response = self.client.get("/expenses/1/")
+
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        self.assertEqual(response.data["amount"], 100)
+
+    @patch.object(ExpenseViewSet, "destroy")
+    @patch.object(CognitoService, "login_user")
+    def test_destroy(self, mock_login_user, mock_destroy):
+        mock_login_user.return_value = {
+            "AuthenticationResult": {"AccessToken": "mock_access_token", "IdToken": "mock_id_token"}
+        }
+        mock_destroy.return_value = Response(status=status.HTTP_204_NO_CONTENT)
+
+        request = self.client.delete("/expenses/1/")
+        request.headers["Authorization"] = "Bearer mock_access_token"
+
+        response = self.client.delete("/expenses/1/")
+
+        self.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT)
+
+    @patch.object(ExpenseViewSet, "partial_update")
+    @patch.object(CognitoService, "login_user")
+    def test_partial_update(self, mock_login_user, mock_partial_update):
+        mock_login_user.return_value = {
+            "AuthenticationResult": {"AccessToken": "mock_access_token", "IdToken": "mock_id_token"}
+        }
+        new_expense = {
+            "expense_type": 1,
+            "category": 1,
+            "bankcard_id": 1,
+            "amount": 200,
+        }
+        mock_partial_update.return_value = Response(status=status.HTTP_200_OK, data=new_expense)
+
+        request = self.client.put("/expenses/1/", {"amount": 200})
+        request.headers["Authorization"] = "Bearer mock_access_token"
+
+        response = self.client.put("/expenses/1/", {"amount": 200})
+
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        self.assertEqual(response.data["amount"], 200)
+
+
+class ExpenseGroupedByTypeAndCategoryViewSetTestCase(TestCase):
+    def setUp(self):
+        self.client = APIClient()
+        self.view = ExpenseGroupedByTypeAndCategoryViewSet()
+        self.expense = {
+            "expense_type": 1,
+            "category": 1,
+            "bankcard_id": 1,
+            "amount": 100,
+        }
+        self.expenses_grouped = {"Personal": {"Comida": 100}}
+
+    @patch.object(ExpenseGroupedByTypeAndCategoryViewSet, "list")
+    @patch.object(CognitoService, "login_user")
+    def test_list(self, mock_login_user, mock_list):
+        mock_login_user.return_value = {
+            "AuthenticationResult": {"AccessToken": "mock_access_token", "IdToken": "mock_id_token"}
+        }
+        mock_list.return_value = Response(status=status.HTTP_200_OK, data=self.expenses_grouped)
+
+        request = self.client.get("/expenses/grouped/")
+        request.headers["Authorization"] = "Bearer mock_access_token"
+
+        response = self.client.get("/expenses/grouped/")
+
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        self.assertIn("Personal", response.data)
+        self.assertIn("Comida", response.data["Personal"])
+        self.assertEqual(response.data["Personal"]["Comida"], 100)

expenses/serializers.py

@@ -0,0 +1,20 @@
+from django.urls import path
+from .views import ExpenseViewSet, ExpenseGroupedByTypeAndCategoryViewSet
+
+urlpatterns = [
+    path(
+        "",
+        ExpenseViewSet.as_view({"get": "list", "post": "create"}),
+        name="expenses",
+    ),
+    path(
+        "<int:pk>/",
+        ExpenseViewSet.as_view({"get": "retrieve", "put": "partial_update", "delete": "destroy"}),
+        name="expense-detail",
+    ),
+    path(
+        "grouped/",
+        ExpenseGroupedByTypeAndCategoryViewSet.as_view({"get": "list"}),
+        name="expenses_grouped",
+    ),
+]