"""Unit Tests for customer_master_master model."""

from abacus_legacy_sync.models.customer_master_master import CustomerMasterMaster

from tests.utils.factories import CustomerMasterMasterFactory


def test_create_customer_master_master():
    """Test creating a customer_master_master."""
    assert CustomerMasterMaster.count() == 0

    customer_name = 'eMusic'
    CustomerMasterMaster.create(customer_name=customer_name)
    result = CustomerMasterMaster.query.all()

    assert len(result) == 1
    assert result[0].customer_name == customer_name


def test_get_by_ids():
    """Test to get customer_master_master by ids."""
    customer_names = ['Audio Lunchbox', 'Download Punk']
    customer_master_masters = [
        CustomerMasterMasterFactory.create(customer_name=customer_name)
        for customer_name in customer_names
    ]
    ids = [
        customer_master_master.customer_master_master_id
        for customer_master_master in customer_master_masters
    ]

    result = CustomerMasterMaster.get_by_ids(ids)

    assert len(result) == 2
    assert result[0].customer_name == customer_names[0]
    assert result[1].customer_name == customer_names[1]