Make the code shorter.
Signed-off-by: Eduardo Habkost <[email protected]>
---
client/tests/kvm/kvm/installer.py | 22 +++++++++++++---------
1 files changed, 13 insertions(+), 9 deletions(-)
diff --git a/client/tests/kvm/kvm/installer.py
b/client/tests/kvm/kvm/installer.py
index 2a9c094..35b041c 100644
--- a/client/tests/kvm/kvm/installer.py
+++ b/client/tests/kvm/kvm/installer.py
@@ -600,18 +600,22 @@ class GitInstaller(SourceDirInstaller):
save_build(self.srcdir, self.results_dir)
+installer_classes = {
+ 'localsrc':SourceDirInstaller,
+ 'localtar':SourceDirInstaller,
+ 'release':SourceDirInstaller,
+ 'snapshot':SourceDirInstaller,
+ 'git':GitInstaller,
+ 'yum':YumInstaller,
+ 'koji':KojiInstaller,
+}
+
def _installer_class(install_mode):
- if install_mode in ['localsrc', 'localtar', 'release', 'snapshot']:
- return SourceDirInstaller
- elif install_mode == 'git':
- return GitInstaller
- elif install_mode == 'yum':
- return YumInstaller
- elif install_mode == 'koji':
- return KojiInstaller
- else:
+ c = installer_classes.get(install_mode)
+ if c is None:
raise error.TestError('Invalid or unsupported'
' install mode: %s' % install_mode)
+ return c
def make_installer(test, params):
mode = params.get("mode")
--
1.7.3.2
_______________________________________________
Autotest mailing list
[email protected]
http://test.kernel.org/cgi-bin/mailman/listinfo/autotest
