LEFT | RIGHT |
1 import os | 1 import os |
2 | 2 |
3 from twisted.python.failure import Failure | 3 from twisted.python.failure import Failure |
4 | 4 |
5 from juju.environment.tests.test_config import EnvironmentsConfigTestBase | 5 from juju.environment.tests.test_config import EnvironmentsConfigTestBase |
6 from juju.errors import ProviderInteractionError, JujuError | 6 from juju.errors import ProviderInteractionError, JujuError |
7 from juju.lib import serializer | 7 from juju.lib import serializer |
8 from juju.lib.testing import TestCase | 8 from juju.lib.testing import TestCase |
9 from juju.providers.common.utils import ( | 9 from juju.providers.common.utils import ( |
10 convert_unknown_error, format_cloud_init, get_user_authorized_keys) | 10 convert_unknown_error, format_cloud_init, get_user_authorized_keys) |
(...skipping 101 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
112 self.assertTrue(config["apt_update"]) | 112 self.assertTrue(config["apt_update"]) |
113 self.assertTrue(config["apt_upgrade"]) | 113 self.assertTrue(config["apt_upgrade"]) |
114 formatted_repos = [dict(source=r) for r in repositories] | 114 formatted_repos = [dict(source=r) for r in repositories] |
115 self.assertEqual(config["apt_sources"], formatted_repos) | 115 self.assertEqual(config["apt_sources"], formatted_repos) |
116 self.assertEqual(config["runcmd"], scripts) | 116 self.assertEqual(config["runcmd"], scripts) |
117 self.assertEqual(config["machine-data"]["magic"], [1, 2, 3]) | 117 self.assertEqual(config["machine-data"]["magic"], [1, 2, 3]) |
118 | 118 |
119 def test_format_cloud_init_when_testing(self): | 119 def test_format_cloud_init_when_testing(self): |
120 """When in testing mode for speed of startup disable update/upgrade. | 120 """When in testing mode for speed of startup disable update/upgrade. |
121 """ | 121 """ |
122 self.change_environment(JUJU_TESTING="YES") | 122 self.change_environment(JUJU_TESTING="fast") |
123 packages = ["python-lxml"] | 123 packages = ["python-lxml"] |
124 scripts = ["wget http://lwn.net > /tmp/out"] | 124 scripts = ["wget http://lwn.net > /tmp/out"] |
125 repositories = ["ppa:juju/pkgs"] | 125 repositories = ["ppa:juju/pkgs"] |
126 output = format_cloud_init( | 126 output = format_cloud_init( |
127 ["zebra"], | 127 ["zebra"], |
128 packages=packages, | 128 packages=packages, |
129 scripts=scripts, | 129 scripts=scripts, |
130 repositories=repositories, | 130 repositories=repositories, |
131 data={"magic": [1, 2, 3]}) | 131 data={"magic": [1, 2, 3]}) |
132 | 132 |
133 lines = output.split("\n") | 133 lines = output.split("\n") |
134 self.assertEqual(lines.pop(0), "#cloud-config") | 134 self.assertEqual(lines.pop(0), "#cloud-config") |
135 config = serializer.yaml_load("\n".join(lines)) | 135 config = serializer.yaml_load("\n".join(lines)) |
136 self.assertFalse(config["apt_update"]) | 136 self.assertFalse(config["apt_update"]) |
137 self.assertFalse(config["apt_upgrade"]) | 137 self.assertFalse(config["apt_upgrade"]) |
| 138 |
| 139 self.change_environment(JUJU_TESTING="yes") |
| 140 output = format_cloud_init( |
| 141 ["zebra"], |
| 142 packages=packages, |
| 143 scripts=scripts, |
| 144 repositories=repositories, |
| 145 data={"magic": [1, 2, 3]}) |
| 146 |
| 147 lines = output.split("\n") |
| 148 config = serializer.yaml_load("\n".join(lines)) |
| 149 self.assertTrue(config["apt_update"]) |
| 150 self.assertTrue(config["apt_upgrade"]) |
LEFT | RIGHT |