python-vaultwarden/tests/e2e/test_bitwarden.py at d28113dd4389ea7df8b0e5ddd2efce849bb36636 · numberly/python-vaultwarden · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
import os
import unittest

from vaultwarden.clients.bitwarden import BitwardenAPIClient
from vaultwarden.models.bitwarden import get_organization

from .docker_helper import start_docker, stop_docker

# Get Bitwarden credentials from environment variables
url = os.environ.get("BITWARDEN_URL", None)
email = os.environ.get("BITWARDEN_EMAIL", None)
password = os.environ.get("BITWARDEN_PASSWORD", None)
client_id = os.environ.get("BITWARDEN_CLIENT_ID", None)
client_secret = os.environ.get("BITWARDEN_CLIENT_SECRET", None)
device_id = os.environ.get("BITWARDEN_DEVICE_ID", None)

# Get test organization id from environment variables
test_organization = os.environ.get("BITWARDEN_TEST_ORGANIZATION", None)


class BitwardenBasic(unittest.TestCase):
    def tearDownClass() -> None:
        stop_docker()

    def setUp(self) -> None:
        start_docker()
        self.bitwarden = BitwardenAPIClient(
            url, email, password, client_id, client_secret, device_id
        )
        self.organization = get_organization(self.bitwarden, test_organization)
        self.test_colls_names = self.organization.collections(as_dict=True)
        self.test_colls_ids = self.organization.collections()
        self.test_users = self.organization.users()
        self.test_org_ciphers = self.organization.ciphers()
        self.test_collection_1_ciphers = self.organization.ciphers(
            self.test_colls_names.get("test-collection").Id
        )
        self.test_collection_2_ciphers = self.organization.ciphers(
            self.test_colls_names.get("test-collection-2").Id
        )
        self.test_collection_1_users = self.test_colls_names.get(
            "test-collection"
        ).users()
        self.test_collection_2_users = self.test_colls_names.get(
            "test-collection-2"
        ).users()

    def tearDown(self) -> None:
        stop_docker()

    def test_get_organization_users(self):
        self.assertEqual(len(self.test_users), 2)

    def test_get_organization_items(self):
        self.assertEqual(len(self.test_org_ciphers), 1)

    def test_get_organization_collection_1_item(self):
        self.assertEqual(len(self.test_collection_1_ciphers), 0)

    def test_get_organization_collection_2_items(self):
        self.assertEqual(len(self.test_collection_2_ciphers), 1)

    def test_get_organizations_collections(self):
        # 2 test collections + default collection
        self.assertEqual(len(self.test_colls_ids), 3)

    def test_get_users_of_collection_1(self):
        # 2 test collections + default collection
        self.assertEqual(len(self.test_collection_1_users), 0)

    def test_get_users_of_collection_2(self):
        self.assertEqual(len(self.test_collection_2_users), 1)

    def test_create_delete_collection(self):
        len_old_colls = len(self.organization.collections(force_refresh=True))
        new_coll = self.organization.create_collection("create_delete_test")
        new_colls = self.organization.collections(force_refresh=True)
        self.assertEqual(len(new_colls), len_old_colls + 1)
        self.organization.delete_collection(new_coll.Id)
        new_colls = self.organization.collections(force_refresh=True)
        self.assertEqual(len(new_colls), len_old_colls)

    def test_set_users_of_collection(self):
        coll = self.test_colls_names.get("test-collection")
        coll.set_users(self.test_collection_2_users)
        users = coll.users()
        self.assertEqual(len(users), 1)
        coll.set_users(self.test_collection_1_users)
        users = coll.users()
        self.assertEqual(len(users), 0)

    def test_add_remove_collection_from_user(self):
        user_org_id = self.test_collection_2_users[0].UserId
        user_infos = self.organization.user(user_org_id)
        collection_2 = self.test_colls_names.get("test-collection-2")
        user_infos.remove_collections([collection_2.Id])
        self.assertEqual(
            len(collection_2.users()),
            0,
        )
        user_infos.add_collections([collection_2.Id])
        users = collection_2.users()
        self.assertEqual(
            len(users),
            1,
        )

    def test_invite_user_than_remove(self):
        resp = self.organization.invite("test-user-3@example.com")
        self.assertTrue(resp.is_success)
        user = self.organization.user_search(
            "test-user-3@example.com", force_refresh=True
        )
        self.assertIsNotNone(user)
        user.delete()

    def test_add_remove_collection_cipher(self):
        cipher = self.test_org_ciphers[0]
        old_colls = cipher.CollectionIds
        collection_1 = self.test_colls_ids[0]
        cipher.add_collections(collections=[collection_1.Id])
        res = self.organization.ciphers(force_refresh=True)
        self.assertEqual(len(res), 1)
        self.assertEqual(len(res[0].CollectionIds), 2)
        cipher.update_collection(old_colls)

    def test_deduplicate(self):
        # Todo build test fixtures and delete them at the end of the test
        return


if __name__ == "__main__":
    unittest.main()