diff options
Diffstat (limited to 'tests/unittests/sources/test_aliyun.py')
| -rw-r--r-- | tests/unittests/sources/test_aliyun.py | 13 |
1 files changed, 12 insertions, 1 deletions
diff --git a/tests/unittests/sources/test_aliyun.py b/tests/unittests/sources/test_aliyun.py index fe4e54b5..f95923a4 100644 --- a/tests/unittests/sources/test_aliyun.py +++ b/tests/unittests/sources/test_aliyun.py @@ -98,6 +98,15 @@ class TestAliYunDatasource(test_helpers.ResponsesTestCase): "instance-identity", ) + @property + def token_url(self): + return os.path.join( + self.metadata_address, + "latest", + "api", + "token", + ) + def register_mock_metaserver(self, base_url, data): def register_helper(register, base_url, body): if isinstance(body, str): @@ -127,6 +136,7 @@ class TestAliYunDatasource(test_helpers.ResponsesTestCase): self.register_mock_metaserver(self.metadata_url, self.default_metadata) self.register_mock_metaserver(self.userdata_url, self.default_userdata) self.register_mock_metaserver(self.identity_url, self.default_identity) + self.responses.add(responses.PUT, self.token_url, "API-TOKEN") def _test_get_data(self): self.assertEqual(self.ds.metadata, self.default_metadata) @@ -151,8 +161,9 @@ class TestAliYunDatasource(test_helpers.ResponsesTestCase): self.default_metadata["hostname"], self.ds.get_hostname().hostname ) + @mock.patch("cloudinit.sources.DataSourceEc2.util.is_resolvable") @mock.patch("cloudinit.sources.DataSourceAliYun._is_aliyun") - def test_with_mock_server(self, m_is_aliyun): + def test_with_mock_server(self, m_is_aliyun, m_resolv): m_is_aliyun.return_value = True self.regist_default_server() ret = self.ds.get_data() |