import amazon


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

    assert len(res) == 1
    email, pwd = res[0]
    assert email == 'test@example.com'
    assert pwd == '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
    email, pwd = res[0]
    assert email == 'test@example.com'
    assert pwd == '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

    email1, pwd1 = res[0]
    assert email1 == 'test@example.com'
    assert pwd1 == 'password123'

    email2, pwd2 = res[1]
    assert email2 == 'test2@example.com'
    assert pwd2 == '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
    email1, pwd1 = res[0]
    assert email1 == 'test@example.com'
    assert pwd1 == 'password123'

    email2, pwd2 = res[1]
    assert email2 == 'test2@example.com'
    assert pwd2 == 'password456'