initial commit

tests/test_amazon.py

@@ -0,0 +1,66 @@
+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'