import amazon


def test_get_accounts():
    env = {
        'AMAZON_EMAIL': 'test@example.com',
        'AMAZON_PASSWORD': 'password123',
    }
    res = amazon.get_accounts(env)

    assert len(res) == 1
    assert res[0].email == 'test@example.com'
    assert res[0].password == 'password123'


def test_get_accounts_numbered():
    env = {
        'AMAZON_EMAIL_0': 'test@example.com',
        'AMAZON_PASSWORD_0': 'password123',
    }
    res = amazon.get_accounts(env)

    assert len(res) == 1
    assert res[0].email == 'test@example.com'
    assert res[0].password == 'password123'


def test_get_accounts_numbered_multi():
    env = {
        'AMAZON_EMAIL_0': 'test@example.com',
        'AMAZON_PASSWORD_0': 'password123',
        'AMAZON_EMAIL_1': 'test2@example.com',
        'AMAZON_PASSWORD_1': 'password456',
    }
    res = amazon.get_accounts(env)

    assert len(res) == 2

    assert res[0].email == 'test@example.com'
    assert res[0].password == 'password123'

    assert res[1].email == 'test2@example.com'
    assert res[1].password == 'password456'


def test_get_accounts_both():
    env = {
        'AMAZON_EMAIL': 'test@example.com',
        'AMAZON_PASSWORD': 'password123',
        'AMAZON_EMAIL_0': 'test2@example.com',
        'AMAZON_PASSWORD_0': 'password456',
    }
    res = amazon.get_accounts(env)

    assert len(res) == 2

    assert res[0].email == 'test@example.com'
    assert res[0].password == 'password123'

    assert res[1].email == 'test2@example.com'
    assert res[1].password == 'password456'