"""Tests for canned_response_category model.""" from product_review.api import db from product_review.models import ( canned_response_category as canned_response_category_model, ) from tests.utils.db_utils import create_test_canned_response_category def test_get_items(): """Test get_items.""" create_test_canned_response_category(name="some cat 1") create_test_canned_response_category(name="some cat 2") result = canned_response_category_model.get_items() assert result == [ {"category_id": 1, "name": "some cat 1"}, {"category_id": 2, "name": "some cat 2"}, ] def test_create_canned_response_category(): """Test create_canned_response_category.""" result = canned_response_category_model.create_category("some cat 1") db.session.commit() assert result.category_id == 1 assert result.name == "some cat 1" def test_create_canned_response_category_exists(): """Test create_canned_response_category category exists.""" create_test_canned_response_category(name="some cat 1") result = canned_response_category_model.create_category("some cat 1") assert result.category_id == 1 assert result.name == "some cat 1"