mirror of
https://github.com/nextcloud/server.git
synced 2025-01-15 23:59:16 +00:00
101 lines
2.6 KiB
PHP
101 lines
2.6 KiB
PHP
<?php
|
|
|
|
declare(strict_types=1);
|
|
|
|
/**
|
|
* SPDX-FileCopyrightText: 2024 Nextcloud GmbH and Nextcloud contributors
|
|
* SPDX-License-Identifier: AGPL-3.0-or-later
|
|
*/
|
|
|
|
namespace Test\DB;
|
|
|
|
use Doctrine\DBAL\Configuration;
|
|
use Doctrine\DBAL\Driver;
|
|
use Doctrine\DBAL\Driver\Connection as DriverConnection;
|
|
use Doctrine\DBAL\Platforms\MySQLPlatform;
|
|
use OC\DB\Adapter;
|
|
use OC\DB\Connection;
|
|
use Test\TestCase;
|
|
|
|
/**
|
|
* @group DB
|
|
*/
|
|
class ConnectionTest extends TestCase {
|
|
|
|
public function testSingleNodeConnectsToPrimaryOnly(): void {
|
|
$connectionParams = [
|
|
'user' => 'test',
|
|
'password' => 'topsecret',
|
|
'host' => 'test',
|
|
];
|
|
$adapter = $this->createMock(Adapter::class);
|
|
$driver = $this->createMock(Driver::class);
|
|
$configuration = $this->createMock(Configuration::class);
|
|
$connection = $this->getMockBuilder(Connection::class)
|
|
->onlyMethods(['connectTo'])
|
|
->setConstructorArgs([
|
|
[
|
|
'adapter' => $adapter,
|
|
'platform' => new MySQLPlatform(),
|
|
'tablePrefix' => 'nctest',
|
|
'primary' => $connectionParams,
|
|
'replica' => [
|
|
$connectionParams,
|
|
],
|
|
],
|
|
$driver,
|
|
$configuration,
|
|
])
|
|
->getMock();
|
|
$driverConnection = $this->createMock(DriverConnection::class);
|
|
$connection->expects(self::once())
|
|
->method('connectTo')
|
|
->with('primary')
|
|
->willReturn($driverConnection);
|
|
|
|
$connection->ensureConnectedToReplica();
|
|
$connection->ensureConnectedToPrimary();
|
|
$connection->ensureConnectedToReplica();
|
|
}
|
|
|
|
public function testClusterConnectsToPrimaryAndReplica(): void {
|
|
$connectionParamsPrimary = [
|
|
'user' => 'test',
|
|
'password' => 'topsecret',
|
|
'host' => 'testprimary',
|
|
];
|
|
$connectionParamsReplica = [
|
|
'user' => 'test',
|
|
'password' => 'topsecret',
|
|
'host' => 'testreplica',
|
|
];
|
|
$adapter = $this->createMock(Adapter::class);
|
|
$driver = $this->createMock(Driver::class);
|
|
$configuration = $this->createMock(Configuration::class);
|
|
$connection = $this->getMockBuilder(Connection::class)
|
|
->onlyMethods(['connectTo'])
|
|
->setConstructorArgs([
|
|
[
|
|
'adapter' => $adapter,
|
|
'platform' => new MySQLPlatform(),
|
|
'tablePrefix' => 'nctest',
|
|
'primary' => $connectionParamsPrimary,
|
|
'replica' => [
|
|
$connectionParamsReplica,
|
|
],
|
|
],
|
|
$driver,
|
|
$configuration,
|
|
])
|
|
->getMock();
|
|
$driverConnection = $this->createMock(DriverConnection::class);
|
|
$connection->expects(self::exactly(2))
|
|
->method('connectTo')
|
|
->willReturn($driverConnection);
|
|
|
|
$connection->ensureConnectedToReplica();
|
|
$connection->ensureConnectedToPrimary();
|
|
$connection->ensureConnectedToReplica();
|
|
}
|
|
|
|
}
|