import pytest
from fansifter_common.auth.account import Account

from dmp.shopify.handlers import GetStoresHandler, GetStoresRequest
from dmp.shopify.models import ShopifyStoreAssociation

from tests.unit.faker import FakerTyped
from tests.unit.types import CreateReportingModel


class TestGetStoresHandler:
    @pytest.mark.db
    def test_get_stores(
        self,
        handler: GetStoresHandler,
        create_reporting_model: CreateReportingModel,
        account: Account,
        fake: FakerTyped,
    ) -> None:
        store = create_reporting_model(
            ShopifyStoreAssociation,
            vendor_id=account.vendor_id,
            subaccount_id=account.subaccount_id,
        )

        stores = handler.handle(GetStoresRequest(identity_id=fake.uuid4_string()))

        assert stores
        assert stores[0].id == store.id