Feature: __add__ and __mul__ are implemented for stdout and stderr ob…

…ject proxies
pylover · Jul 27, 2021 · bec70f7 · bec70f7
1 parent 0119e1f
commit bec70f7
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 2 deletions.
diff --git a/bddcli/__init__.py b/bddcli/__init__.py
@@ -2,4 +2,4 @@
 from .authoring import given, when, story, status, stdout, stderr, Given
 from .application import Application
 
-__version__ = '2.7.0'
+__version__ = '2.8.0'
diff --git a/bddcli/proxy.py b/bddcli/proxy.py
@@ -69,3 +69,9 @@ def __len__(self):
 
     def __getitem__(self, key):
         return self.proxied_object.__getitem__(key)
+
+    def __add__(self, other):
+        return self.proxied_object + other
+
+    def __mul__(self, other):
+        return self.proxied_object * other
diff --git a/tests/test_proxy.py b/tests/test_proxy.py
@@ -14,5 +14,6 @@ def test_proxy():
     with Given(app, ['bar']):
         assert stdout == 'foo bar\n'
         assert len(stdout) == 8
-
         assert stdout[:-1] == 'foo bar'
+        assert stdout + 'baz' == 'foo bar\nbaz'
+        assert stdout * 2 == 'foo bar\nfoo bar\n'