Coverage for mlos_bench/mlos_bench/tests/services/remote/mock/mock_network_service.py: 100%

1# 

2# Copyright (c) Microsoft Corporation. 

3# Licensed under the MIT License. 

4# 

5""" 

6A collection Service functions for mocking managing (Virtual) Networks. 

7""" 

8 

9from typing import Any, Callable, Dict, List, Optional, Union 

10 

11from mlos_bench.services.base_service import Service 

12from mlos_bench.services.types.network_provisioner_type import SupportsNetworkProvisioning 

13from mlos_bench.tests.services.remote.mock import mock_operation 

14 

15 

16class MockNetworkService(Service, SupportsNetworkProvisioning): 

17 """ 

18 Mock Network service for testing. 

19 """ 

20 

21 def __init__(self, config: Optional[Dict[str, Any]] = None, 

22 global_config: Optional[Dict[str, Any]] = None, 

23 parent: Optional[Service] = None, 

24 methods: Union[Dict[str, Callable], List[Callable], None] = None): 

25 """ 

26 Create a new instance of mock network services proxy. 

27 

28 Parameters 

29 ---------- 

30 config : dict 

31 Free-format dictionary that contains the benchmark environment 

32 configuration. 

33 global_config : dict 

34 Free-format dictionary of global parameters. 

35 parent : Service 

36 Parent service that can provide mixin functions. 

37 """ 

38 super().__init__( 

39 config, global_config, parent, 

40 self.merge_methods(methods, { 

41 name: mock_operation for name in ( 

42 # SupportsNetworkProvisioning: 

43 "provision_network", 

44 "deprovision_network", 

45 "wait_network_deployment", 

46 ) 

47 }) 

48 )