Merge pull request DesignPatternsPHP#322 from carousel/master

enhance delegation example

Author: Dominik Liebler

More/Delegation/JuniorDeveloper.php

@@ -8,4 +8,12 @@ public function writeBadCode(): string
     {
         return 'Some junior developer generated code...';
     }
+
+    /**
+     * Junior is authorized to call method on TeamLead (real delegation)
+     */
+    public function writeReallyBadCode(TeamLead $teamLead): string
+    {
+        return $teamLead->writeReallyBadCode();
+    }
 }

More/Delegation/TeamLead.php

@@ -21,4 +21,18 @@ public function writeCode(): string
     {
         return $this->junior->writeBadCode();
     }
+
+    public function writeBadCode(): string
+    {
+        //note that we are passing $this from teamLead context
+        return $this->junior->writeReallyBadCode($this);
+    }
+
+    /**
+     * Junior can call this method
+     */
+    public function writeReallyBadCode(): string
+    {
+        return 'Even team lead can write bad code...';
+    }
 }

More/Delegation/Tests/DelegationTest.php

@@ -7,11 +7,19 @@
 
 class DelegationTest extends TestCase
 {
-    public function testHowTeamLeadWriteCode()
+    public function testTeamLeadCanBlameJuniorForBadCode()
     {
         $junior = new Delegation\JuniorDeveloper();
         $teamLead = new Delegation\TeamLead($junior);
 
         $this->assertEquals($junior->writeBadCode(), $teamLead->writeCode());
     }
+
+    public function testTeamLeadCanWriteBadCode()
+    {
+        $junior = new Delegation\JuniorDeveloper();
+        $teamLead = new Delegation\TeamLead($junior);
+
+        $this->assertEquals($junior->writeReallyBadCode($teamLead), $teamLead->writeBadCode());
+    }
 }