def test_source_list_for_list_returns_existing_local_file_slash(self):
ret = filemod.source_list([self.myfile + '-foo', self.myfile],
'filehash', 'base')
if six.PY3:
self.assertCountEqual(ret, [self.myfile, 'filehash'])
else:
self.assertItemsEqual(ret, [self.myfile, 'filehash'])
def test_source_list_for_list_returns_existing_file(self):
with patch.dict(filemod.__salt__, {'cp.list_master': MagicMock(return_value=['http/httpd.conf.fallback']),
'cp.list_master_dirs': MagicMock(return_value=[])}):
ret = filemod.source_list(['salt://http/httpd.conf',
'salt://http/httpd.conf.fallback'],
'filehash', 'base')
self.assertEqual(list(ret), ['salt://http/httpd.conf.fallback', 'filehash'])
def test_source_list_for_list_returns_existing_local_file_proto(self):
with patch.dict(filemod.__salt__, {'cp.list_master': MagicMock(return_value=[]),
'cp.list_master_dirs': MagicMock(return_value=[])}):
ret = filemod.source_list(['file://' + self.myfile + '-foo',
'file://' + self.myfile],
'filehash', 'base')
self.assertEqual(list(ret), ['file://' + self.myfile, 'filehash'])
def test_source_list_for_list_returns_local_file_slash_from_dict(self):
with patch.dict(
filemod.__salt__, {
'cp.list_master': MagicMock(return_value=[]),
'cp.list_master_dirs': MagicMock(return_value=[])
}):
ret = filemod.source_list([{self.myfile: ''}], 'filehash', 'base')
self.assertEqual(list(ret), [self.myfile, 'filehash'])
def test_source_list_for_list_returns_file_from_dict_via_http(self, remove):
remove.return_value = None
with patch.dict(filemod.__salt__, {'cp.list_master': MagicMock(return_value=[]),
'cp.list_master_dirs': MagicMock(return_value=[]),
'cp.cache_file': MagicMock(return_value='/tmp/http.conf')}):
ret = filemod.source_list(
[{'http://t.est.com/http/httpd.conf': 'filehash'}], '', 'base')
self.assertEqual(list(ret), ['http://t.est.com/http/httpd.conf', 'filehash'])
def test_source_list_for_list_returns_local_file_proto_from_dict(self):
ret = filemod.source_list([{
'file://' + self.myfile: ''
}], 'filehash', 'base')
if six.PY3:
self.assertCountEqual(ret, ['file://' + self.myfile, 'filehash'])
else:
self.assertItemsEqual(ret, ['file://' + self.myfile, 'filehash'])
def test_source_list_for_list_returns_file_from_dict(self):
filemod.__salt__ = {
'cp.list_master': MagicMock(return_value=['http/httpd.conf']),
'cp.list_master_dirs': MagicMock(return_value=[]),
}
ret = filemod.source_list(
[{'salt://http/httpd.conf': ''}], 'filehash', 'base')
self.assertItemsEqual(ret, ['salt://http/httpd.conf', 'filehash'])
def test_source_list_for_list_returns_file_from_dict(self):
filemod.__salt__ = {
'cp.list_master': MagicMock(return_value=['http/httpd.conf']),
'cp.list_master_dirs': MagicMock(return_value=[]),
}
ret = filemod.source_list(
[{'salt://http/httpd.conf': ''}], 'filehash', 'base')
self.assertItemsEqual(ret, ['salt://http/httpd.conf', 'filehash'])
def test_source_list_for_list_returns_local_file_slash_from_dict(myfile):
with patch.dict(
filemod.__salt__,
{
"cp.list_master": MagicMock(return_value=[]),
"cp.list_master_dirs": MagicMock(return_value=[]),
},
):
ret = filemod.source_list([{myfile: ""}], "filehash", "base")
assert list(ret) == [myfile, "filehash"]
def test_source_list_for_list_returns_file_from_dict():
with patch.dict(
filemod.__salt__,
{
"cp.list_master": MagicMock(return_value=["http/httpd.conf"]),
"cp.list_master_dirs": MagicMock(return_value=[]),
},
):
ret = filemod.source_list([{"salt://http/httpd.conf": ""}], "filehash", "base")
assert list(ret) == ["salt://http/httpd.conf", "filehash"]
def test_source_list_for_list_returns_file_from_dict_via_http(self, remove):
remove.return_value = None
filemod.__salt__ = {
'cp.list_master': MagicMock(return_value=[]),
'cp.list_master_dirs': MagicMock(return_value=[]),
'cp.get_url': MagicMock(return_value='/tmp/http.conf'),
}
ret = filemod.source_list(
[{'http://t.est.com/http/httpd.conf': 'filehash'}], '', 'base')
self.assertItemsEqual(ret, ['http://t.est.com/http/httpd.conf',
'filehash'])
def test_source_list_for_list_returns_file_from_other_env(self):
def list_master(env):
dct = {'base': [], 'dev': ['http/httpd.conf']}
return dct[env]
with patch.dict(filemod.__salt__, {'cp.list_master': MagicMock(side_effect=list_master),
'cp.list_master_dirs': MagicMock(return_value=[])}):
ret = filemod.source_list(['salt://http/httpd.conf?saltenv=dev',
'salt://http/httpd.conf.fallback'],
'filehash', 'base')
self.assertEqual(list(ret), ['salt://http/httpd.conf?saltenv=dev', 'filehash'])
def test_source_list_for_list_returns_file_from_dict_via_http(self, remove):
remove.return_value = None
filemod.__salt__ = {
'cp.list_master': MagicMock(return_value=[]),
'cp.list_master_dirs': MagicMock(return_value=[]),
'cp.get_url': MagicMock(return_value='/tmp/http.conf'),
}
ret = filemod.source_list(
[{'http://t.est.com/http/httpd.conf': 'filehash'}], '', 'base')
self.assertItemsEqual(ret, ['http://t.est.com/http/httpd.conf',
'filehash'])
def test_source_list_for_list_returns_existing_file(self):
filemod.__salt__ = {
'cp.list_master': MagicMock(
return_value=['http/httpd.conf.fallback']),
'cp.list_master_dirs': MagicMock(return_value=[]),
}
ret = filemod.source_list(['salt://http/httpd.conf',
'salt://http/httpd.conf.fallback'],
'filehash', 'base')
self.assertItemsEqual(ret, ['salt://http/httpd.conf.fallback',
'filehash'])
def test_source_list_for_list_returns_existing_file(self):
filemod.__salt__ = {
'cp.list_master':
MagicMock(return_value=['http/httpd.conf.fallback']),
'cp.list_master_dirs': MagicMock(return_value=[]),
}
ret = filemod.source_list(
['salt://http/httpd.conf', 'salt://http/httpd.conf.fallback'],
'filehash', 'base')
self.assertItemsEqual(ret,
['salt://http/httpd.conf.fallback', 'filehash'])
def test_source_list_for_list_returns_file_from_other_env(self):
def list_master(env):
dct = {'base': [], 'dev': ['http/httpd.conf']}
return dct[env]
filemod.__salt__ = {
'cp.list_master': MagicMock(side_effect=list_master),
'cp.list_master_dirs': MagicMock(return_value=[]),
}
ret = filemod.source_list(['salt://http/httpd.conf?env=dev',
'salt://http/httpd.conf.fallback'],
'filehash', 'base')
self.assertItemsEqual(ret, ['salt://http/httpd.conf?env=dev',
'filehash'])
def test_source_list_for_list_returns_existing_file():
with patch.dict(
filemod.__salt__,
{
"cp.list_master": MagicMock(return_value=["http/httpd.conf.fallback"]),
"cp.list_master_dirs": MagicMock(return_value=[]),
},
):
ret = filemod.source_list(
["salt://http/httpd.conf", "salt://http/httpd.conf.fallback"],
"filehash",
"base",
)
assert list(ret) == ["salt://http/httpd.conf.fallback", "filehash"]
def test_source_list_for_list_returns_existing_local_file_proto(myfile):
with patch.dict(
filemod.__salt__,
{
"cp.list_master": MagicMock(return_value=[]),
"cp.list_master_dirs": MagicMock(return_value=[]),
},
):
ret = filemod.source_list(
["file://" + myfile + "-foo", "file://" + myfile],
"filehash",
"base",
)
assert list(ret) == ["file://" + myfile, "filehash"]
def test_source_list_for_list_returns_file_from_dict_via_http():
with patch("salt.modules.file.os.remove") as remove:
remove.return_value = None
with patch.dict(
filemod.__salt__,
{
"cp.list_master": MagicMock(return_value=[]),
"cp.list_master_dirs": MagicMock(return_value=[]),
"cp.cache_file": MagicMock(return_value="/tmp/http.conf"),
},
):
with patch("salt.utils.http.query") as http_query:
http_query.return_value = {}
ret = filemod.source_list(
[{"http://t.est.com/http/httpd.conf": "filehash"}], "", "base"
)
assert list(ret) == ["http://t.est.com/http/httpd.conf", "filehash"]
def test_source_list_for_list_returns_file_from_other_env(self):
def list_master(env):
dct = {'base': [], 'dev': ['http/httpd.conf']}
return dct[env]
filemod.__salt__ = {
'cp.list_master': MagicMock(side_effect=list_master),
'cp.list_master_dirs': MagicMock(return_value=[]),
}
filemod.__context__ = {}
ret = filemod.source_list([
'salt://http/httpd.conf?saltenv=dev',
'salt://http/httpd.conf.fallback'
], 'filehash', 'base')
if six.PY3:
self.assertCountEqual(
ret, ['salt://http/httpd.conf?saltenv=dev', 'filehash'])
else:
self.assertItemsEqual(
ret, ['salt://http/httpd.conf?saltenv=dev', 'filehash'])
def test_source_list_for_list_returns_file_from_other_env():
def list_master(env):
dct = {"base": [], "dev": ["http/httpd.conf"]}
return dct[env]
with patch.dict(
filemod.__salt__,
{
"cp.list_master": MagicMock(side_effect=list_master),
"cp.list_master_dirs": MagicMock(return_value=[]),
},
):
ret = filemod.source_list(
[
"salt://http/httpd.conf?saltenv=dev",
"salt://http/httpd.conf.fallback",
],
"filehash",
"base",
)
assert list(ret) == ["salt://http/httpd.conf?saltenv=dev", "filehash"]
def test_source_list_use_requests():
with patch("salt.modules.file.os.remove") as remove:
remove.return_value = None
with patch.dict(
filemod.__salt__,
{
"cp.list_master": MagicMock(return_value=[]),
"cp.list_master_dirs": MagicMock(return_value=[]),
"cp.cache_file": MagicMock(return_value="/tmp/http.conf"),
},
):
expected_call = call(
"http://t.est.com/http/file1",
decode_body=False,
method="HEAD",
)
with patch("salt.utils.http.query",
MagicMock(return_value={})) as http_query:
ret = filemod.source_list(
[{
"http://t.est.com/http/file1": "filehash"
}], "", "base")
assert list(ret) == ["http://t.est.com/http/file1", "filehash"]
assert expected_call in http_query.mock_calls
def test_source_list_for_list_returns_local_file_slash_from_dict(self):
ret = filemod.source_list([{self.myfile: ''}], 'filehash', 'base')
self.assertItemsEqual(ret, [self.myfile, 'filehash'])
def test_source_list_for_list_returns_existing_local_file_proto(self):
ret = filemod.source_list(['file://' + self.myfile + '-foo',
'file://' + self.myfile],
'filehash', 'base')
self.assertItemsEqual(ret, ['file://' + self.myfile, 'filehash'])
def test_source_list_for_list_returns_local_file_proto_from_dict(self):
ret = filemod.source_list(
[{'file://' + self.myfile: ''}], 'filehash', 'base')
self.assertItemsEqual(ret, ['file://' + self.myfile, 'filehash'])
def test_source_list_for_list_returns_local_file_proto_from_dict(self):
ret = filemod.source_list([{
'file://' + self.myfile: ''
}], 'filehash', 'base')
self.assertEqual(list(ret), ['file://' + self.myfile, 'filehash'])
def test_source_list_for_list_returns_existing_local_file_proto(self):
ret = filemod.source_list(
['file://' + self.myfile + '-foo', 'file://' + self.myfile],
'filehash', 'base')
self.assertEqual(list(ret), ['file://' + self.myfile, 'filehash'])
