1be8cef03b
UrlResponse: biggest change... make readurl return bytes, making user
know what to do with it.
util: add load_tfile_or_url for loading text file or url
as read_file_or_url now returns bytes
ec2_utils: all meta-data is text, remove non-obvious string translations
DigitalOcean: adjust for ec2_utils
DataSourceGCE, DataSourceMAAS: user-data is binary other fields are text.
openstack.py: read paths without decoding to text. This is ok as paths
other than user-data are json, and load_json will handle
load_file still returns text, and that is what most things use.
107 lines
3.9 KiB
Python
107 lines
3.9 KiB
Python
from cloudinit import util
|
|
|
|
from cloudinit.config import cc_apt_configure
|
|
from ..helpers import TestCase
|
|
|
|
import os
|
|
import re
|
|
import shutil
|
|
import tempfile
|
|
import unittest
|
|
|
|
|
|
class TestAptProxyConfig(TestCase):
|
|
def setUp(self):
|
|
super(TestAptProxyConfig, self).setUp()
|
|
self.tmp = tempfile.mkdtemp()
|
|
self.addCleanup(shutil.rmtree, self.tmp)
|
|
self.pfile = os.path.join(self.tmp, "proxy.cfg")
|
|
self.cfile = os.path.join(self.tmp, "config.cfg")
|
|
|
|
def _search_apt_config(self, contents, ptype, value):
|
|
return re.search(
|
|
r"acquire::%s::proxy\s+[\"']%s[\"'];\n" % (ptype, value),
|
|
contents, flags=re.IGNORECASE)
|
|
|
|
def test_apt_proxy_written(self):
|
|
cfg = {'apt_proxy': 'myproxy'}
|
|
cc_apt_configure.apply_apt_config(cfg, self.pfile, self.cfile)
|
|
|
|
self.assertTrue(os.path.isfile(self.pfile))
|
|
self.assertFalse(os.path.isfile(self.cfile))
|
|
|
|
contents = util.load_tfile_or_url(self.pfile)
|
|
self.assertTrue(self._search_apt_config(contents, "http", "myproxy"))
|
|
|
|
def test_apt_http_proxy_written(self):
|
|
cfg = {'apt_http_proxy': 'myproxy'}
|
|
cc_apt_configure.apply_apt_config(cfg, self.pfile, self.cfile)
|
|
|
|
self.assertTrue(os.path.isfile(self.pfile))
|
|
self.assertFalse(os.path.isfile(self.cfile))
|
|
|
|
contents = util.load_tfile_or_url(self.pfile)
|
|
self.assertTrue(self._search_apt_config(contents, "http", "myproxy"))
|
|
|
|
def test_apt_all_proxy_written(self):
|
|
cfg = {'apt_http_proxy': 'myproxy_http_proxy',
|
|
'apt_https_proxy': 'myproxy_https_proxy',
|
|
'apt_ftp_proxy': 'myproxy_ftp_proxy'}
|
|
|
|
values = {'http': cfg['apt_http_proxy'],
|
|
'https': cfg['apt_https_proxy'],
|
|
'ftp': cfg['apt_ftp_proxy'],
|
|
}
|
|
|
|
cc_apt_configure.apply_apt_config(cfg, self.pfile, self.cfile)
|
|
|
|
self.assertTrue(os.path.isfile(self.pfile))
|
|
self.assertFalse(os.path.isfile(self.cfile))
|
|
|
|
contents = util.load_tfile_or_url(self.pfile)
|
|
|
|
for ptype, pval in values.items():
|
|
self.assertTrue(self._search_apt_config(contents, ptype, pval))
|
|
|
|
def test_proxy_deleted(self):
|
|
util.write_file(self.cfile, "content doesnt matter")
|
|
cc_apt_configure.apply_apt_config({}, self.pfile, self.cfile)
|
|
self.assertFalse(os.path.isfile(self.pfile))
|
|
self.assertFalse(os.path.isfile(self.cfile))
|
|
|
|
def test_proxy_replaced(self):
|
|
util.write_file(self.cfile, "content doesnt matter")
|
|
cc_apt_configure.apply_apt_config({'apt_proxy': "foo"},
|
|
self.pfile, self.cfile)
|
|
self.assertTrue(os.path.isfile(self.pfile))
|
|
contents = util.load_tfile_or_url(self.pfile)
|
|
self.assertTrue(self._search_apt_config(contents, "http", "foo"))
|
|
|
|
def test_config_written(self):
|
|
payload = 'this is my apt config'
|
|
cfg = {'apt_config': payload}
|
|
|
|
cc_apt_configure.apply_apt_config(cfg, self.pfile, self.cfile)
|
|
|
|
self.assertTrue(os.path.isfile(self.cfile))
|
|
self.assertFalse(os.path.isfile(self.pfile))
|
|
|
|
self.assertEqual(util.load_tfile_or_url(self.cfile), payload)
|
|
|
|
def test_config_replaced(self):
|
|
util.write_file(self.pfile, "content doesnt matter")
|
|
cc_apt_configure.apply_apt_config({'apt_config': "foo"},
|
|
self.pfile, self.cfile)
|
|
self.assertTrue(os.path.isfile(self.cfile))
|
|
self.assertEqual(util.load_tfile_or_url(self.cfile), "foo")
|
|
|
|
def test_config_deleted(self):
|
|
# if no 'apt_config' is provided, delete any previously written file
|
|
util.write_file(self.pfile, "content doesnt matter")
|
|
cc_apt_configure.apply_apt_config({}, self.pfile, self.cfile)
|
|
self.assertFalse(os.path.isfile(self.pfile))
|
|
self.assertFalse(os.path.isfile(self.cfile))
|
|
|
|
|
|
# vi: ts=4 expandtab
|