def test_pick_agents_full(self):
agents = {}
for i in range(10):
agents[i] = {
'id': i, 'mode': 'alone', 'node': 'uno',
}
picked = [set(a['id'] for a in arr)
for arr in server._pick_agents(agents, None)]
self.assertEqual([set(range(10))], picked)
def test_pick_agents_full(self):
agents = {}
for i in range(10):
agents[i] = {
'id': i, 'mode': 'alone', 'node': 'uno',
}
picked = [set(a['id'] for a in arr)
for arr in server._pick_agents(agents, None)]
self.assertEqual([set(range(10))], picked)
def test_pick_agents_linear(self):
agents = {}
for i in range(10):
agents[i] = {
'id': i, 'mode': 'alone', 'node': 'uno',
}
picked = [set(a['id'] for a in arr)
for arr in server._pick_agents(agents, 'linear')]
self.assertEqual([set(range(i + 1)) for i in range(0, 10)],
picked)
def test_pick_agents_linear(self):
agents = {}
for i in range(10):
agents[i] = {
'id': i, 'mode': 'alone', 'node': 'uno',
}
picked = [set(a['id'] for a in arr)
for arr in server._pick_agents(agents, 'linear')]
self.assertEqual([set(range(i + 1)) for i in range(0, 10)],
picked)
def test_pick_agents_quadratic(self):
agents = {}
for i in range(10):
agents[i] = {
'id': i, 'mode': 'alone', 'node': 'uno',
}
picked = [set(a['id'] for a in arr)
for arr in server._pick_agents(agents,
'quadratic')]
self.assertEqual([set(range(1)), set(range(2)),
set(range(5)), set(range(10))],
picked)
def test_pick_agents_quadratic(self):
agents = {}
for i in range(10):
agents[i] = {
'id': i, 'mode': 'alone', 'node': 'uno',
}
picked = [set(a['id'] for a in arr)
for arr in server._pick_agents(agents,
'quadratic')]
self.assertEqual([set(range(1)), set(range(2)),
set(range(5)), set(range(10))],
picked)
def test_pick_agents_full_filter_slaves(self):
agents = {}
for i in range(10):
agents['master_%s' % i] = {
'id': 'master_%s' % i, 'mode': 'master', 'node': 'uno',
}
agents['slave_%s' % i] = {
'id': 'slave_%s' % i, 'mode': 'slave', 'node': 'uno',
}
picked = [set(a['id'] for a in arr)
for arr in server._pick_agents(agents, None)]
self.assertEqual([set('master_%s' % i for i in range(10))],
picked)
def test_pick_agents_full_filter_slaves(self):
agents = {}
for i in range(10):
agents['master_%s' % i] = {
'id': 'master_%s' % i, 'mode': 'master', 'node': 'uno',
}
agents['slave_%s' % i] = {
'id': 'slave_%s' % i, 'mode': 'slave', 'node': 'uno',
}
picked = [set(a['id'] for a in arr)
for arr in server._pick_agents(agents, None)]
self.assertEqual([set('master_%s' % i for i in range(10))],
picked)
def test_pick_agents_full_filter_minions(self):
agents = {}
for i in range(10):
agents['primary_%s' % i] = {
'id': 'primary_%s' % i, 'mode': 'primary', 'node': 'uno',
}
agents['minion_%s' % i] = {
'id': 'minion_%s' % i, 'mode': 'minion', 'node': 'uno',
}
picked = [set(a['id'] for a in arr)
for arr in server._pick_agents(agents, None)]
self.assertEqual([set('primary_%s' % i for i in range(10))],
picked)
