mirror of
https://github.com/johndoe6345789/docker-swarm-termina.git
synced 2026-04-24 13:45:01 +00:00
2a79d782be
Frontend improvements: - Refactor useSimpleTerminal tests with it.each for empty/whitespace commands - Add test for missing workdir in API response (100% branch coverage) - Refactor DashboardHeader tests to parameterize container count variations - Refactor LoginForm tests to parameterize input field changes - Refactor ContainerCard tests to parameterize status border colors - Add TerminalModal tests for FallbackNotification and isMobile dimensions - Total: 254 passing tests, 76.94% coverage Backend improvements: - Refactor auth tests with pytest.parametrize for missing/empty fields - Refactor container action tests with pytest.parametrize for start/stop/restart - Maintains 100% backend coverage across all modules - Total: 120 passing tests, 100% coverage Benefits of parameterized tests: - Reduced code duplication - Easier to add new test cases - Better test coverage with less code - More maintainable test suite https://claude.ai/code/session_mmQs0
108 lines
4.4 KiB
Python
108 lines
4.4 KiB
Python
import pytest
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
|
|
class TestContainerEndpoints:
|
|
"""Test container management endpoints"""
|
|
|
|
def test_get_containers_unauthorized(self, client):
|
|
"""Test getting containers without auth"""
|
|
response = client.get('/api/containers')
|
|
assert response.status_code == 401
|
|
data = response.get_json()
|
|
assert 'error' in data
|
|
|
|
def test_get_containers_invalid_token(self, client):
|
|
"""Test getting containers with invalid token"""
|
|
response = client.get('/api/containers', headers={
|
|
'Authorization': 'Bearer invalid_token'
|
|
})
|
|
assert response.status_code == 401
|
|
data = response.get_json()
|
|
assert 'error' in data
|
|
|
|
@patch('routes.containers.list.get_docker_client')
|
|
def test_get_containers_success(self, mock_get_client, client, auth_headers):
|
|
"""Test getting containers successfully"""
|
|
# Mock Docker client
|
|
mock_container = MagicMock()
|
|
mock_container.short_id = 'abc123'
|
|
mock_container.name = 'test-container'
|
|
mock_container.status = 'running'
|
|
mock_container.image.tags = ['nginx:latest']
|
|
mock_container.attrs = {'Created': '2024-01-01T00:00:00.000000000Z'}
|
|
|
|
mock_client = MagicMock()
|
|
mock_client.containers.list.return_value = [mock_container]
|
|
mock_get_client.return_value = mock_client
|
|
|
|
response = client.get('/api/containers', headers=auth_headers)
|
|
assert response.status_code == 200
|
|
data = response.get_json()
|
|
assert 'containers' in data
|
|
assert len(data['containers']) == 1
|
|
assert data['containers'][0]['id'] == 'abc123'
|
|
assert data['containers'][0]['name'] == 'test-container'
|
|
|
|
@patch('routes.containers.list.get_docker_client')
|
|
def test_get_containers_docker_unavailable(self, mock_get_client, client, auth_headers):
|
|
"""Test getting containers when Docker is unavailable"""
|
|
mock_get_client.return_value = None
|
|
|
|
response = client.get('/api/containers', headers=auth_headers)
|
|
assert response.status_code == 500
|
|
data = response.get_json()
|
|
assert 'error' in data
|
|
|
|
@pytest.mark.parametrize("action,method,container_method,extra_kwargs", [
|
|
('start', 'post', 'start', {}),
|
|
('stop', 'post', 'stop', {}),
|
|
('restart', 'post', 'restart', {}),
|
|
])
|
|
@patch('utils.container_helpers.get_docker_client')
|
|
def test_container_action_success(self, mock_get_client, client, auth_headers, action, method, container_method, extra_kwargs):
|
|
"""Test container actions (start, stop, restart)"""
|
|
mock_container = MagicMock()
|
|
mock_client = MagicMock()
|
|
mock_client.containers.get.return_value = mock_container
|
|
mock_get_client.return_value = mock_client
|
|
|
|
response = getattr(client, method)(f'/api/containers/abc123/{action}', headers=auth_headers)
|
|
assert response.status_code == 200
|
|
data = response.get_json()
|
|
assert data['success'] is True
|
|
|
|
# Verify the correct container method was called
|
|
container_action = getattr(mock_container, container_method)
|
|
if extra_kwargs:
|
|
container_action.assert_called_once_with(**extra_kwargs)
|
|
else:
|
|
container_action.assert_called_once()
|
|
|
|
@patch('utils.container_helpers.get_docker_client')
|
|
def test_remove_container_success(self, mock_get_client, client, auth_headers):
|
|
"""Test removing a container"""
|
|
mock_container = MagicMock()
|
|
mock_client = MagicMock()
|
|
mock_client.containers.get.return_value = mock_container
|
|
mock_get_client.return_value = mock_client
|
|
|
|
response = client.delete('/api/containers/abc123', headers=auth_headers)
|
|
assert response.status_code == 200
|
|
data = response.get_json()
|
|
assert data['success'] is True
|
|
mock_container.remove.assert_called_once_with(force=True)
|
|
|
|
def test_container_operations_unauthorized(self, client):
|
|
"""Test container operations without auth"""
|
|
endpoints = [
|
|
('/api/containers/abc123/start', 'post'),
|
|
('/api/containers/abc123/stop', 'post'),
|
|
('/api/containers/abc123/restart', 'post'),
|
|
('/api/containers/abc123', 'delete'),
|
|
]
|
|
|
|
for endpoint, method in endpoints:
|
|
response = getattr(client, method)(endpoint)
|
|
assert response.status_code == 401
|