Fixed method define does not works when the class has been resolved. (#2695)

* Fixed method `define` does not works when the class has been resolved.

* Added test cases.

Author: limingxinleo

src/Container.php

@@ -100,7 +100,7 @@ public function set(string $name, $entry)
      */
     public function define(string $name, $definition)
     {
-        $this->definitionSource->addDefinition($name, $definition);
+        $this->setDefinition($name, $definition);
     }
 
     /**
@@ -153,7 +153,10 @@ public function getDefinitionSource(): Definition\DefinitionSourceInterface
         return $this->definitionSource;
     }
 
-    protected function setDefinition(string $name, DefinitionInterface $definition): void
+    /**
+     * @param array|callable|string $definition
+     */
+    private function setDefinition(string $name, $definition): void
     {
         // Clear existing entry if it exists
         if (array_key_exists($name, $this->resolvedEntries)) {

tests/ContainerTest.php

@@ -13,8 +13,10 @@
 
 use Hyperf\Di\Container;
 use Hyperf\Di\Definition\DefinitionSource;
+use HyperfTest\Di\Stub\Bar;
 use HyperfTest\Di\Stub\Foo;
 use HyperfTest\Di\Stub\FooInterface;
+use Mockery;
 use PHPUnit\Framework\TestCase;
 
 /**
@@ -23,6 +25,11 @@
  */
 class ContainerTest extends TestCase
 {
+    protected function tearDown(): void
+    {
+        Mockery::close();
+    }
+
     public function testHas()
     {
         $container = new Container(new DefinitionSource([]));
@@ -44,5 +51,10 @@ public function testDefine()
         $container = new Container(new DefinitionSource([]));
         $container->define(FooInterface::class, Foo::class);
         $this->assertInstanceOf(Foo::class, $container->make(FooInterface::class));
+
+        $container->define(FooInterface::class, function () {
+            return Mockery::mock(Bar::class);
+        });
+        $this->assertInstanceOf(Bar::class, $foo = $container->make(FooInterface::class));
     }
 }