"""Test AbacusState."""

from lib.abacus_state import AbacusState


def test_abacus_state_create():
    """Test creating an abacus state."""
    assert AbacusState()


def test_abacus_state_create_with_data():
    """Test creating an abacus state with data populated."""
    mock_state = AbacusState(**{
        'abacus_state_id': 1,
        'action_name': 'foo',
        'action_status': 'baz',
        'message': 'some_message'
    })
    assert mock_state.abacus_state_id == 1
    assert mock_state.action_name == 'foo'
    assert mock_state.action_status == 'baz'