WithContext

src/ContextAwareTrait.php

@@ -35,6 +35,19 @@ protected function _bindContext(ContextAware $to)
     return $this;
   }
 
+  /**
+   * Bind context to a context aware object, and return it
+   *
+   * @param ContextAware $to
+   *
+   * @return ContextAware
+   */
+  protected function _withContext(ContextAware $to)
+  {
+    $to->setContext($this->getContext());
+    return $to;
+  }
+
   /**
    * @return Context
    */

tests/ContextTest.php

@@ -63,8 +63,15 @@ public function testContextAware()
 
     $obj2->setContext($ctx);
     $obj2->bind($obj);
-    $this->assertTrue($obj->hasContext());
+    $this->assertTrue($obj2->hasContext());
     $this->assertSame($ctx, $obj->getContext());
+
+    $obj3 = new TestContextAwareObject();
+    $obj4 = $obj2->with($obj3);
+    $this->assertTrue($obj3->hasContext());
+    $this->assertSame($ctx, $obj3->getContext());
+    $this->assertSame($obj3, $obj4);
+    $this->assertSame($ctx, $obj4->getContext());
   }
 
   public function testConfig()

tests/Supporting/TestContextAwareObject.php

@@ -12,4 +12,9 @@ public function bind(ContextAware $to)
   {
     return $this->_bindContext($to);
   }
+
+  public function with(ContextAware $to)
+  {
+    return $this->_withContext($to);
+  }
 }