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
|
import asyncio
import functools
import random
import time
from testing import Client
from testing import default_test_setup
from testing import gen_data
from testing import gen_points
from testing import gen_series
from testing import InsertError
from testing import PoolError
from testing import QueryError
from testing import run_test
from testing import Series
from testing import Server
from testing import ServerError
from testing import SiriDB
from testing import TestBase
from testing import UserAuthError
from testing import parse_args
class TestPool(TestBase):
title = 'Test pool object'
async def insert(self, client, series, n, timeout=1):
for _ in range(n):
await client.insert_some_series(series, n=0.01, timeout=timeout)
await asyncio.sleep(1.0)
@default_test_setup(4)
async def run(self):
series = gen_series(n=10000)
await self.client0.connect()
task0 = asyncio.ensure_future(self.insert(
self.client0,
series,
200))
await asyncio.sleep(5)
await self.db.add_pool(self.server1, sleep=3)
with self.assertRaises(AssertionError):
await self.db.add_pool(self.server2, sleep=3)
await self.client1.connect()
task1 = asyncio.ensure_future(self.insert(
self.client1,
series,
150))
await asyncio.sleep(5)
await self.assertIsRunning(self.db, self.client0, timeout=200)
await asyncio.sleep(25)
await self.db.add_replica(self.server3, 1, sleep=3)
await self.client3.connect()
await self.assertIsRunning(self.db, self.client3, timeout=200)
await asyncio.sleep(30)
await self.db.add_pool(self.server2, sleep=3)
await self.client2.connect()
task2 = asyncio.ensure_future(self.insert(
self.client2,
series,
100))
await self.assertIsRunning(self.db, self.client0, timeout=600)
await asyncio.wait_for(task0, None)
await asyncio.wait_for(task1, None)
await asyncio.wait_for(task2, None)
await asyncio.sleep(2)
await self.assertSeries(self.client0, series)
await self.assertSeries(self.client1, series)
await self.assertSeries(self.client2, series)
await self.assertSeries(self.client3, series)
self.client0.close()
self.client1.close()
self.client2.close()
self.client3.close()
if __name__ == '__main__':
parse_args()
run_test(TestPool())
|
