"""Tests for target_group model."""

from product_review.models import target_group as target_group_model
from tests.utils.db_utils import create_test_target_group


def test_get_items_found(app_context):
    """Test get_items found."""
    create_test_target_group(target_group_id=1, display_name="hello world")
    create_test_target_group(target_group_id=2, display_name="abcd")
    target_groups = target_group_model.get_items()
    assert target_groups == [
        {"target_group_id": 1, "display_name": "hello world"},
        {"target_group_id": 2, "display_name": "abcd"},
    ]


def test_get_items_not_found(app_context):
    """Test get_items not found."""
    target_groups = target_group_model.get_items()
    assert target_groups == []