Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions server/tests/core/services/generics/fixtures.py
Original file line number Diff line number Diff line change
Expand Up @@ -486,7 +486,7 @@ def get_mac(
caller_instance: typing.Optional[dynamic_userservice.DynamicUserService | dynamic_publication.DynamicPublication],
vmid: str,
*,
force_new: bool = False,
for_unique_id: bool = False,
) -> str:
self.mock.get_mac(caller_instance, vmid)
return '02:04:06:08:0A:0C'
Expand Down Expand Up @@ -689,7 +689,7 @@ def get_mac(
caller_instance: typing.Optional[dynamic_userservice.DynamicUserService | dynamic_publication.DynamicPublication],
vmid: str,
*,
force_new: bool = False,
for_unique_id: bool = False,
) -> str:
return ''

Expand Down
29 changes: 16 additions & 13 deletions server/tests/services/proxmox/fixtures.py
Original file line number Diff line number Diff line change
Expand Up @@ -54,7 +54,7 @@
service_fixed,
publication,
deployment_fixed,
service_linked,
service,
)

from uds.services.Proxmox.proxmox import types as prox_types, exceptions as prox_exceptions
Expand Down Expand Up @@ -513,7 +513,7 @@ def inner(*args: typing.Any, **kwargs: typing.Any) -> T:
'concurrent_removal_limit': 1,
'timeout': 10,
'start_vmid': 100,
'macs_range': '00:00:00:00:00:00-00:00:00:ff:ff:ff',
'macs_range': '00:00:00:00:00:01-00:00:00:ff:ff:ff',
}


Expand Down Expand Up @@ -550,10 +550,13 @@ def create_client_mock() -> mock.Mock:
def patched_provider(
**kwargs: typing.Any,
) -> typing.Generator[provider.ProxmoxProvider, None, None]:
client = create_client_mock()
provider = create_provider(**kwargs)
provider._cached_api = client
yield provider
yield create_mocked_provider(**kwargs)


def create_mocked_provider(**kwargs: typing.Any) -> provider.ProxmoxProvider:
prov = create_provider(**kwargs)
prov._cached_api = create_client_mock()
return prov


def create_provider(**kwargs: typing.Any) -> provider.ProxmoxProvider:
Expand All @@ -565,22 +568,22 @@ def create_provider(**kwargs: typing.Any) -> provider.ProxmoxProvider:

uuid_ = str(uuid.uuid4())
return provider.ProxmoxProvider(
environment=environment.Environment.private_environment(uuid), values=values, uuid=uuid_
environment=environment.Environment.private_environment(uuid_), values=values, uuid=uuid_
)


def create_service_linked(
provider: typing.Optional[provider.ProxmoxProvider] = None, **kwargs: typing.Any
) -> service_linked.ProxmoxServiceLinked:
) -> service.ProxmoxService:
"""
Create a fixed service
"""
uuid_ = str(uuid.uuid4())
values = SERVICE_LINKED_VALUES_DICT.copy()
values.update(kwargs)
srvc = service_linked.ProxmoxServiceLinked(
srvc = service.ProxmoxService(
environment=environment.Environment.private_environment(uuid_),
provider=provider or create_provider(),
provider=provider or create_mocked_provider(),
values=values,
uuid=uuid_,
)
Expand All @@ -605,14 +608,14 @@ def create_service_fixed(
values.update(kwargs)
return service_fixed.ProxmoxServiceFixed(
environment=environment.Environment.private_environment(uuid_),
provider=provider or create_provider(),
provider=provider or create_mocked_provider(),
values=values,
uuid=uuid_,
)


def create_publication(
service: typing.Optional[service_linked.ProxmoxServiceLinked] = None,
service: typing.Optional[service.ProxmoxService] = None,
**kwargs: typing.Any,
) -> 'publication.ProxmoxPublication':
"""
Expand Down Expand Up @@ -647,7 +650,7 @@ def create_userservice_fixed(


def create_userservice_linked(
service: typing.Optional[service_linked.ProxmoxServiceLinked] = None,
service: typing.Optional[service.ProxmoxService] = None,
publication: typing.Optional['publication.ProxmoxPublication'] = None,
) -> deployment_linked.ProxmoxUserserviceLinked:
"""
Expand Down
2 changes: 1 addition & 1 deletion server/tests/services/proxmox/test_service_fixed.py
Original file line number Diff line number Diff line change
Expand Up @@ -152,7 +152,7 @@ def test_process_snapshot(self) -> None:
fixtures.SNAPSHOTS_INFO.clear()
service.snapshot_creation(userservice_instance)
api.get_current_vm_snapshot.assert_called_with(int(vmid))
api.create_snapshot.assert_called_with(int(vmid), name='UDS Snapshot')
api.create_snapshot.assert_called_with(int(vmid), name='UDS_Snapshot')

# Skip snapshot creation
api.reset_mock()
Expand Down
Loading