"""Test SME analytics ingestion preprocessing."""
from copy import deepcopy

from switchboard_consumer.logic.preprocessing.sme_analytics_ingestion \
    import update_label_id

SWB_PRODUCT_EXAMPLE = {
    'some_data_from_sme_response': 98765,
    'someArtistName': 'Artist Name',
    'someFlag': True,
    'project': {},
}


def test_update_without_sme_analytics_flag(create_product_message):
    """Test ordinary Kafka message."""
    product = deepcopy(SWB_PRODUCT_EXAMPLE)
    expected = deepcopy(product)
    update_label_id(create_product_message, product)
    assert expected == product


def test_update_label_id(create_sme_enricher_message):
    """Test SME Enricher Kafka message."""
    product = deepcopy(SWB_PRODUCT_EXAMPLE)
    update_label_id(create_sme_enricher_message, product)
    assert 'labelAccount' in product
    assert 'subAccount' in product
    label_id = create_sme_enricher_message.message['labelAccountId']
    subaccount_id = create_sme_enricher_message.message['subAccountId']
    assert product['labelAccount']['companyCode'] == label_id
    assert product['subAccount']['companyCode'] == subaccount_id
    assert product['project']['labelAccount']['companyCode'] == label_id
    assert product['project']['subAccount']['companyCode'] == subaccount_id