| Index: client/libs/logdog/tests/bootstrap_test.py
|
| diff --git a/client/libs/logdog/tests/bootstrap_test.py b/client/libs/logdog/tests/bootstrap_test.py
|
| index 0d5f968dcd4be3303c753ac01ae7b8e6b2ff7ad2..21b0c674c3b93672357ff656a76a0894d8ad9a8d 100755
|
| --- a/client/libs/logdog/tests/bootstrap_test.py
|
| +++ b/client/libs/logdog/tests/bootstrap_test.py
|
| @@ -12,7 +12,7 @@ ROOT_DIR = os.path.dirname(os.path.abspath(os.path.join(
|
| os.pardir, os.pardir, os.pardir)))
|
| sys.path.insert(0, ROOT_DIR)
|
|
|
| -from libs.logdog import bootstrap
|
| +from libs.logdog import bootstrap, stream
|
|
|
|
|
| class BootstrapTestCase(unittest.TestCase):
|
| @@ -22,6 +22,7 @@ class BootstrapTestCase(unittest.TestCase):
|
| bootstrap.ButlerBootstrap._ENV_PROJECT: 'test-project',
|
| bootstrap.ButlerBootstrap._ENV_PREFIX: 'foo/bar',
|
| bootstrap.ButlerBootstrap._ENV_STREAM_SERVER_PATH: 'fake:path',
|
| + bootstrap.ButlerBootstrap._ENV_COORDINATOR_HOST: 'example.appspot.com',
|
| }
|
|
|
| def testProbeSucceeds(self):
|
| @@ -29,7 +30,8 @@ class BootstrapTestCase(unittest.TestCase):
|
| self.assertEqual(bs, bootstrap.ButlerBootstrap(
|
| project='test-project',
|
| prefix='foo/bar',
|
| - streamserver_uri='fake:path'))
|
| + streamserver_uri='fake:path',
|
| + coordinator_host='example.appspot.com'))
|
|
|
| def testProbeNoBootstrapRaisesError(self):
|
| self.assertRaises(bootstrap.NotBootstrappedError,
|
| @@ -50,6 +52,26 @@ class BootstrapTestCase(unittest.TestCase):
|
| self.assertRaises(bootstrap.NotBootstrappedError,
|
| bootstrap.ButlerBootstrap.probe, env=self.env)
|
|
|
| + def testCreateStreamClient(self):
|
| + class TestStreamClient(stream.StreamClient):
|
| + @classmethod
|
| + def _create(cls, _value, **kwargs):
|
| + return cls(**kwargs)
|
| +
|
| + def _connect_raw(self):
|
| + raise NotImplementedError()
|
| +
|
| + reg = stream.StreamProtocolRegistry()
|
| + reg.register_protocol('test', TestStreamClient)
|
| + bs = bootstrap.ButlerBootstrap(
|
| + project='test-project',
|
| + prefix='foo/bar',
|
| + streamserver_uri='test:',
|
| + coordinator_host='example.appspot.com')
|
| + sc = bs.stream_client(reg=reg)
|
| + self.assertEqual(sc.prefix, 'foo/bar')
|
| + self.assertEqual(sc.coordinator_host, 'example.appspot.com')
|
| +
|
|
|
| if __name__ == '__main__':
|
| unittest.main()
|
|
|