175 lines
5.2 KiB
PHP
175 lines
5.2 KiB
PHP
|
<?php
|
||
|
/**
|
||
|
* @author Lukas Reschke <lukas@owncloud.com>
|
||
|
*
|
||
|
* @copyright Copyright (c) 2015, ownCloud, Inc.
|
||
|
* @license AGPL-3.0
|
||
|
*
|
||
|
* This code is free software: you can redistribute it and/or modify
|
||
|
* it under the terms of the GNU Affero General Public License, version 3,
|
||
|
* as published by the Free Software Foundation.
|
||
|
*
|
||
|
* This program is distributed in the hope that it will be useful,
|
||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||
|
* GNU Affero General Public License for more details.
|
||
|
*
|
||
|
* You should have received a copy of the GNU Affero General Public License, version 3,
|
||
|
* along with this program. If not, see <http://www.gnu.org/licenses/>
|
||
|
*
|
||
|
*/
|
||
|
|
||
|
namespace OC\Settings\Controller;
|
||
|
|
||
|
use OCP\AppFramework\Http;
|
||
|
use OCP\AppFramework\Http\DataResponse;
|
||
|
use OCP\IRequest;
|
||
|
use OCP\IL10N;
|
||
|
use OCP\ICertificateManager;
|
||
|
|
||
|
/**
|
||
|
* Class CertificateControllerTest
|
||
|
*
|
||
|
* @package OC\Settings\Controller
|
||
|
*/
|
||
|
class CertificateControllerTest extends \Test\TestCase {
|
||
|
/** @var CertificateController */
|
||
|
private $certificateController;
|
||
|
/** @var IRequest */
|
||
|
private $request;
|
||
|
/** @var ICertificateManager */
|
||
|
private $certificateManager;
|
||
|
/** @var IL10N */
|
||
|
private $l10n;
|
||
|
|
||
|
public function setUp() {
|
||
|
parent::setUp();
|
||
|
|
||
|
$this->request = $this->getMock('\OCP\IRequest');
|
||
|
$this->certificateManager = $this->getMock('\OCP\ICertificateManager');
|
||
|
$this->l10n = $this->getMock('\OCP\IL10N');
|
||
|
|
||
|
$this->certificateController = new CertificateController(
|
||
|
'settings',
|
||
|
$this->request,
|
||
|
$this->certificateManager,
|
||
|
$this->l10n
|
||
|
);
|
||
|
}
|
||
|
|
||
|
public function testAddPersonalRootCertificateWithEmptyFile() {
|
||
|
$this->request
|
||
|
->expects($this->once())
|
||
|
->method('getUploadedFile')
|
||
|
->with('rootcert_import')
|
||
|
->will($this->returnValue(null));
|
||
|
|
||
|
$expected = new DataResponse(['message' => 'No file uploaded'], Http::STATUS_UNPROCESSABLE_ENTITY);
|
||
|
$this->assertEquals($expected, $this->certificateController->addPersonalRootCertificate());
|
||
|
}
|
||
|
|
||
|
public function testAddPersonalRootCertificateValidCertificate() {
|
||
|
$uploadedFile = [
|
||
|
'tmp_name' => __DIR__ . '/../../data/certificates/goodCertificate.crt',
|
||
|
'name' => 'goodCertificate.crt',
|
||
|
];
|
||
|
|
||
|
$certificate = $this->getMock('\OCP\ICertificate');
|
||
|
$certificate
|
||
|
->expects($this->once())
|
||
|
->method('getName')
|
||
|
->will($this->returnValue('Name'));
|
||
|
$certificate
|
||
|
->expects($this->once())
|
||
|
->method('getCommonName')
|
||
|
->will($this->returnValue('CommonName'));
|
||
|
$certificate
|
||
|
->expects($this->once())
|
||
|
->method('getOrganization')
|
||
|
->will($this->returnValue('Organization'));
|
||
|
$certificate
|
||
|
->expects($this->exactly(2))
|
||
|
->method('getIssueDate')
|
||
|
->will($this->returnValue(new \DateTime('@1429099555')));
|
||
|
$certificate
|
||
|
->expects($this->exactly(2))
|
||
|
->method('getExpireDate')
|
||
|
->will($this->returnValue(new \DateTime('@1529099555')));
|
||
|
$certificate
|
||
|
->expects($this->once())
|
||
|
->method('getIssuerName')
|
||
|
->will($this->returnValue('Issuer'));
|
||
|
$certificate
|
||
|
->expects($this->once())
|
||
|
->method('getIssuerOrganization')
|
||
|
->will($this->returnValue('IssuerOrganization'));
|
||
|
|
||
|
$this->request
|
||
|
->expects($this->once())
|
||
|
->method('getUploadedFile')
|
||
|
->with('rootcert_import')
|
||
|
->will($this->returnValue($uploadedFile));
|
||
|
$this->certificateManager
|
||
|
->expects($this->once())
|
||
|
->method('addCertificate')
|
||
|
->with(file_get_contents($uploadedFile['tmp_name'], 'goodCertificate.crt'))
|
||
|
->will($this->returnValue($certificate));
|
||
|
|
||
|
$this->l10n
|
||
|
->expects($this->at(0))
|
||
|
->method('l')
|
||
|
->with('date', new \DateTime('@1429099555'))
|
||
|
->will($this->returnValue('Valid From as String'));
|
||
|
$this->l10n
|
||
|
->expects($this->at(1))
|
||
|
->method('l')
|
||
|
->with('date', new \DateTime('@1529099555'))
|
||
|
->will($this->returnValue('Valid Till as String'));
|
||
|
|
||
|
|
||
|
$expected = new DataResponse([
|
||
|
'name' => 'Name',
|
||
|
'commonName' => 'CommonName',
|
||
|
'organization' => 'Organization',
|
||
|
'validFrom' => 1429099555,
|
||
|
'validTill' => 1529099555,
|
||
|
'validFromString' => 'Valid From as String',
|
||
|
'validTillString' => 'Valid Till as String',
|
||
|
'issuer' => 'Issuer',
|
||
|
'issuerOrganization' => 'IssuerOrganization',
|
||
|
]);
|
||
|
$this->assertEquals($expected, $this->certificateController->addPersonalRootCertificate());
|
||
|
}
|
||
|
|
||
|
public function testAddPersonalRootCertificateInValidCertificate() {
|
||
|
$uploadedFile = [
|
||
|
'tmp_name' => __DIR__ . '/../../data/certificates/badCertificate.crt',
|
||
|
'name' => 'badCertificate.crt',
|
||
|
];
|
||
|
|
||
|
$this->request
|
||
|
->expects($this->once())
|
||
|
->method('getUploadedFile')
|
||
|
->with('rootcert_import')
|
||
|
->will($this->returnValue($uploadedFile));
|
||
|
$this->certificateManager
|
||
|
->expects($this->once())
|
||
|
->method('addCertificate')
|
||
|
->with(file_get_contents($uploadedFile['tmp_name'], 'goodCertificate.crt'))
|
||
|
->will($this->throwException(new \Exception()));
|
||
|
|
||
|
$expected = new DataResponse('An error occurred.', Http::STATUS_INTERNAL_SERVER_ERROR);
|
||
|
$this->assertEquals($expected, $this->certificateController->addPersonalRootCertificate());
|
||
|
}
|
||
|
|
||
|
public function testRemoveCertificate() {
|
||
|
$this->certificateManager
|
||
|
->expects($this->once())
|
||
|
->method('removeCertificate')
|
||
|
->with('CertificateToRemove');
|
||
|
|
||
|
$this->assertEquals(new DataResponse(), $this->certificateController->removePersonalRootCertificate('CertificateToRemove'));
|
||
|
}
|
||
|
|
||
|
}
|