2011-08-09 20:40:56 +00:00
|
|
|
import argparse
|
2011-09-13 21:53:02 +00:00
|
|
|
from ..orchestra import cluster
|
2011-08-09 20:40:56 +00:00
|
|
|
|
|
|
|
|
from nose.tools import (
|
|
|
|
|
eq_ as eq,
|
|
|
|
|
assert_raises,
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
from .. import misc
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class FakeRemote(object):
|
|
|
|
|
pass
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def test_get_clients_simple():
|
|
|
|
|
ctx = argparse.Namespace()
|
|
|
|
|
remote = FakeRemote()
|
|
|
|
|
ctx.cluster = cluster.Cluster(
|
|
|
|
|
remotes=[
|
|
|
|
|
(remote, ['client.0', 'client.1'])
|
|
|
|
|
],
|
|
|
|
|
)
|
|
|
|
|
g = misc.get_clients(ctx=ctx, roles=['client.1'])
|
|
|
|
|
got = next(g)
|
|
|
|
|
eq(len(got), 2)
|
|
|
|
|
eq(got[0], ('1'))
|
|
|
|
|
assert got[1] is remote
|
|
|
|
|
assert_raises(StopIteration, next, g)
|
