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 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152
|
import asyncio
import functools
import random
import time
import math
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
DATA = {
'series-001 float': [
[1471254705, 1.5],
[1471254707, -3.5],
[1471254710, -7.3]],
'series-001 integer': [
[1471254705, 5],
[1471254708, -3],
[1471254710, -7]],
'series-002 float': [
[1471254705, 3.5],
[1471254707, -2.5],
[1471254710, -8.3]],
'series-002 integer': [
[1471254705, 4],
[1471254708, -1],
[1471254710, -8]],
'aggr': [
[1447249033, 531], [1447249337, 534],
[1447249633, 535], [1447249937, 531],
[1447250249, 532], [1447250549, 537],
[1447250868, 530], [1447251168, 520],
[1447251449, 54], [1447251749, 54],
[1447252049, 513], [1447252349, 537],
[1447252649, 528], [1447252968, 531],
[1447253244, 533], [1447253549, 538],
[1447253849, 534], [1447254149, 532],
[1447254449, 533], [1447254748, 537]],
'huge': [
[1471254705, 9223372036854775807],
[1471254706, 9223372036854775806],
[1471254707, 9223372036854775805],
[1471254708, 9223372036854775804]],
'equal ts': [
[1471254705, 0], [1471254705, 1], [1471254705, 1],
[1471254707, 0], [1471254707, 1], [1471254708, 0],
],
'variance': [
[1471254705, 2.75], [1471254706, 1.75], [1471254707, 1.25],
[1471254708, 0.25], [1471254709, 0.5], [1471254710, 1.25],
[1471254711, 3.5]
],
'pvariance': [
[1471254705, 0.0], [1471254706, 0.25], [1471254707, 0.25],
[1471254708, 1.25], [1471254709, 1.5], [1471254710, 1.75],
[1471254711, 2.75], [1471254712, 3.25]
],
'filter': [
[1471254705, 5],
[1471254710, -3],
[1471254715, -7],
[1471254720, 7]
],
'one': [
[1471254710, 1]
],
'log': [
[1471254710, 'log line one'],
[1471254712, 'log line two'],
[1471254714, 'another line (three)'],
[1471254716, 'and yet one more'],
],
'special': [
[1471254705, 0.1],
[1471254706, math.nan],
[1471254707, math.inf],
[1471254708, -math.inf],
]
}
class TestCluster(TestBase):
title = 'Test siridb-cluster'
@default_test_setup(4, time_precision='s')
async def run(self):
await self.client0.connect()
# await self.client0.insert(DATA)
# await self.client0.query('''
# alter series /series.*/ tag `SERIES`
# ''')
# await asyncio.sleep(3.0)
# await self.client0.query('''
# alter series /.*/ - `SERIES` tag `OTHER`
# ''')
await self.db.add_replica(self.server1, 0)
await self.assertIsRunning(self.db, self.client0, timeout=30)
# await asyncio.sleep(35)
# await self.db.add_replica(self.server2, 0)
# await self.assertIsRunning(self.db, self.client0, timeout=30)
# await asyncio.sleep(45)
# await self.db.add_replica(self.server3, 1)
# await self.assertIsRunning(self.db, self.client0, timeout=12)
await asyncio.sleep(35)
await self.db.add_pool(self.server2)
await self.assertIsRunning(self.db, self.client0, timeout=12)
await asyncio.sleep(35)
# await self.db.add_pool(self.server5)
# await self.assertIsRunning(self.db, self.client0, timeout=12)
await self.db.add_replica(self.server3, 1)
await asyncio.sleep(5)
# await self.db.add_replica(self.server3, 1)
# await asyncio.sleep(5)
# await self.db.add_replica(self.server5, 2)
# await self.assertIsRunning(self.db, self.client0, timeout=35)
# self.client0.close()
if __name__ == '__main__':
parse_args()
run_test(TestCluster())
|