DXR is a code search and navigation tool aimed at making sense of large projects. It supports full-text and regex searches as well as structural queries.

Mercurial (7ac42b6b5ee8)

VCS Links

Line Code
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
# ***** BEGIN LICENSE BLOCK *****
# Version: MPL 1.1/GPL 2.0/LGPL 2.1
#
# The contents of this file are subject to the Mozilla Public License Version
# 1.1 (the "License"); you may not use this file except in compliance with
# the License. You may obtain a copy of the License at
# http://www.mozilla.org/MPL/
#
# Software distributed under the License is distributed on an "AS IS" basis,
# WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
# for the specific language governing rights and limitations under the
# License.
#
# The Original Code is Weave Basic Object Server
#
# The Initial Developer of the Original Code is
# Mozilla Services.
# Portions created by the Initial Developer are Copyright (C) 2010
# the Initial Developer. All Rights Reserved.
#
# Contributor(s):
#    Toby Elliott (telliott@mozilla.com)
#
# Alternatively, the contents of this file may be used under the terms of
# either the GNU General Public License Version 2 or later (the "GPL"), or
# the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
# in which case the provisions of the GPL or the LGPL are applicable instead
# of those above. If you wish to allow use of your version of this file only
# under the terms of either the GPL or the LGPL, and not to allow others to
# use your version of this file under the terms of the MPL, indicate your
# decision by deleting the provisions above and replace them with the notice
# and other provisions required by the GPL or the LGPL. If you do not delete
# the provisions above, a recipient may use your version of this file under
# the terms of any one of the MPL, the GPL or the LGPL.
#
# ***** END LICENSE BLOCK *****


# A script to do a framework for weave session testing
#
# This script shows the recommended style for scripts, with a
# TestRunner class. The script is executed just once by each worker
# process and defines the TestRunner class. The Grinder creates an
# instance of TestRunner for each worker thread, and repeatedly calls
# the instance for each run of that thread.

# Detailed documentation can be found at https://wiki.mozilla.org/Labs/Weave/GrinderTesting

from net.grinder.script.Grinder import grinder
from net.grinder.script import Test
from net.grinder.plugin.http import HTTPRequest
from net.grinder.plugin.http import HTTPPluginControl
from math import floor
import random
import time

# A shorter alias for the grinder.logger.output() method.
log = grinder.logger.output
verbose = 10


VERSION = '1.0'

collections = {'clients': 1, 'crypto': 7, 'forms': 158, 'history': 1214,
                'keys': 2, 'meta': 1, 'bookmarks': 303, 'prefs': 1, 'passwords': 28}
utilities = HTTPPluginControl.getHTTPUtilities()


def requestSequence():
    node = 0
    while node < 21:
        userid = random.randint(1,100000)
        node = int(floor(userid/100) % 20 + 10)
    username = "user" + str(userid)
    host = "https://dev-sync%i.services.mozilla.com" % node
    request = HTTPRequest(url=host)
    auth = utilities.basicAuthorizationHeader(username, 'password')
    request.setHeaders([auth])

    #GET https://server/pathname/version/username/info/collections
    getCollectionsString = "/%s/%s/info/collections" % (VERSION, username)
    response = request.GET(getCollectionsString)
    if response.getStatusCode() not in [200, 404]:
        raise RuntimeError("status code: %i" % response.getStatusCode())
    if verbose > 0:
        log("GET %s - %s" % (username, response.getText()))

    for name, total in collections.items():
        payloadarray = []
        for i in range(1, total):
            payloadarray.append(payload(username))
            if not i % 100:
                post_payloads(request, username, name, payloadarray)
                payloadarray = []
        if len(payloadarray):
            post_payloads(request,username, name, payloadarray)


def post_payloads(request, username, collectionname, payload):
    postCollectionsString = "/%s/%s/storage/%s" % (VERSION, username, collectionname)
    postData = "[%s]" % (",".join(payload))
    response = request.POST(postCollectionsString, postData)
    if response.getStatusCode() != 200:
        raise RuntimeError("status code: %i" % response.getStatusCode())
    if verbose > 0:
        log("POST %s (%i) (%s) - %s" % (collectionname, len(payload), username, response.getText()))


def payload(username):
    return "{\"id\": %i, \"payload\": \"%s\"}" % (random.randint(1, 1000), username * random.randint(50, 200))


weaveTest = Test(2, "New User").wrap(requestSequence)


# A TestRunner instance is created for each thread. It can be used to
# store thread-specific data.
class TestRunner:

    # This method is called for every run.
    def __call__(self):
        weaveTest()