From 6ef25deac9a624f115459e3fbc7cdfb2320d1b2a Mon Sep 17 00:00:00 2001
From: Romain Dorgueil <romain@dorgueil.net>
Date: Wed, 5 Jul 2017 12:01:37 +0200
Subject: [PATCH] [config] Adds test for requires() decorator.

---
 bonobo/errors.py              | 11 +++++++----
 tests/config/test_services.py | 22 +++++++++++++++++++++-
 2 files changed, 28 insertions(+), 5 deletions(-)

diff --git a/bonobo/errors.py b/bonobo/errors.py
index 8510a50..08b97d4 100644
--- a/bonobo/errors.py
+++ b/bonobo/errors.py
@@ -58,19 +58,22 @@ class ConfigurationError(Exception):
     pass
 
 
-class MissingServiceImplementationError(KeyError):
-    pass
-
-
 class UnrecoverableError(Exception):
     """Flag for errors that must interrupt the workflow, either because they will happen for sure on each node run, or
     because you know that your transformation has no point continuing runnning after a bad event."""
 
+
 class UnrecoverableValueError(UnrecoverableError, ValueError):
     pass
 
+
 class UnrecoverableRuntimeError(UnrecoverableError, RuntimeError):
     pass
 
+
 class UnrecoverableNotImplementedError(UnrecoverableError, NotImplementedError):
     pass
+
+
+class MissingServiceImplementationError(UnrecoverableError, KeyError):
+    pass
diff --git a/tests/config/test_services.py b/tests/config/test_services.py
index b762dbe..ff81e82 100644
--- a/tests/config/test_services.py
+++ b/tests/config/test_services.py
@@ -3,7 +3,7 @@ import time
 
 import pytest
 
-from bonobo.config import Configurable, Container, Exclusive, Service
+from bonobo.config import Configurable, Container, Exclusive, Service, requires
 from bonobo.config.services import validate_service_name
 
 
@@ -94,3 +94,23 @@ def test_exclusive():
         'hello', '0 0', '0 1', '0 2', '0 3', '0 4', '1 0', '1 1', '1 2', '1 3', '1 4', '2 0', '2 1', '2 2', '2 3',
         '2 4', '3 0', '3 1', '3 2', '3 3', '3 4', '4 0', '4 1', '4 2', '4 3', '4 4'
     ]
+
+
+def test_requires():
+    vcr = VCR()
+
+    services = Container(
+        output=vcr.append
+    )
+
+    @requires('output')
+    def append(out, x):
+        out(x)
+
+    svcargs = services.args_for(append)
+    assert len(svcargs) == 1
+    assert svcargs[0] == vcr.append
+
+
+
+