Add more tests on ObjectDefinition

PHP-DI · Jan 4, 2017 · f24d749 · f24d749
1 parent c2cee94
commit f24d749
Showing 1 changed file with 27 additions and 0 deletions.
diff --git a/tests/UnitTest/Definition/ObjectDefinitionTest.php b/tests/UnitTest/Definition/ObjectDefinitionTest.php
@@ -4,6 +4,8 @@
 
 use DI\Definition\CacheableDefinition;
 use DI\Definition\ObjectDefinition;
+use DI\Definition\ObjectDefinition\MethodInjection;
+use DI\Definition\ObjectDefinition\PropertyInjection;
 use DI\Scope;
 use DI\Test\UnitTest\Definition\Fixture\NonInstantiableClass;
 
@@ -58,6 +60,31 @@ public function test_is_instantiable()
         self::assertTrue($definition->classExists());
     }
 
+    public function test_property_injections()
+    {
+        $definition = new ObjectDefinition('foo');
+        $definition->addPropertyInjection(new PropertyInjection('property1', 'Property1'));
+        $definition->addPropertyInjection(new PropertyInjection('property2', 'Property2'));
+        $this->assertEquals([
+            'property1' => new PropertyInjection('property1', 'Property1'),
+            'property2' => new PropertyInjection('property2', 'Property2'),
+        ], $definition->getPropertyInjections());
+    }
+
+    public function test_method_injections()
+    {
+        $definition = new ObjectDefinition('foo');
+        $definition->setConstructorInjection(MethodInjection::constructor());
+        $definition->addMethodInjection(new MethodInjection('method1', ['foo']));
+        $definition->addMethodInjection(new MethodInjection('method2'));
+
+        $this->assertNotNull($definition->getConstructorInjection());
+        $this->assertEquals([
+            new MethodInjection('method1', ['foo']),
+            new MethodInjection('method2'),
+        ], $definition->getMethodInjections());
+    }
+
     /**
      * @test
      */