67 lines
1.6 KiB
Python
67 lines
1.6 KiB
Python
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'
|