Allow generate link for itself

src/ZfrRest/Mvc/Controller/MethodHandler/PostHandler.php

@@ -24,6 +24,7 @@
 use ZfrRest\Http\Exception\Client\MethodNotAllowedException;
 use ZfrRest\Options\ControllerBehavioursOptions;
 use ZfrRest\Resource\ResourceInterface;
+use ZfrRest\View\Model\ResourceModel;
 
 /**
  * Handler for the POST method verb
@@ -99,10 +100,11 @@ public function handleMethod(AbstractController $controller, ResourceInterface $
         $result = $controller->post($data, $singleResource->getMetadata());
 
         // Set the Location header with the URL of the newly created resource
-        if (is_object($result)) {
-            // @TODO: when we have a functional router, we should be able to automatically set the Location
-            //        header with the URI of the newly created resource
-            $controller->getResponse()->setStatusCode(201);
+        if ($result instanceof ResourceModel) {
+            $location = $controller->url()->fromRoute(null, ['resource' => $result->getResource()]);
+
+            $controller->getResponse()->setStatusCode(201)
+                                      ->getHeaders()->addHeaderLine('Location', $location);
         }
 
         return $result;

src/ZfrRest/Router/Http/ResourceGraphRoute.php

@@ -81,9 +81,41 @@ public function __construct(MetadataFactory $metadataFactory, BaseSubPathMatcher
      */
     public function assemble(array $params = [], array $options = [])
     {
-        // @TODO: not sure about how to do this correctly...
+        $trimmedRoute = trim($this->route, '/');
 
-        throw new RuntimeException('ResourceGraphRoute does not support yet assembling route');
+        // If no resource it's equal to "self"
+        if (!isset($params['resource'])) {
+            return '/' . $trimmedRoute;
+        }
+
+        /* @var \ZfrRest\Resource\ResourceInterface $resource */
+        $resource         = $params['resource'];
+        $resourceMetadata = $resource->getMetadata();
+
+        $classMetadata = $resourceMetadata->getClassMetadata();
+        $identifiers   = $classMetadata->getIdentifierValues($resource->getData());
+
+        if (count($identifiers) > 1) {
+            throw new RuntimeException('ZfrRest assembling does not support composite identifiers');
+        }
+
+        $route = '/' . $trimmedRoute . '/' . current($identifiers);
+
+        if (!isset($params['association'])) {
+            return $route;
+        }
+
+        if (!$resourceMetadata->hasAssociationMetadata($params['association'])) {
+            throw new RuntimeException(sprintf(
+                'You are trying to generate a URL for the association "%s", which does not exist or is not exposed',
+                $params['association']
+            ));
+        }
+
+        $associationMetadata = $resourceMetadata->getAssociationMetadata($params['association']);
+        $route               .= '/' . $associationMetadata['path'];
+
+        return $route;
     }
 
     /**

tests/ZfrRestTest/Router/Http/ResourceGraphRouteTest.php

@@ -54,9 +54,80 @@ public function testReturnNullIfNotAHttpRequest()
             $this->metadataFactory,
             $this->baseSubPathMatcher,
             new \stdClass(),
-            'route'
+            '/route'
         );
 
         $this->assertNull($resourceGraphRoute->match($this->getMock('Zend\Stdlib\RequestInterface')));
     }
+
+    public function testCanAssembleWithoutResource()
+    {
+        $resourceGraphRoute = new ResourceGraphRoute(
+            $this->metadataFactory,
+            $this->baseSubPathMatcher,
+            new \stdClass(),
+            '/route'
+        );
+
+        $this->assertEquals('/route', $resourceGraphRoute->assemble());
+    }
+
+    public function testCanAssembleWithResource()
+    {
+        $resourceGraphRoute = new ResourceGraphRoute(
+            $this->metadataFactory,
+            $this->baseSubPathMatcher,
+            new \stdClass(),
+            '/route'
+        );
+
+        $classMetadata = $this->getMock('Doctrine\Common\Persistence\Mapping\ClassMetadata');
+        $classMetadata->expects($this->once())
+                      ->method('getIdentifierValues')
+                      ->will($this->returnValue(['id' => 2]));
+
+        $metadata = $this->getMock('ZfrRest\Resource\Metadata\ResourceMetadataInterface');
+        $metadata->expects($this->once())->method('getClassMetadata')->will($this->returnValue($classMetadata));
+
+        $resource = $this->getMock('ZfrRest\Resource\ResourceInterface');
+        $resource->expects($this->once())->method('getMetadata')->will($this->returnValue($metadata));
+
+        $this->assertEquals('/route/2', $resourceGraphRoute->assemble(['resource' => $resource]));
+    }
+
+    public function testCanAssembleWithResourceAndAssociation()
+    {
+        $resourceGraphRoute = new ResourceGraphRoute(
+            $this->metadataFactory,
+            $this->baseSubPathMatcher,
+            new \stdClass(),
+            '/route'
+        );
+
+        $classMetadata = $this->getMock('Doctrine\Common\Persistence\Mapping\ClassMetadata');
+        $classMetadata->expects($this->once())
+                      ->method('getIdentifierValues')
+                      ->will($this->returnValue(['id' => 2]));
+
+        $metadata = $this->getMock('ZfrRest\Resource\Metadata\ResourceMetadataInterface');
+        $metadata->expects($this->once())->method('getClassMetadata')->will($this->returnValue($classMetadata));
+
+        $metadata->expects($this->once())
+                 ->method('hasAssociationMetadata')
+                 ->with('tweets')
+                 ->will($this->returnValue(true));
+
+        $metadata->expects($this->once())
+                 ->method('getAssociationMetadata')
+                 ->with('tweets')
+                 ->will($this->returnValue(['path' => 'tweets']));
+
+        $resource = $this->getMock('ZfrRest\Resource\ResourceInterface');
+        $resource->expects($this->once())->method('getMetadata')->will($this->returnValue($metadata));
+
+        $this->assertEquals('/route/2/tweets', $resourceGraphRoute->assemble([
+            'resource'    => $resource,
+            'association' => 'tweets'
+        ]));
+    }
 }