"""Tests for account_note model."""

from datetime import datetime

from product_review.api import db
from product_review.models import account_note as account_note_model
from tests.utils.db_utils import create_test_account_note


def test_get_items():
    """Test get_items."""
    create_test_account_note(vendor_id=1, note="abc", updated_by="user-123")
    create_test_account_note(vendor_id=2, note="def", updated_by="user-123")
    result = account_note_model.get_items()
    for item in result:
        assert isinstance(item.pop("updated_timestamp"), datetime)
    assert result == [
        {"vendor_id": 1, "note": "abc", "updated_by": "user-123"},
        {"vendor_id": 2, "note": "def", "updated_by": "user-123"},
    ]


def test_get_account_note():
    """Test get_account_note."""
    create_test_account_note(vendor_id=1, note="abc", updated_by="user-123")
    result = account_note_model.get_account_note(1)
    assert isinstance(result.pop("updated_timestamp"), datetime)
    assert result == {"vendor_id": 1, "note": "abc", "updated_by": "user-123"}


def test_get_account_note_none():
    """Test get_account_note None."""
    result = account_note_model.get_account_note(2)
    assert result is None


def test_create_account_note():
    """Test create_account_note."""
    account_note_model.create_account_note(1, "abc", "user-123")
    db.session.commit()
    item = account_note_model.get_account_note(1)
    assert isinstance(item.pop("updated_timestamp"), datetime)
    assert item == {"vendor_id": 1, "note": "abc", "updated_by": "user-123"}


def test_update_account_note():
    """Test update_account_note."""
    mock_obj = create_test_account_note(vendor_id=1, note="abc", updated_by="user-123")
    prev_updated_timestamp = mock_obj.updated_timestamp
    account_note_model.update_account_note(1, "def", "user-123")
    db.session.commit()
    item = account_note_model.get_account_note(1)
    assert item.pop("updated_timestamp") > prev_updated_timestamp
    assert item == {"vendor_id": 1, "note": "def", "updated_by": "user-123"}