def test_serial_console_closed_on_error(self, get_serial, metadata_client): class OurException(Exception): pass metadata_client.side_effect = OurException try: DataSourceSmartOS.query_data('noun', 'device', 0) except OurException: pass self.assertEqual(1, get_serial.return_value.close.call_count)
def _get_ds(self, mockdata=None, mode=DataSourceSmartOS.SMARTOS_ENV_KVM, sys_cfg=None, ds_cfg=None): self.jmc_cfact.return_value = PsuedoJoyentClient(mockdata) self.get_smartos_environ.return_value = mode tmpd = self.tmp_dir() dirs = { 'cloud_dir': self.tmp_path('cloud_dir', tmpd), 'run_dir': self.tmp_path('run_dir') } for d in dirs.values(): os.mkdir(d) paths = c_helpers.Paths(dirs) if sys_cfg is None: sys_cfg = {} if ds_cfg is not None: sys_cfg['datasource'] = sys_cfg.get('datasource', {}) sys_cfg['datasource']['SmartOS'] = ds_cfg return DataSourceSmartOS.DataSourceSmartOS(sys_cfg, distro=None, paths=paths)
def _get_ds( self, mockdata=None, mode=DataSourceSmartOS.SMARTOS_ENV_KVM, sys_cfg=None, ds_cfg=None, ): self.jmc_cfact.return_value = PsuedoJoyentClient(mockdata) self.get_smartos_environ.return_value = mode tmpd = self.tmp_dir() dirs = { "cloud_dir": self.tmp_path("cloud_dir", tmpd), "run_dir": self.tmp_path("run_dir"), } for d in dirs.values(): os.mkdir(d) paths = c_helpers.Paths(dirs) if sys_cfg is None: sys_cfg = {} if ds_cfg is not None: sys_cfg["datasource"] = sys_cfg.get("datasource", {}) sys_cfg["datasource"]["SmartOS"] = ds_cfg return DataSourceSmartOS.DataSourceSmartOS(sys_cfg, distro=None, paths=paths)
def _get_ds(self, mockdata=None, mode=DataSourceSmartOS.SMARTOS_ENV_KVM, sys_cfg=None, ds_cfg=None): self.jmc_cfact.return_value = PsuedoJoyentClient(mockdata) self.get_smartos_environ.return_value = mode if sys_cfg is None: sys_cfg = {} if ds_cfg is not None: sys_cfg['datasource'] = sys_cfg.get('datasource', {}) sys_cfg['datasource']['SmartOS'] = ds_cfg return DataSourceSmartOS.DataSourceSmartOS( sys_cfg, distro=None, paths=self.paths)
def test_convert_simple(self): expected = { "version": 1, "config": [ { "name": "net0", "type": "physical", "subnets": [ { "type": "static", "gateway": "8.12.42.1", "netmask": "255.255.255.0", "address": "8.12.42.102/24", } ], "mtu": 1500, "mac_address": "90:b8:d0:f5:e4:f5", }, { "name": "net1", "type": "physical", "subnets": [ { "type": "static", "gateway": "192.168.128.1", "netmask": "255.255.252.0", "address": "192.168.128.93/22", } ], "mtu": 8500, "mac_address": "90:b8:d0:a5:ff:cd", }, ], } found = DataSourceSmartOS.convert_smartos_network_data(SDC_NICS) self.assertEqual(expected, found)
def _get_serial_client(self): self.serial.timeout = 1 return DataSourceSmartOS.JoyentMetadataSerialClient(None, fp=self.serial)
def _get_client(self): return DataSourceSmartOS.JoyentMetadataClient( fp=self.serial, smartos_type=DataSourceSmartOS.SMARTOS_ENV_KVM)
def _get_client(self): return DataSourceSmartOS.JoyentMetadataClient(self.serial)