[gentoo-commits] proj/g-sorcery:master commit in: g_sorcery/, tests/dummy_backend/, tests/

[Jauhien Piatlicki]

commit:     c9ad38ff53c9dc8baa38a9b888c3e83a7b47dc1b
Author:     Jauhien Piatlicki (jauhien) <piatlicki <AT> gmail <DOT> com>
AuthorDate: Mon Jul  1 21:42:06 2013 +0000
Commit:     Jauhien Piatlicki <piatlicki <AT> gmail <DOT> com>
CommitDate: Mon Jul  1 21:42:06 2013 +0000
URL:        http://git.overlays.gentoo.org/gitweb/?p=proj/g-sorcery.git;a=commit;h=c9ad38ff

g_sorcery/g_sorcery: get_backend

---
 g_sorcery/g_sorcery.py          | 11 ++++++++++-
 tests/dummy_backend/__init__.py |  4 ++++
 tests/dummy_backend/backend.py  |  4 ++++
 tests/test_g_sorcery.py         | 16 ++++++++++++++++
 4 files changed, 34 insertions(+), 1 deletion(-)

diff --git a/g_sorcery/g_sorcery.py b/g_sorcery/g_sorcery.py
index 4b1784a..c815ec9 100644
--- a/g_sorcery/g_sorcery.py
+++ b/g_sorcery/g_sorcery.py
@@ -11,7 +11,7 @@
     :license: GPL-2, see LICENSE for more details.
 """
 
-import os, sys
+import importlib, os, sys
 
 from .fileutils import FileJSON
 
@@ -41,5 +41,14 @@ def main():
             return -1
         pass
 
+def get_backend(package):
+    try:
+        module = importlib.import_module(package + '.backend')
+    except ImportError:
+        return None
+    
+    return module.instance
+
+
 if __name__ == "__main__":
     sys.exit(main())

diff --git a/tests/dummy_backend/__init__.py b/tests/dummy_backend/__init__.py
new file mode 100644
index 0000000..2ad1e71
--- /dev/null
+++ b/tests/dummy_backend/__init__.py
@@ -0,0 +1,4 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+__all__ = ['backend']

diff --git a/tests/dummy_backend/backend.py b/tests/dummy_backend/backend.py
new file mode 100644
index 0000000..b23785c
--- /dev/null
+++ b/tests/dummy_backend/backend.py
@@ -0,0 +1,4 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+instance = 'test'

diff --git a/tests/test_g_sorcery.py b/tests/test_g_sorcery.py
index 7ff9992..ece991d 100644
--- a/tests/test_g_sorcery.py
+++ b/tests/test_g_sorcery.py
@@ -13,6 +13,10 @@
 
 import os, subprocess, tempfile, unittest
 
+from g_sorcery import g_sorcery
+
+from tests.dummy_backend import backend as dummyBackend
+
 class TestBin(unittest.TestCase):
     def setUp(self):
         self.tempdir = tempfile.TemporaryDirectory()
@@ -42,9 +46,21 @@ class TestBin(unittest.TestCase):
         self.assertRaises(subprocess.CalledProcessError, subprocess.check_output, './g-empty')
         os.chdir(prev)
 
+class TestGSorcery(unittest.TestCase):
+    def setUp(self):
+        self.tempdir = tempfile.TemporaryDirectory()
+
+    def tearDown(self):
+        del self.tempdir
+
+    def test_get_backend(self):
+        self.assertEqual(g_sorcery.get_backend('nonexistent_backend'), None)
+        self.assertEqual(g_sorcery.get_backend('dummy_backend'), dummyBackend.instance)
+
 def suite():
     suite = unittest.TestSuite()
     suite.addTest(TestBin('test_g_sorcery'))
     suite.addTest(TestBin('test_nonexistent_backend'))
     suite.addTest(TestBin('test_empty_config'))
+    suite.addTest(TestGSorcery('test_get_backend'))
     return suite