| 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
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..2f6d9150092ba1f4f6f256f1424ab1ccdf0c2a12
|
| --- /dev/null
|
| +++ b/client/libs/logdog/tests/bootstrap_test.py
|
| @@ -0,0 +1,49 @@
|
| +# Copyright 2016 The LUCI Authors. All rights reserved.
|
| +# Use of this source code is governed by the Apache v2.0 license that can be
|
| +# found in the LICENSE file.
|
| +
|
| +import unittest
|
| +import itertools
|
| +import StringIO
|
| +
|
| +from client.libs.logdog import bootstrap
|
| +
|
| +
|
| +class BootstrapTestCase(unittest.TestCase):
|
| +
|
| + def setUp(self):
|
| + self.env = {
|
| + bootstrap.ButlerBootstrap._ENV_PROJECT: 'test-project',
|
| + bootstrap.ButlerBootstrap._ENV_PREFIX: 'foo/bar',
|
| + bootstrap.ButlerBootstrap._ENV_STREAM_SERVER_PATH: 'fake:path',
|
| + }
|
| +
|
| + def testProbeSucceeds(self):
|
| + bs = bootstrap.ButlerBootstrap.probe(self.env)
|
| + self.assertEqual(bs, bootstrap.ButlerBootstrap(
|
| + project='test-project',
|
| + prefix='foo/bar',
|
| + streamserver_uri='fake:path'))
|
| +
|
| + def testProbeNoBootstrapRaisesError(self):
|
| + self.assertRaises(bootstrap.NotBootstrappedError,
|
| + bootstrap.ButlerBootstrap.probe, env={})
|
| +
|
| + def testProbeMissingProjectRaisesError(self):
|
| + self.env.pop(bootstrap.ButlerBootstrap._ENV_PROJECT)
|
| + self.assertRaises(bootstrap.NotBootstrappedError,
|
| + bootstrap.ButlerBootstrap.probe, env=self.env)
|
| +
|
| + def testProbeMissingPrefixRaisesError(self):
|
| + self.env.pop(bootstrap.ButlerBootstrap._ENV_PREFIX)
|
| + self.assertRaises(bootstrap.NotBootstrappedError,
|
| + bootstrap.ButlerBootstrap.probe, env=self.env)
|
| +
|
| + def testProbeInvalidPrefixRaisesError(self):
|
| + self.env[bootstrap.ButlerBootstrap._ENV_PREFIX] = '!!! not valid !!!'
|
| + self.assertRaises(bootstrap.NotBootstrappedError,
|
| + bootstrap.ButlerBootstrap.probe, env=self.env)
|
| +
|
| +
|
| +if __name__ == '__main__':
|
| + unittest.main()
|
|
|