OilfieldToolsNet/shlink
4
0
Fork 0
mirror of https://github.com/shlinkio/shlink.git synced 2025-02-25 18:45:27 -06:00
Code Issues Packages Projects Releases Wiki Activity

Ensured CloseDbConnectionMiddleware clears the entity manager

Browse Source
This commit is contained in:
Alejandro Celaya 2018-11-24 13:24:43 +01:00
parent 0c26490e3f
commit f0330e9ae3
2 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
module/Common/src/Middleware/CloseDbConnectionMiddleware.php
View File

@ -27,6 +27,7 @@ class CloseDbConnectionMiddleware implements MiddlewareInterface
{ {
$handledRequest = $handler->handle($request); $handledRequest = $handler->handle($request);
$this->em->getConnection()->close(); $this->em->getConnection()->close();
$this->em->clear();
return $handledRequest; return $handledRequest;
} }

3
module/Common/test/Middleware/CloseDbConnectionMiddlewareTest.php
View File

@ -41,6 +41,8 @@ class CloseDbConnectionMiddlewareTest extends TestCase
$closeConn = $conn->close()->will(function () { $closeConn = $conn->close()->will(function () {
}); });
$getConn = $this->em->getConnection()->willReturn($conn->reveal()); $getConn = $this->em->getConnection()->willReturn($conn->reveal());
$clear = $this->em->clear()->will(function () {
});
$handle = $this->handler->handle($req)->willReturn($resp); $handle = $this->handler->handle($req)->willReturn($resp);
$result = $this->middleware->process($req, $this->handler->reveal()); $result = $this->middleware->process($req, $this->handler->reveal());
@ -48,6 +50,7 @@ class CloseDbConnectionMiddlewareTest extends TestCase
$this->assertSame($result, $resp); $this->assertSame($result, $resp);
$getConn->shouldHaveBeenCalledOnce(); $getConn->shouldHaveBeenCalledOnce();
$closeConn->shouldHaveBeenCalledOnce(); $closeConn->shouldHaveBeenCalledOnce();
$clear->shouldHaveBeenCalledOnce();
$handle->shouldHaveBeenCalledOnce(); $handle->shouldHaveBeenCalledOnce();
} }
} }