Sith/counter/tests/test_auto_actions.py

import json
from datetime import timedelta
from typing import TYPE_CHECKING

import pytest
from django.conf import settings
from django.urls import reverse
from django.utils.timezone import now
from django_celery_beat.models import ClockedSchedule
from model_bakery import baker

from core.models import Group, User
from counter.baker_recipes import counter_recipe, product_recipe
from counter.forms import (
    ProductForm,
    ScheduledProductActionForm,
    ScheduledProductActionFormSet,
)
from counter.models import Product, ProductType, ScheduledProductAction

if TYPE_CHECKING:
    from django.test import Client


@pytest.mark.django_db
def test_edit_product(client: Client):
    client.force_login(
        baker.make(
            User, groups=[Group.objects.get(id=settings.SITH_GROUP_COUNTER_ADMIN_ID)]
        )
    )
    product = product_recipe.make()
    url = reverse("counter:product_edit", kwargs={"product_id": product.id})
    res = client.get(url)
    assert res.status_code == 200

    res = client.post(url, data={})
    # This is actually a failure, but we just want to check that
    # we don't have a 403 or a 500.
    # The actual behaviour will be tested directly on the form.
    assert res.status_code == 200


@pytest.mark.django_db
def test_create_actions_alongside_product():
    """The form should work when the product and the actions are created alongside."""
    # non-persisted instance
    product: Product = product_recipe.prepare(_save_related=True)
    trigger_at = now() + timedelta(minutes=10)
    form = ProductForm(
        data={
            "name": "foo",
            "product_type": ProductType.objects.first(),
            "club": product.club_id,
            "code": "FOO",
            "purchase_price": 1.0,
            "selling_price": 1.0,
            "special_selling_price": 1.0,
            "limit_age": 0,
            "price-TOTAL_FORMS": "0",
            "price-INITIAL_FORMS": "0",
            "action-TOTAL_FORMS": "1",
            "action-INITIAL_FORMS": "0",
            "action-0-task": "counter.tasks.archive_product",
            "action-0-trigger_at": trigger_at,
        },
    )
    assert form.is_valid()
    product = form.save()
    action = ScheduledProductAction.objects.last()
    assert action.clocked.clocked_time == trigger_at
    assert action.enabled is True
    assert action.one_off is True
    assert action.task == "counter.tasks.archive_product"
    assert action.kwargs == json.dumps({"product_id": product.id})


@pytest.mark.django_db
class TestProductActionForm:
    def test_single_form_archive(self):
        product = product_recipe.make()
        trigger_at = now() + timedelta(minutes=10)
        form = ScheduledProductActionForm(
            product=product,
            data={
                "scheduled-task": "counter.tasks.archive_product",
                "scheduled-trigger_at": trigger_at,
            },
        )
        assert form.is_valid()
        instance = form.save()
        assert instance.clocked.clocked_time == trigger_at
        assert instance.enabled is True
        assert instance.one_off is True
        assert instance.task == "counter.tasks.archive_product"
        assert instance.kwargs == json.dumps({"product_id": product.id})

    def test_single_form_change_counters(self):
        product = product_recipe.make()
        counter = counter_recipe.make()
        trigger_at = now() + timedelta(minutes=10)
        form = ScheduledProductActionForm(
            product=product,
            data={
                "scheduled-task": "counter.tasks.change_counters",
                "scheduled-trigger_at": trigger_at,
                "scheduled-counters": [counter.id],
            },
        )
        assert form.is_valid()
        instance = form.save()
        instance.refresh_from_db()
        assert instance.clocked.clocked_time == trigger_at
        assert instance.enabled is True
        assert instance.one_off is True
        assert instance.task == "counter.tasks.change_counters"
        assert instance.kwargs == json.dumps(
            {"product_id": product.id, "counters": [counter.id]}
        )

    def test_delete(self):
        product = product_recipe.make()
        clocked = baker.make(ClockedSchedule, clocked_time=now() + timedelta(minutes=2))
        task = baker.make(
            ScheduledProductAction,
            product=product,
            one_off=True,
            clocked=clocked,
            task="counter.tasks.archive_product",
        )
        formset = ScheduledProductActionFormSet(product=product)
        formset.delete_existing(task)
        assert not ScheduledProductAction.objects.filter(id=task.id).exists()
        assert not ClockedSchedule.objects.filter(id=clocked.id).exists()


@pytest.mark.django_db
class TestProductActionFormSet:
    def test_ok(self):
        product = product_recipe.make()
        counter = counter_recipe.make()
        trigger_at = now() + timedelta(minutes=10)
        formset = ScheduledProductActionFormSet(
            product=product,
            data={
                "form-TOTAL_FORMS": "2",
                "form-INITIAL_FORMS": "0",
                "form-0-task": "counter.tasks.archive_product",
                "form-0-trigger_at": trigger_at,
                "form-1-task": "counter.tasks.change_counters",
                "form-1-trigger_at": trigger_at,
                "form-1-counters": [counter.id],
            },
        )
        assert formset.is_valid()
        formset.save()
        assert ScheduledProductAction.objects.filter(product=product).count() == 2