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
|
import pytest
from twisted.internet.defer import gatherResults
@pytest.inlineCallbacks
def load_up(client, amount, payload):
# create a bunch of local documents
deferreds = []
for i in xrange(amount):
deferreds.append(client.create_doc({'content': payload}))
yield gatherResults(deferreds)
# Each test created with this function will:
#
# - get a fresh client.
# - iterate:
# - setup: create N docs of a certain size
# - benchmark: sync() -- uploads N docs.
def create_upload(uploads, size):
@pytest.inlineCallbacks
@pytest.mark.benchmark(group="test_upload")
def test(soledad_client, txbenchmark_with_setup, payload):
client = soledad_client()
def setup():
return load_up(client, uploads, payload(size))
yield txbenchmark_with_setup(setup, client.sync)
return test
# ATTENTION: update the documentation in ../docs/benchmarks.rst if you change
# the number of docs or the doc sizes for the tests below.
test_upload_20_500k = create_upload(20, 500 * 1000)
test_upload_100_100k = create_upload(100, 100 * 1000)
test_upload_1000_10k = create_upload(1000, 10 * 1000)
# Each test created with this function will:
#
# - get a fresh client.
# - create N docs of a certain size
# - sync (uploads those docs)
# - iterate:
# - setup: get a fresh client with empty local db
# - benchmark: sync() -- downloads N docs.
def create_download(downloads, size):
@pytest.inlineCallbacks
@pytest.mark.benchmark(group="test_download")
def test(soledad_client, txbenchmark_with_setup, payload):
client = soledad_client()
yield load_up(client, downloads, payload(size))
yield client.sync()
# We could create them directly on couch, but sending them
# ensures we are dealing with properly encrypted docs
def setup():
return soledad_client(force_fresh_db=True)
def sync(clean_client):
return clean_client.sync()
yield txbenchmark_with_setup(setup, sync)
return test
# ATTENTION: update the documentation in ../docs/benchmarks.rst if you change
# the number of docs or the doc sizes for the tests below.
test_download_20_500k = create_download(20, 500 * 1000)
test_download_100_100k = create_download(100, 100 * 1000)
test_download_1000_10k = create_download(1000, 10 * 1000)
@pytest.inlineCallbacks
@pytest.mark.benchmark(group="test_nothing_to_sync")
def test_nothing_to_sync(soledad_client, txbenchmark_with_setup):
def setup():
return soledad_client()
def sync(clean_client):
return clean_client.sync()
yield txbenchmark_with_setup(setup, sync)
|
