def test_get_all_metrics(self):
"""Gets group's metrics"""
def _game(groups, log, _print=False):
self.assertEqual(log, "log")
return [Effect(Constant(["foo", "bar"])), Effect(Constant(["baz"]))]
d = get_all_metrics(base_dispatcher, object(), "log", get_all_metrics_effects=_game)
self.assertEqual(set(self.successResultOf(d)), set(["foo", "bar", "baz"]))
def test_ignore_error_results(self):
"""
When get_all_metrics_effects returns a list containing a None, those
elements are ignored.
"""
def _game(groups, log, _print=False):
self.assertEqual(log, "log")
return [Effect(Constant(None)), Effect(Constant(["foo"]))]
d = get_all_metrics(base_dispatcher, object(), "log", get_all_metrics_effects=_game)
self.assertEqual(self.successResultOf(d), ["foo"])
def test_ignore_error_results(self):
"""
When get_all_metrics_effects returns a list containing a None, those
elements are ignored.
"""
def _game(groups, log, _print=False):
self.assertEqual(log, "log")
return [Effect(Constant(None)), Effect(Constant(['foo']))]
d = get_all_metrics(base_dispatcher,
object(),
"log",
get_all_metrics_effects=_game)
self.assertEqual(self.successResultOf(d), ['foo'])
def test_get_all_metrics(self):
"""Gets group's metrics"""
def _game(groups, log, _print=False):
self.assertEqual(log, "log")
return [
Effect(Constant(['foo', 'bar'])),
Effect(Constant(['baz']))
]
d = get_all_metrics(base_dispatcher,
object(),
"log",
get_all_metrics_effects=_game)
self.assertEqual(set(self.successResultOf(d)),
set(['foo', 'bar', 'baz']))
def test_get_all_metrics(self):
"""
Gets group's metrics
"""
servers_t1 = {'g1': [{'status': 'ACTIVE'}] * 3 + [{'status': 'BUILD'}] * 2,
'g2': [{'status': 'ACTIVE'}]}
servers_t2 = {'g4': [{'status': 'ACTIVE'}, {'status': 'BUILD'}]}
groups = [{'tenantId': 't1', 'groupId': 'g1', 'desired': 3},
{'tenantId': 't1', 'groupId': 'g2', 'desired': 4},
{'tenantId': 't2', 'groupId': 'g4', 'desired': 2}]
self.tenant_servers['t1'] = servers_t1
self.tenant_servers['t2'] = servers_t2
d = get_all_metrics(groups, 'a', 'n', 'r', clock='c')
self.assertEqual(
set(self.successResultOf(d)),
set([GroupMetrics('t1', 'g1', 3, 3, 2), GroupMetrics('t1', 'g2', 4, 1, 0),
GroupMetrics('t2', 'g4', 2, 1, 1)]))
