vertical-cooperative/easy_my_coop_api/tests/test_subscription_requests.py

# Copyright 2020 Coop IT Easy SCRL fs
#   Robin Keunen <robin@coopiteasy.be>
# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl.html).

import json
from datetime import timedelta

from werkzeug.exceptions import BadRequest

import odoo
from odoo.fields import Date

from odoo.addons.base_rest.controllers.main import _PseudoCollection
from odoo.addons.component.core import WorkContext

from .common import BaseEMCRestCase


class TestSRController(BaseEMCRestCase):
    def setUp(self):
        super().setUp()
        collection = _PseudoCollection("emc.services", self.env)
        emc_services_env = WorkContext(
            model_name="rest.service.registration", collection=collection
        )

        self.sr_service = emc_services_env.component(
            usage="subscription-request"
        )

    def test_service(self):
        # kept as example
        # useful if you need to change data in database and check db type

        result = self.sr_service.get(self.demo_request_1.external_id)
        self.assertEquals(self.demo_request_1_dict, result)

        all_sr = self.sr_service.search()
        self.assertTrue(all_sr)

        sr_date = self.demo_request_1.date
        date_from = Date.to_string(sr_date - timedelta(days=1))
        date_to = Date.to_string(sr_date + timedelta(days=1))

        date_sr = self.sr_service.search(date_from=date_from, date_to=date_to)
        self.assertTrue(date_sr)

    def test_route_get(self):
        id_ = self.demo_request_1.external_id
        route = "/api/subscription-request/%s" % id_
        content = self.http_get_content(route)
        self.assertEquals(self.demo_request_1_dict, content)

    @odoo.tools.mute_logger("odoo.addons.base_rest.http")
    def test_route_get_returns_not_found(self):
        route = "/api/subscription-request/%s" % "99999"
        response = self.http_get(route)
        self.assertEquals(response.status_code, 404)

    def test_route_get_string_returns_method_not_allowed(self):
        route = "/api/subscription-request/%s" % "abc"
        response = self.http_get(route)
        self.assertEquals(response.status_code, 405)

    def test_route_search_all(self):
        route = "/api/subscription-request"
        content = self.http_get_content(route)
        self.assertIn(self.demo_request_1_dict, content["rows"])

    def test_route_search_by_date(self):
        sr_date = self.demo_request_1.date
        date_from = Date.to_string(sr_date - timedelta(days=1))
        date_to = Date.to_string(sr_date + timedelta(days=1))

        route = "/api/subscription-request?date_from=%s" % date_from
        content = self.http_get_content(route)
        self.assertIn(self.demo_request_1_dict, content["rows"])

        route = "/api/subscription-request?date_to=%s" % date_to
        content = self.http_get_content(route)
        self.assertIn(self.demo_request_1_dict, content["rows"])

        route = "/api/subscription-request?date_from={}&date_to={}".format(
            date_from, date_to
        )
        content = self.http_get_content(route)
        self.assertIn(self.demo_request_1_dict, content["rows"])

        route = "/api/subscription-request?date_from=%s" % "2300-01-01"
        content = self.http_get_content(route)
        self.assertEquals(content["count"], 0)

        route = "/api/subscription-request?date_to=%s" % "1900-01-01"
        content = self.http_get_content(route)
        self.assertEquals(content["count"], 0)

    @odoo.tools.mute_logger("odoo.addons.base_rest.http")
    def test_route_search_acd_date_returns_bad_request(self):
        route = "/api/subscription-request?date_from=%s" % "20200101"
        response = self.http_get(route)
        self.assertEquals(response.status_code, 400)

    def test_route_create(self):
        url = "/api/subscription-request"
        data = {
            "name": "Lisa des Danses",
            "email": "lisa@desdanses.be",
            "ordered_parts": 3,
            "share_product": self.demo_share_product.id,
            "address": {
                "street": "schaerbeekstraat",
                "zip_code": "1111",
                "city": "Brussels",
                "country": "BE",
            },
            "lang": "en_US",
        }

        response = self.http_post(url, data=data)
        self.assertEquals(response.status_code, 200)
        content = json.loads(response.content.decode("utf-8"))

        content.pop("id")  # can't know id in advance
        expected = {
            **data,
            **{
                "date": Date.to_string(Date.today()),
                "state": "draft",
                "share_product": {
                    "id": self.demo_share_product.id,
                    "name": self.demo_share_product.name,
                },
                "capital_release_request": [],
            },
        }
        self.assertEquals(expected, content)

    def test_route_update(self):
        url = "/api/subscription-request/%s" % self.demo_request_1.external_id
        data = {"state": "done"}

        response = self.http_post(url, data=data)
        self.assertEquals(response.status_code, 200)
        content = json.loads(response.content.decode("utf-8"))

        expected = self.demo_request_1_dict
        expected["state"] = "done"
        self.assertEquals(expected, content)

    def test_route_validate(self):
        url = (
            "/api/subscription-request/%s/validate"
            % self.demo_request_1.external_id
        )
        response = self.http_post(url, data={})
        self.assertEquals(response.status_code, 200)
        content = json.loads(response.content.decode("utf-8"))

        state = content.get("state")
        self.assertEquals(state, "done")

    def test_service_validate_draft_request(self):
        self.sr_service.validate(self.demo_request_1.external_id)
        self.assertEquals(self.demo_request_1.state, "done")
        self.assertTrue(len(self.demo_request_1.capital_release_request) > 0)

    def test_service_validate_done_request(self):
        with self.assertRaises(BadRequest):
            self.sr_service.validate(self.demo_request_2.external_id)