Select backend based on backend name

redkyn · Feb 12, 2018 · 43bad19 · 43bad19
1 parent d5a0e13
commit 43bad19
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/assigner/backends/decorators.py b/assigner/backends/decorators.py
@@ -19,7 +19,7 @@ def wrapper(config, cmdargs, *args, **kwargs):
             )
             return func(config, GitlabBackend, cmdargs, *args, **kwargs)
 
-        if config.backend == "gitlab":
+        if config.backend["name"] == "gitlab":
             return func(config, GitlabBackend, cmdargs, *args, **kwargs)
         return func(config, MockBackend, cmdargs, *args, **kwargs)
     return wrapper
diff --git a/assigner/tests/utils.py b/assigner/tests/utils.py
@@ -46,4 +46,8 @@ def setUp(self):
         )
 
         self.mock_config = self.mock_config_class.return_value.__enter__.return_value
-        self.mock_config.backend = "gitlab"
+        self.mock_config.backend = {
+            "name": "gitlab",
+            "host": " xxx gitlab host xxx ",
+            "token": " xxx gitlab token xxx ",
+        }