mirror of
https://github.com/tiennm99/coolify.git
synced 2026-04-17 21:20:29 +00:00
feat(proxy): enhance proxy configuration regeneration by extracting custom commands
- Added a new function to extract custom proxy commands from existing Traefik configurations before regenerating the proxy configuration. - Updated the proxy configuration generation logic to include these custom commands, ensuring they are preserved during regeneration. - Introduced unit tests to validate the extraction of custom commands and handle various scenarios, including invalid YAML and different proxy types.
This commit is contained in:
137
tests/Unit/ProxyCustomCommandsTest.php
Normal file
137
tests/Unit/ProxyCustomCommandsTest.php
Normal file
@@ -0,0 +1,137 @@
|
||||
<?php
|
||||
|
||||
use App\Enums\ProxyTypes;
|
||||
use Symfony\Component\Yaml\Yaml;
|
||||
|
||||
it('extracts custom proxy commands from existing traefik configuration', function () {
|
||||
// Create a sample config with custom trustedIPs commands
|
||||
$existingConfig = [
|
||||
'services' => [
|
||||
'traefik' => [
|
||||
'command' => [
|
||||
'--ping=true',
|
||||
'--api.dashboard=true',
|
||||
'--entrypoints.http.address=:80',
|
||||
'--entrypoints.https.address=:443',
|
||||
'--entrypoints.http.forwardedHeaders.trustedIPs=173.245.48.0/20,103.21.244.0/22',
|
||||
'--entrypoints.https.forwardedHeaders.trustedIPs=173.245.48.0/20,103.21.244.0/22',
|
||||
'--providers.docker=true',
|
||||
'--providers.docker.exposedbydefault=false',
|
||||
],
|
||||
],
|
||||
],
|
||||
];
|
||||
|
||||
$yamlConfig = Yaml::dump($existingConfig);
|
||||
|
||||
// Mock a server with Traefik proxy type
|
||||
$server = Mockery::mock('App\Models\Server');
|
||||
$server->shouldReceive('proxyType')->andReturn(ProxyTypes::TRAEFIK->value);
|
||||
|
||||
$customCommands = extractCustomProxyCommands($server, $yamlConfig);
|
||||
|
||||
expect($customCommands)
|
||||
->toBeArray()
|
||||
->toHaveCount(2)
|
||||
->toContain('--entrypoints.http.forwardedHeaders.trustedIPs=173.245.48.0/20,103.21.244.0/22')
|
||||
->toContain('--entrypoints.https.forwardedHeaders.trustedIPs=173.245.48.0/20,103.21.244.0/22');
|
||||
});
|
||||
|
||||
it('returns empty array when only default commands exist', function () {
|
||||
// Config with only default commands
|
||||
$existingConfig = [
|
||||
'services' => [
|
||||
'traefik' => [
|
||||
'command' => [
|
||||
'--ping=true',
|
||||
'--api.dashboard=true',
|
||||
'--entrypoints.http.address=:80',
|
||||
'--entrypoints.https.address=:443',
|
||||
'--providers.docker=true',
|
||||
'--providers.docker.exposedbydefault=false',
|
||||
],
|
||||
],
|
||||
],
|
||||
];
|
||||
|
||||
$yamlConfig = Yaml::dump($existingConfig);
|
||||
|
||||
$server = Mockery::mock('App\Models\Server');
|
||||
$server->shouldReceive('proxyType')->andReturn(ProxyTypes::TRAEFIK->value);
|
||||
|
||||
$customCommands = extractCustomProxyCommands($server, $yamlConfig);
|
||||
|
||||
expect($customCommands)->toBeArray()->toBeEmpty();
|
||||
});
|
||||
|
||||
it('handles invalid yaml gracefully', function () {
|
||||
$invalidYaml = 'this is not: valid: yaml::: content';
|
||||
|
||||
$server = Mockery::mock('App\Models\Server');
|
||||
$server->shouldReceive('proxyType')->andReturn(ProxyTypes::TRAEFIK->value);
|
||||
|
||||
$customCommands = extractCustomProxyCommands($server, $invalidYaml);
|
||||
|
||||
expect($customCommands)->toBeArray()->toBeEmpty();
|
||||
});
|
||||
|
||||
it('returns empty array for caddy proxy type', function () {
|
||||
$existingConfig = [
|
||||
'services' => [
|
||||
'caddy' => [
|
||||
'environment' => ['SOME_VAR=value'],
|
||||
],
|
||||
],
|
||||
];
|
||||
|
||||
$yamlConfig = Yaml::dump($existingConfig);
|
||||
|
||||
$server = Mockery::mock('App\Models\Server');
|
||||
$server->shouldReceive('proxyType')->andReturn(ProxyTypes::CADDY->value);
|
||||
|
||||
$customCommands = extractCustomProxyCommands($server, $yamlConfig);
|
||||
|
||||
expect($customCommands)->toBeArray()->toBeEmpty();
|
||||
});
|
||||
|
||||
it('returns empty array when config is empty', function () {
|
||||
$server = Mockery::mock('App\Models\Server');
|
||||
$server->shouldReceive('proxyType')->andReturn(ProxyTypes::TRAEFIK->value);
|
||||
|
||||
$customCommands = extractCustomProxyCommands($server, '');
|
||||
|
||||
expect($customCommands)->toBeArray()->toBeEmpty();
|
||||
});
|
||||
|
||||
it('correctly identifies multiple custom command types', function () {
|
||||
$existingConfig = [
|
||||
'services' => [
|
||||
'traefik' => [
|
||||
'command' => [
|
||||
'--ping=true',
|
||||
'--api.dashboard=true',
|
||||
'--entrypoints.http.forwardedHeaders.trustedIPs=173.245.48.0/20',
|
||||
'--entrypoints.https.forwardedHeaders.trustedIPs=173.245.48.0/20',
|
||||
'--entrypoints.http.forwardedHeaders.insecure=true',
|
||||
'--metrics.prometheus=true',
|
||||
'--providers.docker=true',
|
||||
],
|
||||
],
|
||||
],
|
||||
];
|
||||
|
||||
$yamlConfig = Yaml::dump($existingConfig);
|
||||
|
||||
$server = Mockery::mock('App\Models\Server');
|
||||
$server->shouldReceive('proxyType')->andReturn(ProxyTypes::TRAEFIK->value);
|
||||
|
||||
$customCommands = extractCustomProxyCommands($server, $yamlConfig);
|
||||
|
||||
expect($customCommands)
|
||||
->toBeArray()
|
||||
->toHaveCount(4)
|
||||
->toContain('--entrypoints.http.forwardedHeaders.trustedIPs=173.245.48.0/20')
|
||||
->toContain('--entrypoints.https.forwardedHeaders.trustedIPs=173.245.48.0/20')
|
||||
->toContain('--entrypoints.http.forwardedHeaders.insecure=true')
|
||||
->toContain('--metrics.prometheus=true');
|
||||
});
|
||||
Reference in New Issue
Block a user