"""Test the store object."""

import pytest

from abacus_common_data.store import Store


def test_store_create_uma():
    """Test UMA store."""
    uma = Store(666)
    assert uma.store_id == 666
    assert uma.store_name == 'UMA'


def test_store_unknown_id():
    """Tests passing an unknown id."""
    with pytest.raises(KeyError):
        Store(9999999999)


def test_list_all_stores():
    """Test list_all_stores."""
    stores = Store.list_all_stores()
    store = stores[0]
    assert store.get('store_id')
    assert store.get('store_name')