|
1 | 1 | from posit import connect |
2 | 2 |
|
3 | 3 |
|
4 | | -class TestAttributeContent: |
| 4 | +class TestUser: |
| 5 | + @classmethod |
| 6 | + def setup_class(cls): |
| 7 | + cls.client = client = connect.Client() |
| 8 | + cls.aron = client.users.create( |
| 9 | + username="aron", email="[email protected]", password="s3cur3p@ssword" |
| 10 | + ) |
| 11 | + cls.bill = client.users.create( |
| 12 | + username="bill", email="[email protected]", password="s3cur3p@ssword" |
| 13 | + ) |
| 14 | + cls.cole = client.users.create( |
| 15 | + username="cole", email="[email protected]", password="s3cur3p@ssword" |
| 16 | + ) |
| 17 | + |
| 18 | + def test_lock(self): |
| 19 | + for user in (self.aron, self.bill, self.cole): |
| 20 | + user.lock() |
| 21 | + assert len(self.client.users.find(account_status="locked")) == 1 |
| 22 | + user.unlock() |
| 23 | + assert len(self.client.users.find(account_status="locked")) == 0 |
| 24 | + |
| 25 | + def test_count(self): |
| 26 | + # aron, bill, cole, and me |
| 27 | + assert self.client.users.count() == 4 |
| 28 | + |
| 29 | + def test_find(self): |
| 30 | + assert self.client.users.find(prefix="aron") == [self.aron] |
| 31 | + assert self.client.users.find(prefix="bill") == [self.bill] |
| 32 | + assert self.client.users.find(prefix="cole") == [self.cole] |
| 33 | + |
| 34 | + def test_find_one(self): |
| 35 | + assert self.client.users.find_one(prefix="aron") == self.aron |
| 36 | + assert self.client.users.find_one(prefix="bill") == self.bill |
| 37 | + assert self.client.users.find_one(prefix="cole") == self.cole |
| 38 | + |
| 39 | + def test_get(self): |
| 40 | + assert self.client.users.get(self.aron["guid"]) == self.aron |
| 41 | + assert self.client.users.get(self.bill["guid"]) == self.bill |
| 42 | + assert self.client.users.get(self.cole["guid"]) == self.cole |
| 43 | + |
| 44 | + |
| 45 | +class TestUserContent: |
5 | 46 | """Checks behavior of the content attribute.""" |
6 | 47 |
|
7 | 48 | @classmethod |
8 | 49 | def setup_class(cls): |
9 | | - cls.client = connect.Client() |
10 | | - cls.user = cls.client.me |
11 | | - cls.user.content.create( |
| 50 | + cls.client = client = connect.Client() |
| 51 | + cls.me = me = client.me |
| 52 | + cls.content = me.content.create( |
12 | 53 | name="Sample", |
13 | 54 | description="Simple sample content for testing", |
14 | 55 | access_type="acl", |
15 | 56 | ) |
16 | 57 |
|
17 | 58 | @classmethod |
18 | 59 | def teardown_class(cls): |
19 | | - assert cls.user.content.find_one().delete() is None |
20 | | - assert cls.user.content.count() == 0 |
| 60 | + assert cls.content.delete() is None |
| 61 | + assert cls.me.content.count() == 0 |
21 | 62 |
|
22 | 63 | def test_count(self): |
23 | | - assert self.user.content.count() == 1 |
| 64 | + assert self.me.content.count() == 1 |
24 | 65 |
|
25 | 66 | def test_find(self): |
26 | | - assert self.user.content.find() |
| 67 | + assert self.me.content.find() |
27 | 68 |
|
28 | 69 | def test_find_one(self): |
29 | | - assert self.user.content.find_one() |
| 70 | + assert self.me.content.find_one() |
| 71 | + |
| 72 | + def test_multiple_users(self): |
| 73 | + user = self.client.users.create( |
| 74 | + username="example", email="[email protected]", password="s3cur3p@ssword" |
| 75 | + ) |
| 76 | + # assert filtering limits to the provided user. |
| 77 | + assert self.me.content.find_one() == self.content |
| 78 | + assert user.content.find_one() is None |
| 79 | + user.lock() |
0 commit comments