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_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_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'])