summaryrefslogtreecommitdiff
path: root/service/test/integration/tags_test.py
diff options
context:
space:
mode:
Diffstat (limited to 'service/test/integration/tags_test.py')
-rw-r--r--service/test/integration/tags_test.py9
1 files changed, 4 insertions, 5 deletions
diff --git a/service/test/integration/tags_test.py b/service/test/integration/tags_test.py
index bb3bdbdb..03d6f4b3 100644
--- a/service/test/integration/tags_test.py
+++ b/service/test/integration/tags_test.py
@@ -14,9 +14,8 @@
# You should have received a copy of the GNU Affero General Public License
# along with Pixelated. If not, see <http://www.gnu.org/licenses/>.
import json
-import unittest
-from test.support.integration_helper import MailBuilder, SoledadTestBase
+from test.support.integration import *
class TagsTest(SoledadTestBase):
@@ -32,7 +31,7 @@ class TagsTest(SoledadTestBase):
def test_add_tag_to_an_inbox_mail_and_query(self):
mail = MailBuilder().with_subject('Mail with tags').build_input_mail()
- self.add_mail_to_inbox(mail)
+ self.client.add_mail_to_inbox(mail)
self.post_tags(mail.ident, self._tags_json(['IMPORTANT']))
@@ -44,10 +43,10 @@ class TagsTest(SoledadTestBase):
def test_addition_of_reserved_tags_is_not_allowed(self):
mail = MailBuilder().with_subject('Mail with tags').build_input_mail()
- self.add_mail_to_inbox(mail)
+ self.client.add_mail_to_inbox(mail)
response = self.post_tags(mail.ident, self._tags_json(['DRAFTS']))
self.assertEquals("None of the following words can be used as tags: drafts", response)
- mail = self.mailboxes.inbox().mail(mail.ident)
+ mail = self.client.mailboxes.inbox().mail(mail.ident)
self.assertNotIn('drafts', mail.tags)