|
2 | 2 |
|
3 | 3 | import pytest |
4 | 4 |
|
5 | | -from alive_progress.core.progress import _GatedAssignFunction, _GatedFunction, _render_title |
| 5 | +from alive_progress.core.progress import _AssignFunction, _Function, _GatedFunction, _render_title |
6 | 6 | from alive_progress.utils.cells import join_cells |
7 | 7 |
|
8 | 8 |
|
@@ -32,23 +32,42 @@ def test_render_title(length, text, expected): |
32 | 32 |
|
33 | 33 | def test_gated_properties(): |
34 | 34 | class AClass: |
| 35 | + always = _Function() |
35 | 36 | readonly = _GatedFunction() |
36 | | - assignable = _GatedAssignFunction() |
| 37 | + assignable = _AssignFunction() |
37 | 38 |
|
38 | 39 | instance, m = AClass(), mock.Mock() |
39 | | - instance._handle = True |
| 40 | + instance._always = lambda: 'ok' |
40 | 41 | instance._readonly = lambda: 1 |
41 | 42 | instance._assignable = m |
42 | 43 |
|
43 | | - assert instance.readonly() == 1 |
| 44 | + instance._handle = True |
| 45 | + assert instance.always() == 'ok' |
| 46 | + assert instance.readonly() == 1 # works. |
44 | 47 |
|
45 | 48 | instance.assignable() |
46 | 49 | m.assert_called_once_with() |
47 | 50 |
|
48 | 51 | m.reset_mock() |
49 | | - instance.assignable(2) |
| 52 | + instance.assignable(2) # as parameter. |
| 53 | + m.assert_called_once_with(2) |
| 54 | + |
| 55 | + m.reset_mock() |
| 56 | + instance.assignable = 3 # as attribute. |
| 57 | + m.assert_called_once_with(3) |
| 58 | + |
| 59 | + instance._handle = False |
| 60 | + assert instance.always() == 'ok' |
| 61 | + assert instance.readonly() is None # is None when handle is False. |
| 62 | + |
| 63 | + m.reset_mock() |
| 64 | + instance.assignable() |
| 65 | + m.assert_called_once_with() |
| 66 | + |
| 67 | + m.reset_mock() |
| 68 | + instance.assignable(2) # as parameter. |
50 | 69 | m.assert_called_once_with(2) |
51 | 70 |
|
52 | 71 | m.reset_mock() |
53 | | - instance.assignable = 3 |
| 72 | + instance.assignable = 3 # as attribute. |
54 | 73 | m.assert_called_once_with(3) |
0 commit comments