1
0
mirror of https://github.com/GNS3/gns3-server synced 2024-12-26 00:38:10 +00:00

Fix tests.

This commit is contained in:
grossmj 2021-04-15 18:30:22 +09:30
parent 9c850e0f2b
commit 6b8ce8219c
5 changed files with 15 additions and 18 deletions

View File

@ -75,14 +75,13 @@ async def test_vbox_get(app: FastAPI, client: AsyncClient, compute_project: Proj
async def test_vbox_start(app: FastAPI, client: AsyncClient, vm: dict) -> None: async def test_vbox_start(app: FastAPI, client: AsyncClient, vm: dict) -> None:
with asyncio_patch("gns3server.compute.virtualbox.virtualbox_vm.VirtualBoxVM.check_hw_virtualization", return_value=True): with asyncio_patch("gns3server.compute.virtualbox.virtualbox_vm.VirtualBoxVM.start", return_value=True) as mock:
with asyncio_patch("gns3server.compute.virtualbox.virtualbox_vm.VirtualBoxVM.start", return_value=True) as mock:
response = await client.post(app.url_path_for("start_virtualbox_node", response = await client.post(app.url_path_for("start_virtualbox_node",
project_id=vm["project_id"], project_id=vm["project_id"],
node_id=vm["node_id"])) node_id=vm["node_id"]))
assert mock.called assert mock.called
assert response.status_code == status.HTTP_204_NO_CONTENT assert response.status_code == status.HTTP_204_NO_CONTENT
async def test_vbox_stop(app: FastAPI, client: AsyncClient, vm: dict) -> None: async def test_vbox_stop(app: FastAPI, client: AsyncClient, vm: dict) -> None:

View File

@ -82,14 +82,12 @@ async def test_vmware_get(app: FastAPI, client: AsyncClient, compute_project: Pr
async def test_vmware_start(app: FastAPI, client: AsyncClient, vm: dict) -> None: async def test_vmware_start(app: FastAPI, client: AsyncClient, vm: dict) -> None:
with asyncio_patch("gns3server.compute.vmware.vmware_vm.VMwareVM.check_hw_virtualization", return_value=True) as mock1: with asyncio_patch("gns3server.compute.vmware.vmware_vm.VMwareVM.start", return_value=True) as mock:
with asyncio_patch("gns3server.compute.vmware.vmware_vm.VMwareVM.start", return_value=True) as mock2: response = await client.post(app.url_path_for("start_vmware_node",
response = await client.post(app.url_path_for("start_vmware_node", project_id=vm["project_id"],
project_id=vm["project_id"], node_id=vm["node_id"]))
node_id=vm["node_id"])) assert mock.called
assert mock1.called assert response.status_code == status.HTTP_204_NO_CONTENT
assert mock2.called
assert response.status_code == status.HTTP_204_NO_CONTENT
async def test_vmware_stop(app: FastAPI, client: AsyncClient, vm: dict) -> None: async def test_vmware_stop(app: FastAPI, client: AsyncClient, vm: dict) -> None:

View File

@ -21,7 +21,7 @@ import pytest
from fastapi import FastAPI, status from fastapi import FastAPI, status
from httpx import AsyncClient from httpx import AsyncClient
from gns3server.schemas.computes import Compute from gns3server.schemas.controller.computes import Compute
pytestmark = pytest.mark.asyncio pytestmark = pytest.mark.asyncio

View File

@ -27,7 +27,7 @@ from gns3server.db.repositories.users import UsersRepository
from gns3server.services import auth_service from gns3server.services import auth_service
from gns3server.services.authentication import DEFAULT_JWT_SECRET_KEY from gns3server.services.authentication import DEFAULT_JWT_SECRET_KEY
from gns3server.config import Config from gns3server.config import Config
from gns3server.schemas.users import User from gns3server.schemas.controller.users import User
pytestmark = pytest.mark.asyncio pytestmark = pytest.mark.asyncio

View File

@ -23,7 +23,7 @@ from gns3server.db.repositories.users import UsersRepository
from gns3server.db.repositories.computes import ComputesRepository from gns3server.db.repositories.computes import ComputesRepository
from gns3server.api.routes.controller.dependencies.database import get_db_session from gns3server.api.routes.controller.dependencies.database import get_db_session
from gns3server import schemas from gns3server import schemas
from gns3server.schemas.computes import Protocol from gns3server.schemas.controller.computes import Protocol
from gns3server.services import auth_service from gns3server.services import auth_service
from gns3server.services.authentication import DEFAULT_JWT_SECRET_KEY from gns3server.services.authentication import DEFAULT_JWT_SECRET_KEY