summaryrefslogtreecommitdiff
path: root/service/test/perf/search/test_Search.py
blob: 63636789c71f358eeb4b5b17888c04584b5a69d8 (plain)
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
#
# Copyright (c) 2014 ThoughtWorks, Inc.
#
# Pixelated is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# Pixelated is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU Affero General Public License for more details.
#
# 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 unittest
import json

from funkload.FunkLoadTestCase import FunkLoadTestCase
from funkload.utils import Data
from test.support.integration import AppTestClient


class Search(FunkLoadTestCase):

    def setUpBench(self):
        client = AppTestClient()

        # setup data
        client.add_multiple_to_mailbox(10, 'INBOX', to='to@inbox.com', cc='cc@inbox.com', bcc='bcc@inbox.com', tags=['inbox'])
        client.add_multiple_to_mailbox(10, 'TRASH', to='to@trash.com', cc='cc@trash.com', bcc='bcc@trash.com', tags=['trash'])
        client.add_multiple_to_mailbox(10, 'DRAFTS', to='to@drafts.com', cc='cc@drafts.com', bcc='bcc@drafts.com', tags=['drafts'])

        self.call_to_terminate = client.run_on_a_thread(logfile='results/app.log')

    def tearDownBench(self):
        self.call_to_terminate()

    def setUp(self):
        """Setting up test."""
        self.server_url = self.conf_get('main', 'url')
        self.mails_by_tag_url = self.server_url + '/mails?q=%%22tag:%s%%22&w=25&p=0'

    def idents_by_tag(self, tag):
        return list(mail['ident'] for mail in json.loads(self.get(self.mails_by_tag_url % tag, description='Query mails by tag').body)['mails'])

    def test_search(self):
        """ Query contacts and tags. Write a new tag, updating index. Query again. """
        contacts_url = self.server_url + '/contacts?q=rec'
        mail_tags_url = self.server_url + '/mail/%s/tags'
        all_tags_url = self.server_url + '/tags'
        tags = ['inbox', 'trash', 'drafts']
        nb_time = self.conf_getInt('main', 'nb_time')

        for i in range(nb_time):
            self.get(contacts_url, description='Query for contacts')
            idents = self.idents_by_tag(tags[i % 3])
            tag_data = Data('application/json', '{"newtags":["newtag%s"]}' % i)
            self.post(mail_tags_url % idents[i % 3], tag_data, description='Change tags on a mail')
            self.idents_by_tag(tags[i % 3])
            self.get(all_tags_url, description='Query for all tags listing')

if __name__ in ('main', '__main__'):
    unittest.main()