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
134
135
136
|
# -*- coding: utf-8 -*-
# Copyright (c) 2016, Menno Smits
# Released subject to the New BSD License
# Please see http://en.wikipedia.org/wiki/BSD_licenses
from unittest.mock import Mock, patch, sentinel
from imapclient.imapclient import ANSWERED, DELETED, DRAFT, FLAGGED, RECENT, SEEN
from .imapclient_test import IMAPClientTest
class TestFlagsConsts(IMAPClientTest):
def test_flags_are_bytes(self):
for flag in DELETED, SEEN, ANSWERED, FLAGGED, DRAFT, RECENT:
if not isinstance(flag, bytes):
self.fail("%r flag is not bytes" % flag)
class TestFlags(IMAPClientTest):
def setUp(self):
super(TestFlags, self).setUp()
self.client._command_and_check = Mock()
def test_get(self):
with patch.object(
self.client,
"fetch",
autospec=True,
return_value={123: {b"FLAGS": [b"foo", b"bar"]}, 444: {b"FLAGS": [b"foo"]}},
):
out = self.client.get_flags(sentinel.messages)
self.client.fetch.assert_called_with(sentinel.messages, ["FLAGS"])
self.assertEqual(out, {123: [b"foo", b"bar"], 444: [b"foo"]})
def test_set(self):
self.check(self.client.set_flags, b"FLAGS")
def test_add(self):
self.check(self.client.add_flags, b"+FLAGS")
def test_remove(self):
self.check(self.client.remove_flags, b"-FLAGS")
def check(self, meth, expected_command):
self._check(meth, expected_command)
self._check(meth, expected_command, silent=True)
def _check(self, meth, expected_command, silent=False):
if silent:
expected_command += b".SILENT"
cc = self.client._command_and_check
cc.return_value = [
b"11 (FLAGS (blah foo) UID 1)",
b"11 (UID 1 OTHER (dont))",
b"22 (FLAGS (foo) UID 2)",
b"22 (UID 2 OTHER (care))",
]
resp = meth([1, 2], "foo", silent=silent)
cc.assert_called_once_with("store", b"1,2", expected_command, "(foo)", uid=True)
if silent:
self.assertIsNone(resp)
else:
self.assertEqual(
resp,
{
1: (b"blah", b"foo"),
2: (b"foo",),
},
)
cc.reset_mock()
class TestGmailLabels(IMAPClientTest):
def setUp(self):
super(TestGmailLabels, self).setUp()
self.client._command_and_check = Mock()
def test_get(self):
with patch.object(
self.client,
"fetch",
autospec=True,
return_value={
123: {b"X-GM-LABELS": [b"foo", b"&AUE-abel"]},
444: {b"X-GM-LABELS": [b"foo"]},
},
):
out = self.client.get_gmail_labels(sentinel.messages)
self.client.fetch.assert_called_with(sentinel.messages, [b"X-GM-LABELS"])
self.assertEqual(out, {123: ["foo", "Łabel"], 444: ["foo"]})
def test_set(self):
self.check(self.client.set_gmail_labels, b"X-GM-LABELS")
def test_add(self):
self.check(self.client.add_gmail_labels, b"+X-GM-LABELS")
def test_remove(self):
self.check(self.client.remove_gmail_labels, b"-X-GM-LABELS")
def check(self, meth, expected_command):
self._check(meth, expected_command)
self._check(meth, expected_command, silent=True)
def _check(self, meth, expected_command, silent=False):
if silent:
expected_command += b".SILENT"
cc = self.client._command_and_check
cc.return_value = [
b'11 (X-GM-LABELS (&AUE-abel "f\\"o\\"o") UID 1)',
b'22 (X-GM-LABELS ("f\\"o\\"o") UID 2)',
b"11 (UID 1 FLAGS (dont))",
b"22 (UID 2 FLAGS (care))",
]
resp = meth([1, 2], 'f"o"o', silent=silent)
cc.assert_called_once_with(
"store", b"1,2", expected_command, '("f\\"o\\"o")', uid=True
)
if silent:
self.assertIsNone(resp)
else:
self.assertEqual(
resp,
{
1: ["Łabel", 'f"o"o'],
2: [
'f"o"o',
],
},
)
cc.reset_mock()
|
