62 lines
1.5 KiB
Python
62 lines
1.5 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
|
|
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'
|