How to use once method in Playwright Internal

Best JavaScript code snippet using playwright-internal

test_servers.py

Source:test_servers.py Github

copy

Full Screen

1# Copyright 2013: Mirantis Inc.2# All Rights Reserved.3#4# Licensed under the Apache License, Version 2.0 (the "License"); you may5# not use this file except in compliance with the License. You may obtain6# a copy of the License at7#8# http://www.apache.org/licenses/LICENSE-2.09#10# Unless required by applicable law or agreed to in writing, software11# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT12# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the13# License for the specific language governing permissions and limitations14# under the License.15from unittest import mock16import ddt17from rally import exceptions as rally_exceptions18from rally_openstack.task.scenarios.nova import servers19from tests.unit import fakes20from tests.unit import test21NOVA_SERVERS_MODULE = "rally_openstack.task.scenarios.nova.servers"22NOVA_SERVERS = NOVA_SERVERS_MODULE + ".NovaServers"23@ddt.ddt24class NovaServersTestCase(test.ScenarioTestCase):25 @ddt.data(("rescue_unrescue", ["_rescue_server", "_unrescue_server"], 1),26 ("stop_start", ["_stop_server", "_start_server"], 2),27 ("pause_unpause", ["_pause_server", "_unpause_server"], 3),28 ("suspend_resume", ["_suspend_server", "_resume_server"], 4),29 ("lock_unlock", ["_lock_server", "_unlock_server"], 5),30 ("shelve_unshelve", ["_shelve_server", "_unshelve_server"], 6))31 @ddt.unpack32 def test_action_pair(self, action_pair, methods, nof_calls):33 actions = [{action_pair: nof_calls}]34 fake_server = mock.MagicMock()35 scenario = servers.BootAndBounceServer(self.context)36 scenario._boot_server = mock.MagicMock(return_value=fake_server)37 scenario._delete_server = mock.MagicMock()38 scenario.generate_random_name = mock.MagicMock(return_value="name")39 for method in methods:40 setattr(scenario, method, mock.MagicMock())41 scenario.run("img", 1, actions=actions)42 scenario._boot_server.assert_called_once_with("img", 1)43 server_calls = []44 for i in range(nof_calls):45 server_calls.append(mock.call(fake_server))46 for method in methods:47 mocked_method = getattr(scenario, method)48 self.assertEqual(nof_calls, mocked_method.call_count,49 "%s not called %d times" % (method, nof_calls))50 mocked_method.assert_has_calls(server_calls)51 scenario._delete_server.assert_called_once_with(fake_server,52 force=False)53 def test_multiple_bounce_actions(self):54 actions = [{"hard_reboot": 5}, {"stop_start": 8},55 {"rescue_unrescue": 3}, {"pause_unpause": 2},56 {"suspend_resume": 4}, {"lock_unlock": 6},57 {"shelve_unshelve": 7}]58 fake_server = mock.MagicMock()59 scenario = servers.BootAndBounceServer(self.context)60 scenario._boot_server = mock.MagicMock(return_value=fake_server)61 scenario._delete_server = mock.MagicMock()62 scenario._reboot_server = mock.MagicMock()63 scenario._stop_and_start_server = mock.MagicMock()64 scenario._rescue_and_unrescue_server = mock.MagicMock()65 scenario._pause_and_unpause_server = mock.MagicMock()66 scenario._suspend_and_resume_server = mock.MagicMock()67 scenario._lock_and_unlock_server = mock.MagicMock()68 scenario._shelve_and_unshelve_server = mock.MagicMock()69 scenario.generate_random_name = mock.MagicMock(return_value="name")70 scenario.run("img", 1, actions=actions)71 scenario._boot_server.assert_called_once_with("img", 1)72 server_calls = []73 for i in range(5):74 server_calls.append(mock.call(fake_server))75 self.assertEqual(5, scenario._reboot_server.call_count,76 "Reboot not called 5 times")77 scenario._reboot_server.assert_has_calls(server_calls)78 server_calls = []79 for i in range(8):80 server_calls.append(mock.call(fake_server))81 self.assertEqual(8, scenario._stop_and_start_server.call_count,82 "Stop/Start not called 8 times")83 scenario._stop_and_start_server.assert_has_calls(server_calls)84 server_calls = []85 for i in range(3):86 server_calls.append(mock.call(fake_server))87 self.assertEqual(3, scenario._rescue_and_unrescue_server.call_count,88 "Rescue/Unrescue not called 3 times")89 scenario._rescue_and_unrescue_server.assert_has_calls(server_calls)90 server_calls = []91 for i in range(2):92 server_calls.append(mock.call(fake_server))93 self.assertEqual(2, scenario._pause_and_unpause_server.call_count,94 "Pause/Unpause not called 2 times")95 scenario._pause_and_unpause_server.assert_has_calls(server_calls)96 server_calls = []97 for i in range(4):98 server_calls.append(mock.call(fake_server))99 self.assertEqual(4, scenario._suspend_and_resume_server.call_count,100 "Suspend/Resume not called 4 times")101 scenario._suspend_and_resume_server.assert_has_calls(server_calls)102 server_calls = []103 for i in range(6):104 server_calls.append(mock.call(fake_server))105 self.assertEqual(6, scenario._lock_and_unlock_server.call_count,106 "Lock/Unlock not called 6 times")107 scenario._lock_and_unlock_server.assert_has_calls(server_calls)108 server_calls = []109 for i in range(7):110 server_calls.append(mock.call(fake_server))111 self.assertEqual(7, scenario._shelve_and_unshelve_server.call_count,112 "Shelve/Unshelve not called 7 times")113 scenario._shelve_and_unshelve_server.assert_has_calls(server_calls)114 scenario._delete_server.assert_called_once_with(fake_server,115 force=False)116 def test_boot_lock_unlock_and_delete(self):117 server = fakes.FakeServer()118 image = fakes.FakeImage()119 flavor = fakes.FakeFlavor()120 scenario = servers.BootLockUnlockAndDelete(self.context)121 scenario._boot_server = mock.Mock(return_value=server)122 scenario._lock_server = mock.Mock(side_effect=lambda s: s.lock())123 scenario._unlock_server = mock.Mock(side_effect=lambda s: s.unlock())124 scenario._delete_server = mock.Mock(125 side_effect=lambda s, **kwargs:126 self.assertFalse(getattr(s, "OS-EXT-STS:locked", False)))127 scenario.run(image, flavor, fakearg="fakearg")128 scenario._boot_server.assert_called_once_with(image, flavor,129 fakearg="fakearg")130 scenario._lock_server.assert_called_once_with(server)131 scenario._unlock_server.assert_called_once_with(server)132 scenario._delete_server.assert_called_once_with(server, force=False)133 @ddt.data("hard_reboot", "soft_reboot", "stop_start",134 "rescue_unrescue", "pause_unpause", "suspend_resume",135 "lock_unlock", "shelve_unshelve")136 def test_validate_actions(self, action):137 scenario = servers.BootAndBounceServer(self.context)138 self.assertRaises(rally_exceptions.InvalidConfigException,139 scenario.run,140 1, 1, actions=[{action: "no"}])141 self.assertRaises(rally_exceptions.InvalidConfigException,142 scenario.run,143 1, 1, actions=[{action: -1}])144 self.assertRaises(rally_exceptions.InvalidConfigException,145 scenario.run,146 1, 1, actions=[{action: 0}])147 def test_validate_actions_additional(self):148 scenario = servers.BootAndBounceServer(self.context)149 self.assertRaises(rally_exceptions.InvalidConfigException,150 scenario.run,151 1, 1, actions=[{"not_existing_action": "no"}])152 # NOTE: next should fail because actions parameter is a just a153 # dictionary, not an array of dictionaries154 self.assertRaises(rally_exceptions.InvalidConfigException,155 scenario.run,156 1, 1, actions={"hard_reboot": 1})157 def _verify_reboot(self, soft=True):158 actions = [{"soft_reboot" if soft else "hard_reboot": 5}]159 fake_server = mock.MagicMock()160 scenario = servers.BootAndBounceServer(self.context)161 scenario._reboot_server = mock.MagicMock()162 scenario._soft_reboot_server = mock.MagicMock()163 scenario._boot_server = mock.MagicMock(return_value=fake_server)164 scenario._delete_server = mock.MagicMock()165 scenario.generate_random_name = mock.MagicMock(return_value="name")166 scenario.run("img", 1, actions=actions)167 scenario._boot_server.assert_called_once_with("img", 1)168 server_calls = []169 for i in range(5):170 server_calls.append(mock.call(fake_server))171 if soft:172 self.assertEqual(5, scenario._soft_reboot_server.call_count,173 "Reboot not called 5 times")174 scenario._soft_reboot_server.assert_has_calls(server_calls)175 else:176 self.assertEqual(5, scenario._reboot_server.call_count,177 "Reboot not called 5 times")178 scenario._reboot_server.assert_has_calls(server_calls)179 scenario._delete_server.assert_called_once_with(fake_server,180 force=False)181 def test_boot_soft_reboot(self):182 self._verify_reboot(soft=True)183 def test_boot_hard_reboot(self):184 self._verify_reboot(soft=False)185 def test_boot_and_delete_server(self):186 fake_server = object()187 scenario = servers.BootAndDeleteServer(self.context)188 scenario.generate_random_name = mock.MagicMock(return_value="name")189 scenario._boot_server = mock.MagicMock(return_value=fake_server)190 scenario._delete_server = mock.MagicMock()191 scenario.sleep_between = mock.MagicMock()192 scenario.run("img", 0, 10, 20, fakearg="fakearg")193 scenario._boot_server.assert_called_once_with("img", 0,194 fakearg="fakearg")195 scenario.sleep_between.assert_called_once_with(10, 20)196 scenario._delete_server.assert_called_once_with(fake_server,197 force=False)198 def test_boot_and_delete_multiple_servers(self):199 scenario = servers.BootAndDeleteMultipleServers(self.context)200 scenario._boot_servers = mock.Mock()201 scenario._delete_servers = mock.Mock()202 scenario.sleep_between = mock.Mock()203 scenario.run("img", "flavor", count=15, min_sleep=10,204 max_sleep=20, fakearg="fakearg")205 scenario._boot_servers.assert_called_once_with("img", "flavor", 1,206 instances_amount=15,207 fakearg="fakearg")208 scenario.sleep_between.assert_called_once_with(10, 20)209 scenario._delete_servers.assert_called_once_with(210 scenario._boot_servers.return_value, force=False)211 def test_boot_and_list_server(self):212 scenario = servers.BootAndListServer(self.context)213# scenario.generate_random_name = mock.MagicMock(return_value="name")214 img_name = "img"215 flavor_uuid = 0216 details = True217 fake_server_name = mock.MagicMock()218 scenario._boot_server = mock.MagicMock()219 scenario._list_servers = mock.MagicMock()220 scenario._list_servers.return_value = [mock.MagicMock(),221 fake_server_name,222 mock.MagicMock()]223 # Positive case224 scenario._boot_server.return_value = fake_server_name225 scenario.run(img_name, flavor_uuid, fakearg="fakearg")226 scenario._boot_server.assert_called_once_with(img_name, flavor_uuid,227 fakearg="fakearg")228 scenario._list_servers.assert_called_once_with(details)229 # Negative case1: server isn't created230 scenario._boot_server.return_value = None231 self.assertRaises(rally_exceptions.RallyAssertionError,232 scenario.run,233 img_name, flavor_uuid, fakearg="fakearg")234 scenario._boot_server.assert_called_with(img_name, flavor_uuid,235 fakearg="fakearg")236 # Negative case2: server not in the list of available servers237 scenario._boot_server.return_value = mock.MagicMock()238 self.assertRaises(rally_exceptions.RallyAssertionError,239 scenario.run,240 img_name, flavor_uuid, fakearg="fakearg")241 scenario._boot_server.assert_called_with(img_name, flavor_uuid,242 fakearg="fakearg")243 scenario._list_servers.assert_called_with(details)244 def test_suspend_and_resume_server(self):245 fake_server = object()246 scenario = servers.SuspendAndResumeServer(self.context)247 scenario.generate_random_name = mock.MagicMock(return_value="name")248 scenario._boot_server = mock.MagicMock(return_value=fake_server)249 scenario._suspend_server = mock.MagicMock()250 scenario._resume_server = mock.MagicMock()251 scenario._delete_server = mock.MagicMock()252 scenario.run("img", 0, fakearg="fakearg")253 scenario._boot_server.assert_called_once_with("img", 0,254 fakearg="fakearg")255 scenario._suspend_server.assert_called_once_with(fake_server)256 scenario._resume_server.assert_called_once_with(fake_server)257 scenario._delete_server.assert_called_once_with(fake_server,258 force=False)259 def test_pause_and_unpause_server(self):260 fake_server = object()261 scenario = servers.PauseAndUnpauseServer(self.context)262 scenario.generate_random_name = mock.MagicMock(return_value="name")263 scenario._boot_server = mock.MagicMock(return_value=fake_server)264 scenario._pause_server = mock.MagicMock()265 scenario._unpause_server = mock.MagicMock()266 scenario._delete_server = mock.MagicMock()267 scenario.run("img", 0, fakearg="fakearg")268 scenario._boot_server.assert_called_once_with("img", 0,269 fakearg="fakearg")270 scenario._pause_server.assert_called_once_with(fake_server)271 scenario._unpause_server.assert_called_once_with(fake_server)272 scenario._delete_server.assert_called_once_with(fake_server,273 force=False)274 def test_shelve_and_unshelve_server(self):275 fake_server = mock.MagicMock()276 scenario = servers.ShelveAndUnshelveServer(self.context)277 scenario._boot_server = mock.MagicMock(return_value=fake_server)278 scenario._shelve_server = mock.MagicMock()279 scenario._unshelve_server = mock.MagicMock()280 scenario._delete_server = mock.MagicMock()281 scenario.run("img", 0, fakearg="fakearg")282 scenario._boot_server.assert_called_once_with("img", 0,283 fakearg="fakearg")284 scenario._shelve_server.assert_called_once_with(fake_server)285 scenario._unshelve_server.assert_called_once_with(fake_server)286 scenario._delete_server.assert_called_once_with(fake_server,287 force=False)288 def test_list_servers(self):289 scenario = servers.ListServers(self.context)290 scenario._list_servers = mock.MagicMock()291 scenario.run(True)292 scenario._list_servers.assert_called_once_with(True)293 @mock.patch("rally_openstack.common.services.storage.block.BlockStorage")294 def test_boot_server_from_volume(self, mock_block_storage):295 fake_server = object()296 scenario = servers.BootServerFromVolume(297 self.context, clients=mock.Mock())298 scenario._boot_server = mock.MagicMock(return_value=fake_server)299 fake_volume = fakes.FakeVolumeManager().create()300 fake_volume.id = "volume_id"301 cinder = mock_block_storage.return_value302 cinder.create_volume.return_value = fake_volume303 scenario.run("img", 0, 5, volume_type=None,304 auto_assign_nic=False, fakearg="f")305 cinder.create_volume.assert_called_once_with(5, imageRef="img",306 volume_type=None)307 scenario._boot_server.assert_called_once_with(308 None, 0, auto_assign_nic=False,309 block_device_mapping={"vda": "volume_id:::0"},310 fakearg="f")311 @mock.patch("rally_openstack.common.services.storage.block.BlockStorage")312 def test_boot_server_from_volume_and_delete(self, mock_block_storage):313 fake_server = object()314 scenario = servers.BootServerFromVolumeAndDelete(315 self.context, clients=mock.Mock())316 scenario._boot_server = mock.MagicMock(return_value=fake_server)317 scenario.sleep_between = mock.MagicMock()318 scenario._delete_server = mock.MagicMock()319 fake_volume = fakes.FakeVolumeManager().create()320 fake_volume.id = "volume_id"321 cinder = mock_block_storage.return_value322 cinder.create_volume.return_value = fake_volume323 scenario.run("img", 0, 5, None, 10, 20, fakearg="f")324 cinder.create_volume.assert_called_once_with(5, imageRef="img",325 volume_type=None)326 scenario._boot_server.assert_called_once_with(327 None, 0,328 block_device_mapping={"vda": "volume_id:::0"},329 fakearg="f")330 scenario.sleep_between.assert_called_once_with(10, 20)331 scenario._delete_server.assert_called_once_with(fake_server,332 force=False)333 def _prepare_boot(self, nic=None, assert_nic=False):334 fake_server = mock.MagicMock()335 scenario = servers.BootServer(self.context)336 scenario._boot_server = mock.MagicMock(return_value=fake_server)337 scenario.generate_random_name = mock.MagicMock(return_value="name")338 kwargs = {"fakearg": "f"}339 expected_kwargs = {"fakearg": "f"}340 assert_nic = nic or assert_nic341 if nic:342 kwargs["nics"] = nic343 if assert_nic:344 self.clients("nova").networks.create("net-1")345 expected_kwargs["nics"] = nic or [{"net-id": "net-2"}]346 return scenario, kwargs, expected_kwargs347 def _verify_boot_server(self, nic=None, assert_nic=False):348 scenario, kwargs, expected_kwargs = self._prepare_boot(349 nic=nic, assert_nic=assert_nic)350 scenario.run("img", 0, **kwargs)351 scenario._boot_server.assert_called_once_with(352 "img", 0, auto_assign_nic=False, **expected_kwargs)353 def test_boot_server_no_nics(self):354 self._verify_boot_server(nic=None, assert_nic=False)355 def test_boot_server_with_nic(self):356 self._verify_boot_server(nic=[{"net-id": "net-1"}], assert_nic=True)357 def test_snapshot_server(self):358 fake_server = object()359 fake_image = fakes.FakeImageManager()._create()360 fake_image.id = "image_id"361 scenario = servers.SnapshotServer(self.context)362 scenario.generate_random_name = mock.MagicMock(return_value="name")363 scenario._boot_server = mock.MagicMock(return_value=fake_server)364 scenario._create_image = mock.MagicMock(return_value=fake_image)365 scenario._delete_server = mock.MagicMock()366 scenario._delete_image = mock.MagicMock()367 scenario.run("i", 0, fakearg=2)368 scenario._boot_server.assert_has_calls([369 mock.call("i", 0, fakearg=2),370 mock.call("image_id", 0, fakearg=2)])371 scenario._create_image.assert_called_once_with(fake_server)372 scenario._delete_server.assert_has_calls([373 mock.call(fake_server, force=False),374 mock.call(fake_server, force=False)])375 scenario._delete_image.assert_called_once_with(fake_image)376 def _test_resize(self, confirm=False):377 fake_server = object()378 fake_image = fakes.FakeImageManager()._create()379 fake_image.id = "image_id"380 flavor = mock.MagicMock()381 to_flavor = mock.MagicMock()382 scenario = servers.ResizeServer(self.context)383 scenario.generate_random_name = mock.MagicMock(return_value="name")384 scenario._boot_server = mock.MagicMock(return_value=fake_server)385 scenario._resize_confirm = mock.MagicMock()386 scenario._resize_revert = mock.MagicMock()387 scenario._resize = mock.MagicMock()388 scenario._delete_server = mock.MagicMock()389 kwargs = {"confirm": confirm}390 scenario.run(fake_image, flavor, to_flavor, **kwargs)391 scenario._resize.assert_called_once_with(fake_server, to_flavor)392 if confirm:393 scenario._resize_confirm.assert_called_once_with(fake_server)394 else:395 scenario._resize_revert.assert_called_once_with(fake_server)396 def test_resize_with_confirm(self):397 self._test_resize(confirm=True)398 def test_resize_with_revert(self):399 self._test_resize(confirm=False)400 @ddt.data({"confirm": True},401 {"confirm": False})402 @ddt.unpack403 def test_resize_shoutoff_server(self, confirm=False):404 fake_server = object()405 flavor = mock.MagicMock()406 to_flavor = mock.MagicMock()407 scenario = servers.ResizeShutoffServer(self.context)408 scenario.generate_random_name = mock.MagicMock(return_value="name")409 scenario._boot_server = mock.MagicMock(return_value=fake_server)410 scenario._stop_server = mock.MagicMock()411 scenario._resize_confirm = mock.MagicMock()412 scenario._resize_revert = mock.MagicMock()413 scenario._resize = mock.MagicMock()414 scenario._delete_server = mock.MagicMock()415 scenario.run("img", flavor, to_flavor, confirm=confirm)416 scenario._boot_server.assert_called_once_with("img", flavor)417 scenario._stop_server.assert_called_once_with(fake_server)418 scenario._resize.assert_called_once_with(fake_server, to_flavor)419 if confirm:420 scenario._resize_confirm.assert_called_once_with(fake_server,421 "SHUTOFF")422 else:423 scenario._resize_revert.assert_called_once_with(fake_server,424 "SHUTOFF")425 scenario._delete_server.assert_called_once_with(fake_server,426 force=False)427 @ddt.data({"confirm": True, "do_delete": True},428 {"confirm": False, "do_delete": True})429 @ddt.unpack430 @mock.patch("rally_openstack.common.services.storage.block.BlockStorage")431 def test_boot_server_attach_created_volume_and_resize(432 self, mock_block_storage, confirm=False, do_delete=False):433 fake_volume = mock.MagicMock()434 fake_server = mock.MagicMock()435 flavor = mock.MagicMock()436 to_flavor = mock.MagicMock()437 fake_attachment = mock.MagicMock()438 cinder = mock_block_storage.return_value439 cinder.create_volume.return_value = fake_volume440 scenario = servers.BootServerAttachCreatedVolumeAndResize(441 self.context, clients=mock.Mock())442 scenario.generate_random_name = mock.MagicMock(return_value="name")443 scenario._boot_server = mock.MagicMock(return_value=fake_server)444 scenario._attach_volume = mock.MagicMock(return_value=fake_attachment)445 scenario._resize_confirm = mock.MagicMock()446 scenario._resize_revert = mock.MagicMock()447 scenario._resize = mock.MagicMock()448 scenario._detach_volume = mock.MagicMock()449 scenario._delete_server = mock.MagicMock()450 scenario.sleep_between = mock.MagicMock()451 volume_size = 10452 scenario.run("img", flavor, to_flavor, volume_size, min_sleep=10,453 max_sleep=20, confirm=confirm, do_delete=do_delete)454 scenario._boot_server.assert_called_once_with("img", flavor)455 cinder.create_volume.assert_called_once_with(volume_size)456 scenario._attach_volume.assert_called_once_with(fake_server,457 fake_volume)458 scenario._detach_volume.assert_called_once_with(fake_server,459 fake_volume)460 scenario.sleep_between.assert_called_once_with(10, 20)461 scenario._resize.assert_called_once_with(fake_server, to_flavor)462 if confirm:463 scenario._resize_confirm.assert_called_once_with(fake_server)464 else:465 scenario._resize_revert.assert_called_once_with(fake_server)466 if do_delete:467 scenario._detach_volume.assert_called_once_with(fake_server,468 fake_volume)469 cinder.delete_volume.assert_called_once_with(fake_volume)470 scenario._delete_server.assert_called_once_with(fake_server,471 force=False)472 @mock.patch("rally_openstack.common.services.storage.block.BlockStorage")473 def test_list_attachments(self, mock_block_storage):474 mock_volume_service = mock_block_storage.return_value475 fake_volume = mock.MagicMock()476 fake_server = mock.MagicMock()477 flavor = mock.MagicMock()478 fake_attachment = mock.MagicMock()479 list_attachments = [mock.MagicMock(),480 fake_attachment,481 mock.MagicMock()]482 context = self.context483 context.update({484 "admin": {485 "id": "fake_user_id",486 "credential": mock.MagicMock()487 },488 "user": {"id": "fake_user_id",489 "credential": mock.MagicMock()},490 "tenant": {"id": "fake", "name": "fake",491 "volumes": [{"id": "uuid", "size": 1}],492 "servers": [1]}})493 scenario = servers.BootServerAttachVolumeAndListAttachments(494 context)495 scenario._boot_server = mock.MagicMock(return_value=fake_server)496 scenario._attach_volume = mock.MagicMock()497 scenario._list_attachments = mock.MagicMock()498 mock_volume_service.create_volume.return_value = fake_volume499 scenario._list_attachments.return_value = list_attachments500 img_name = "img"501 volume_size = 10502 volume_num = 1503 scenario._attach_volume.return_value = fake_attachment504 scenario.run(img_name, flavor, volume_size, volume_num)505 scenario._boot_server.assert_called_once_with(img_name, flavor)506 mock_volume_service.create_volume.assert_called_once_with(volume_size)507 scenario._attach_volume.assert_called_once_with(fake_server,508 fake_volume)509 scenario._list_attachments.assert_called_once_with(fake_server.id)510 @mock.patch("rally_openstack.common.services.storage.block.BlockStorage")511 def test_list_attachments_fails(self, mock_block_storage):512 mock_volume_service = mock_block_storage.return_value513 fake_volume = mock.MagicMock()514 fake_server = mock.MagicMock()515 flavor = mock.MagicMock()516 fake_attachment = mock.MagicMock()517 list_attachments = [mock.MagicMock(),518 mock.MagicMock(),519 mock.MagicMock()]520 context = self.context521 context.update({522 "admin": {523 "id": "fake_user_id",524 "credential": mock.MagicMock()525 },526 "user": {"id": "fake_user_id",527 "credential": mock.MagicMock()},528 "tenant": {"id": "fake", "name": "fake",529 "volumes": [{"id": "uuid", "size": 1}],530 "servers": [1]}})531 scenario = servers.BootServerAttachVolumeAndListAttachments(532 context)533 scenario._boot_server = mock.MagicMock(return_value=fake_server)534 mock_volume_service.create_volume.return_value = fake_volume535 scenario._attach_volume = mock.MagicMock()536 scenario._list_attachments = mock.MagicMock()537 scenario._attach_volume.return_value = fake_attachment538 scenario._list_attachments.return_value = list_attachments539 img_name = "img"540 volume_size = 10541 # Negative case: attachment not included into list of542 # available attachments543 self.assertRaises(rally_exceptions.RallyAssertionError,544 scenario.run,545 img_name, flavor, volume_size)546 scenario._boot_server.assert_called_with(img_name, flavor)547 mock_volume_service.create_volume.assert_called_with(volume_size)548 scenario._attach_volume.assert_called_with(fake_server,549 fake_volume)550 scenario._list_attachments.assert_called_with(fake_server.id)551 @ddt.data({"confirm": True, "do_delete": True},552 {"confirm": False, "do_delete": True})553 @ddt.unpack554 @mock.patch("rally_openstack.common.services.storage.block.BlockStorage")555 def test_boot_server_from_volume_and_resize(556 self, mock_block_storage, confirm=False, do_delete=False):557 fake_server = object()558 flavor = mock.MagicMock()559 to_flavor = mock.MagicMock()560 scenario = servers.BootServerFromVolumeAndResize(self.context,561 clients=mock.Mock())562 scenario._boot_server = mock.MagicMock(return_value=fake_server)563 scenario.generate_random_name = mock.MagicMock(return_value="name")564 scenario._resize_confirm = mock.MagicMock()565 scenario._resize_revert = mock.MagicMock()566 scenario._resize = mock.MagicMock()567 scenario.sleep_between = mock.MagicMock()568 scenario._delete_server = mock.MagicMock()569 fake_volume = fakes.FakeVolumeManager().create()570 fake_volume.id = "volume_id"571 cinder = mock_block_storage.return_value572 cinder.create_volume.return_value = fake_volume573 volume_size = 10574 scenario.run("img", flavor, to_flavor, volume_size, min_sleep=10,575 max_sleep=20, confirm=confirm, do_delete=do_delete)576 cinder.create_volume.assert_called_once_with(10, imageRef="img")577 scenario._boot_server.assert_called_once_with(578 None, flavor,579 block_device_mapping={"vda": "volume_id:::0"})580 scenario.sleep_between.assert_called_once_with(10, 20)581 scenario._resize.assert_called_once_with(fake_server, to_flavor)582 if confirm:583 scenario._resize_confirm.assert_called_once_with(fake_server)584 else:585 scenario._resize_revert.assert_called_once_with(fake_server)586 if do_delete:587 scenario._delete_server.assert_called_once_with(fake_server,588 force=False)589 def test_boot_and_live_migrate_server(self):590 fake_server = mock.MagicMock()591 scenario = servers.BootAndLiveMigrateServer(self.context)592 scenario.generate_random_name = mock.MagicMock(return_value="name")593 scenario._boot_server = mock.MagicMock(return_value=fake_server)594 scenario.sleep_between = mock.MagicMock()595 scenario._live_migrate = mock.MagicMock()596 scenario._delete_server = mock.MagicMock()597 scenario.run("img", 0, min_sleep=10, max_sleep=20, fakearg="fakearg")598 scenario._boot_server.assert_called_once_with("img", 0,599 fakearg="fakearg")600 scenario.sleep_between.assert_called_once_with(10, 20)601 scenario._live_migrate.assert_called_once_with(fake_server,602 False, False)603 scenario._delete_server.assert_called_once_with(fake_server)604 @mock.patch("rally_openstack.common.services.storage.block.BlockStorage")605 def test_boot_server_from_volume_and_live_migrate(self,606 mock_block_storage):607 fake_server = mock.MagicMock()608 scenario = servers.BootServerFromVolumeAndLiveMigrate(609 self.context, clients=mock.Mock())610 scenario.generate_random_name = mock.MagicMock(return_value="name")611 scenario._boot_server = mock.MagicMock(return_value=fake_server)612 scenario.sleep_between = mock.MagicMock()613 scenario._live_migrate = mock.MagicMock()614 scenario._delete_server = mock.MagicMock()615 fake_volume = fakes.FakeVolumeManager().create()616 fake_volume.id = "volume_id"617 cinder = mock_block_storage.return_value618 cinder.create_volume.return_value = fake_volume619 scenario.run("img", 0, 5, volume_type=None,620 min_sleep=10, max_sleep=20, fakearg="f")621 cinder.create_volume.assert_called_once_with(5, imageRef="img",622 volume_type=None)623 scenario._boot_server.assert_called_once_with(624 None, 0,625 block_device_mapping={"vda": "volume_id:::0"},626 fakearg="f")627 scenario.sleep_between.assert_called_once_with(10, 20)628 scenario._live_migrate.assert_called_once_with(fake_server,629 False, False)630 scenario._delete_server.assert_called_once_with(fake_server,631 force=False)632 @mock.patch("rally_openstack.common.services.storage.block.BlockStorage")633 def test_boot_server_attach_created_volume_and_live_migrate(634 self, mock_block_storage):635 fake_volume = mock.MagicMock()636 fake_server = mock.MagicMock()637 fake_attachment = mock.MagicMock()638 clients = mock.Mock()639 cinder = mock_block_storage.return_value640 cinder.create_volume.return_value = fake_volume641 scenario = servers.BootServerAttachCreatedVolumeAndLiveMigrate(642 self.context, clients=clients)643 scenario._attach_volume = mock.MagicMock(return_value=fake_attachment)644 scenario._detach_volume = mock.MagicMock()645 scenario.sleep_between = mock.MagicMock()646 scenario._live_migrate = mock.MagicMock()647 scenario._boot_server = mock.MagicMock(return_value=fake_server)648 scenario._delete_server = mock.MagicMock()649 image = "img"650 flavor = "flavor"651 size = 5652 boot_kwargs = {"some_var": "asd"}653 scenario.run(image, flavor, size, min_sleep=10, max_sleep=20,654 boot_server_kwargs=boot_kwargs)655 scenario._boot_server.assert_called_once_with(image, flavor,656 **boot_kwargs)657 cinder.create_volume.assert_called_once_with(size)658 scenario._attach_volume.assert_called_once_with(fake_server,659 fake_volume)660 scenario._detach_volume.assert_called_once_with(fake_server,661 fake_volume)662 scenario.sleep_between.assert_called_once_with(10, 20)663 scenario._live_migrate.assert_called_once_with(fake_server,664 False, False)665 cinder.delete_volume.assert_called_once_with(fake_volume)666 scenario._delete_server.assert_called_once_with(fake_server)667 def _test_boot_and_migrate_server(self, confirm=False):668 fake_server = mock.MagicMock()669 scenario = servers.BootAndMigrateServer(self.context)670 scenario.generate_random_name = mock.MagicMock(return_value="name")671 scenario._boot_server = mock.MagicMock(return_value=fake_server)672 scenario._migrate = mock.MagicMock()673 scenario._resize_confirm = mock.MagicMock()674 scenario._resize_revert = mock.MagicMock()675 scenario._delete_server = mock.MagicMock()676 kwargs = {"confirm": confirm}677 scenario.run("img", 0, fakearg="fakearg", **kwargs)678 scenario._boot_server.assert_called_once_with("img", 0,679 fakearg="fakearg",680 confirm=confirm)681 scenario._migrate.assert_called_once_with(fake_server)682 if confirm:683 scenario._resize_confirm.assert_called_once_with(fake_server,684 status="ACTIVE")685 else:686 scenario._resize_revert.assert_called_once_with(fake_server,687 status="ACTIVE")688 scenario._delete_server.assert_called_once_with(fake_server)689 def test_boot_and_migrate_server_with_confirm(self):690 self._test_boot_and_migrate_server(confirm=True)691 def test_boot_and_migrate_server_with_revert(self):692 self._test_boot_and_migrate_server(confirm=False)693 def test_boot_and_rebuild_server(self):694 scenario = servers.BootAndRebuildServer(self.context)695 scenario._boot_server = mock.Mock()696 scenario._rebuild_server = mock.Mock()697 scenario._delete_server = mock.Mock()698 from_image = "img1"699 to_image = "img2"700 flavor = "flavor"701 scenario.run(from_image, to_image, flavor, fakearg="fakearg")702 scenario._boot_server.assert_called_once_with(from_image, flavor,703 fakearg="fakearg")704 server = scenario._boot_server.return_value705 scenario._rebuild_server.assert_called_once_with(server, to_image)706 scenario._delete_server.assert_called_once_with(server)707 def test_boot_and_show_server(self):708 server = fakes.FakeServer()709 image = fakes.FakeImage()710 flavor = fakes.FakeFlavor()711 scenario = servers.BootAndShowServer(self.context)712 scenario._boot_server = mock.MagicMock(return_value=server)713 scenario._show_server = mock.MagicMock()714 scenario.run(image, flavor, fakearg="fakearg")715 scenario._boot_server.assert_called_once_with(image, flavor,716 fakearg="fakearg")717 scenario._show_server.assert_called_once_with(server)718 def test_boot_server_and_list_interfaces(self):719 server = fakes.FakeServer()720 image = fakes.FakeImage()721 flavor = fakes.FakeFlavor()722 scenario = servers.BootServerAndListInterfaces(self.context)723 scenario._boot_server = mock.MagicMock(return_value=server)724 scenario._list_interfaces = mock.MagicMock()725 scenario.run(image, flavor, fakearg="fakearg")726 scenario._boot_server.assert_called_once_with(image, flavor,727 fakearg="fakearg")728 scenario._list_interfaces.assert_called_once_with(server)729 @ddt.data({"length": None},730 {"length": 10})731 @ddt.unpack732 def test_boot_and_get_console_server(self, length):733 server = fakes.FakeServer()734 image = fakes.FakeImage()735 flavor = fakes.FakeFlavor()736 kwargs = {"fakearg": "fakearg"}737 scenario = servers.BootAndGetConsoleOutput(self.context)738 scenario._boot_server = mock.MagicMock(return_value=server)739 scenario._get_server_console_output = mock.MagicMock()740 scenario.run(image, flavor, length, **kwargs)741 scenario._boot_server.assert_called_once_with(image, flavor,742 **kwargs)743 scenario._get_server_console_output.assert_called_once_with(server,744 length)745 def test_boot_and_get_console_url(self):746 server = fakes.FakeServer()747 image = fakes.FakeImage()748 flavor = fakes.FakeFlavor()749 kwargs = {"fakearg": "fakearg"}750 scenario = servers.BootAndGetConsoleUrl(self.context)751 scenario._boot_server = mock.MagicMock(return_value=server)752 scenario._get_console_url_server = mock.MagicMock()753 scenario.run(image, flavor, console_type="novnc", **kwargs)754 scenario._boot_server.assert_called_once_with(image, flavor,755 **kwargs)756 scenario._get_console_url_server.assert_called_once_with(757 server, "novnc")758 def test_boot_and_associate_floating_ip(self):759 clients = mock.MagicMock(credential=mock.MagicMock(api_info={}))760 neutronclient = clients.neutron.return_value761 floatingip = "floatingip"762 neutronclient.create_floatingip.return_value = {763 "floatingip": floatingip}764 scenario = servers.BootAndAssociateFloatingIp(self.context,765 clients=clients)766 server = mock.Mock()767 scenario._boot_server = mock.Mock(return_value=server)768 scenario._associate_floating_ip = mock.Mock()769 image = "img"770 flavor = "flavor"771 scenario.run(image, flavor, fakearg="fakearg")772 scenario._boot_server.assert_called_once_with(image, flavor,773 fakearg="fakearg")774 neutronclient.create_floatingip.assert_called_once_with(775 {"floatingip": mock.ANY}776 )777 scenario._associate_floating_ip.assert_called_once_with(778 server, floatingip)779 # check ext_network780 neutronclient.list_networks.return_value = {781 "networks": [782 {"id": "id1", "name": "net1", "router:external": True},783 {"id": "id2", "name": "net2", "router:external": True},784 {"id": "id3", "name": "net3", "router:external": True},785 ]786 }787 neutronclient.create_floatingip.reset_mock()788 # case 1: new argument is used789 scenario.run(image, flavor, floating_network="net3")790 neutronclient.create_floatingip.assert_called_once_with(791 {"floatingip": {"description": mock.ANY,792 "floating_network_id": "id3"}}793 )794 # case 2: new argument is transmitted with an old one795 neutronclient.create_floatingip.reset_mock()796 scenario.run(image, flavor, floating_network="net3",797 create_floating_ip_args={"ext_network": "net2"})798 neutronclient.create_floatingip.assert_called_once_with(799 {"floatingip": {"description": mock.ANY,800 "floating_network_id": "id3"}}801 )802 # case 3: new argument is transmitted with an semi-old one803 neutronclient.create_floatingip.reset_mock()804 scenario.run(image, flavor, floating_network="net3",805 create_floating_ip_args={"floating_network": "net1"})806 neutronclient.create_floatingip.assert_called_once_with(807 {"floatingip": {"description": mock.ANY,808 "floating_network_id": "id3"}}809 )810 # case 4: only old argument is transmitted811 neutronclient.create_floatingip.reset_mock()812 scenario.run(image, flavor,813 create_floating_ip_args={"ext_network": "net2"})814 neutronclient.create_floatingip.assert_called_once_with(815 {"floatingip": {"description": mock.ANY,816 "floating_network_id": "id2"}}817 )818 # case 5: only semi-old argument is transmitted819 neutronclient.create_floatingip.reset_mock()820 scenario.run(image, flavor,821 create_floating_ip_args={"floating_network": "net1"})822 neutronclient.create_floatingip.assert_called_once_with(823 {"floatingip": {"description": mock.ANY,824 "floating_network_id": "id1"}}825 )826 def test_boot_server_associate_and_dissociate_floating_ip(self):827 clients = mock.MagicMock(credential=mock.MagicMock(api_info={}))828 neutronclient = clients.neutron.return_value829 floatingip = "floatingip"830 neutronclient.create_floatingip.return_value = {831 "floatingip": floatingip}832 scenario = servers.BootServerAssociateAndDissociateFloatingIP(833 self.context, clients=clients)834 server = mock.Mock()835 scenario._boot_server = mock.Mock(return_value=server)836 scenario._associate_floating_ip = mock.Mock()837 scenario._dissociate_floating_ip = mock.Mock()838 image = "img"839 flavor = "flavor"840 scenario.run(image, flavor, fakearg="fakearg")841 scenario._boot_server.assert_called_once_with(image, flavor,842 fakearg="fakearg")843 neutronclient.create_floatingip.assert_called_once_with(844 {"floatingip": mock.ANY}845 )846 scenario._associate_floating_ip.assert_called_once_with(847 server, floatingip)848 scenario._dissociate_floating_ip.assert_called_once_with(849 server, floatingip)850 # check ext_network851 neutronclient.list_networks.return_value = {852 "networks": [853 {"id": "id1", "name": "net1", "router:external": True},854 {"id": "id2", "name": "net2", "router:external": True},855 {"id": "id3", "name": "net3", "router:external": True},856 ]857 }858 neutronclient.create_floatingip.reset_mock()859 # case 1: new argument is used860 scenario.run(image, flavor, floating_network="net3")861 neutronclient.create_floatingip.assert_called_once_with(862 {"floatingip": {"description": mock.ANY,863 "floating_network_id": "id3"}}864 )865 # case 2: new argument is transmitted with an old one866 neutronclient.create_floatingip.reset_mock()867 scenario.run(image, flavor, floating_network="net3",868 create_floating_ip_args={"ext_network": "net2"})869 neutronclient.create_floatingip.assert_called_once_with(870 {"floatingip": {"description": mock.ANY,871 "floating_network_id": "id3"}}872 )873 # case 3: new argument is transmitted with an semi-old one874 neutronclient.create_floatingip.reset_mock()875 scenario.run(image, flavor, floating_network="net3",876 create_floating_ip_args={"floating_network": "net1"})877 neutronclient.create_floatingip.assert_called_once_with(878 {"floatingip": {"description": mock.ANY,879 "floating_network_id": "id3"}}880 )881 # case 4: only old argument is transmitted882 neutronclient.create_floatingip.reset_mock()883 scenario.run(image, flavor,884 create_floating_ip_args={"ext_network": "net2"})885 neutronclient.create_floatingip.assert_called_once_with(886 {"floatingip": {"description": mock.ANY,887 "floating_network_id": "id2"}}888 )889 # case 5: only semi-old argument is transmitted890 neutronclient.create_floatingip.reset_mock()891 scenario.run(image, flavor,892 create_floating_ip_args={"floating_network": "net1"})893 neutronclient.create_floatingip.assert_called_once_with(894 {"floatingip": {"description": mock.ANY,895 "floating_network_id": "id1"}}896 )897 def test_boot_and_update_server(self):898 scenario = servers.BootAndUpdateServer(self.context)899 scenario._boot_server = mock.Mock()900 scenario._update_server = mock.Mock()901 scenario.run("img", "flavor", "desp", fakearg="fakearg")902 scenario._boot_server.assert_called_once_with("img", "flavor",903 fakearg="fakearg")904 scenario._update_server.assert_called_once_with(905 scenario._boot_server.return_value, "desp")906 def test_boot_server_and_attach_interface(self):907 network_create_args = {"router:external": True}908 subnet_create_args = {"allocation_pools": []}909 subnet_cidr_start = "10.1.0.0/16"910 boot_server_args = {}911 net = mock.MagicMock()912 subnet = mock.MagicMock()913 server = mock.MagicMock()914 scenario = servers.BootServerAndAttachInterface(self.context)915 scenario._get_or_create_network = mock.Mock(return_value=net)916 scenario._create_subnet = mock.Mock(return_value=subnet)917 scenario._boot_server = mock.Mock(return_value=server)918 scenario._attach_interface = mock.Mock()919 scenario.run("image", "flavor",920 network_create_args=network_create_args,921 subnet_create_args=subnet_create_args,922 subnet_cidr_start=subnet_cidr_start,923 boot_server_args=boot_server_args)924 scenario._get_or_create_network.assert_called_once_with(925 network_create_args)926 scenario._create_subnet.assert_called_once_with(net,927 subnet_create_args,928 subnet_cidr_start)929 scenario._boot_server.assert_called_once_with("image", "flavor",930 **boot_server_args)931 scenario._attach_interface.assert_called_once_with(932 server, net_id=net["network"]["id"])933 @mock.patch("rally_openstack.common.services.storage.block.BlockStorage")934 def test_boot_server_from_volume_snapshot(self, mock_block_storage):935 fake_volume = mock.MagicMock(id="volume_id")936 fake_snapshot = mock.MagicMock(id="snapshot_id")937 cinder = mock_block_storage.return_value938 cinder.create_volume.return_value = fake_volume939 cinder.create_snapshot.return_value = fake_snapshot940 scenario = servers.BootServerFromVolumeSnapshot(self.context,941 clients=mock.Mock())942 scenario._boot_server = mock.MagicMock()943 scenario.run("img", "flavor", 1, volume_type=None,944 auto_assign_nic=False, fakearg="f")945 cinder.create_volume.assert_called_once_with(1, imageRef="img",946 volume_type=None)947 cinder.create_snapshot.assert_called_once_with("volume_id",948 force=False)949 scenario._boot_server.assert_called_once_with(950 None, "flavor", auto_assign_nic=False,951 block_device_mapping={"vda": "snapshot_id:snap::1"},...

Full Screen

Full Screen

test_volumes.py

Source:test_volumes.py Github

copy

Full Screen

1# Copyright 2013 Huawei Technologies Co.,LTD.2# All Rights Reserved.3#4# Licensed under the Apache License, Version 2.0 (the "License"); you may5# not use this file except in compliance with the License. You may obtain6# a copy of the License at7#8# http://www.apache.org/licenses/LICENSE-2.09#10# Unless required by applicable law or agreed to in writing, software11# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT12# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the13# License for the specific language governing permissions and limitations14# under the License.15from unittest import mock16import ddt17from rally_openstack.task.scenarios.cinder import volumes18from tests.unit import test19CINDER_VOLUMES = ("rally_openstack.task.scenarios.cinder.volumes")20@ddt.ddt21class CinderServersTestCase(test.ScenarioTestCase):22 def _get_context(self):23 context = test.get_test_context()24 context.update({25 "admin": {26 "id": "fake_user_id",27 "credential": mock.MagicMock()28 },29 "user": {"id": "fake_user_id",30 "credential": mock.MagicMock()},31 "tenant": {"id": "fake", "name": "fake",32 "volumes": [{"id": "uuid", "size": 1}],33 "servers": [1]}})34 return context35 def setUp(self):36 super(CinderServersTestCase, self).setUp()37 patch = mock.patch(38 "rally_openstack.common.services.storage.block.BlockStorage")39 self.addCleanup(patch.stop)40 self.mock_cinder = patch.start()41 def test_create_and_list_volume(self):42 mock_service = self.mock_cinder.return_value43 scenario = volumes.CreateAndListVolume(self._get_context())44 scenario.run(1, True, fakearg="f")45 mock_service.create_volume.assert_called_once_with(1, fakearg="f")46 mock_service.list_volumes.assert_called_once_with(True)47 def test_create_and_get_volume(self):48 mock_service = self.mock_cinder.return_value49 scenario = volumes.CreateAndGetVolume(self._get_context())50 scenario.run(1, fakearg="f")51 mock_service.create_volume.assert_called_once_with(1, fakearg="f")52 mock_service.get_volume.assert_called_once_with(53 mock_service.create_volume.return_value.id)54 def test_list_volumes(self):55 mock_service = self.mock_cinder.return_value56 scenario = volumes.ListVolumes(self._get_context())57 scenario.run(True)58 mock_service.list_volumes.assert_called_once_with(59 True, limit=None, marker=None, search_opts=None, sort=None)60 def test_list_types(self):61 mock_service = self.mock_cinder.return_value62 scenario = volumes.ListTypes(self._get_context())63 scenario.run(None, is_public=None)64 mock_service.list_types.assert_called_once_with(None,65 is_public=None)66 def test_list_transfers(self):67 mock_service = self.mock_cinder.return_value68 scenario = volumes.ListTransfers(self._get_context())69 scenario._list_transfers = mock.MagicMock()70 scenario.run(True, search_opts=None)71 mock_service.list_transfers.assert_called_once_with(72 True, search_opts=None)73 @ddt.data({"update_args": {"description": "desp"},74 "expected": {"description": "desp"}},75 {"update_args": {"update_name": True, "description": "desp"},76 "expected": {"name": "new_name", "description": "desp"}})77 @ddt.unpack78 def test_create_and_update_volume(self, update_args, expected):79 mock_service = self.mock_cinder.return_value80 scenario = volumes.CreateAndUpdateVolume(self._get_context())81 scenario.generate_random_name = mock.MagicMock()82 scenario.generate_random_name.return_value = "new_name"83 scenario.run(1, update_volume_kwargs=update_args)84 mock_service.create_volume.assert_called_once_with(1)85 mock_service.update_volume.assert_called_once_with(86 mock_service.create_volume.return_value, **expected)87 if update_args.get("update_name", False):88 scenario.generate_random_name.assert_called_once_with()89 def test_create_volume_and_update_readonly_flag(self):90 mock_service = self.mock_cinder.return_value91 scenario = volumes.CreateVolumeAndUpdateReadonlyFlag(92 self._get_context())93 scenario.run(1, image=None, read_only=True, fakearg="f")94 mock_service.create_volume.assert_called_once_with(1, fakearg="f")95 mock_service.update_readonly_flag.assert_called_once_with(96 mock_service.create_volume.return_value.id, read_only=True)97 def test_create_and_delete_volume(self):98 mock_service = self.mock_cinder.return_value99 scenario = volumes.CreateAndDeleteVolume(self._get_context())100 scenario.sleep_between = mock.MagicMock()101 scenario.run(size=1, min_sleep=10, max_sleep=20, fakearg="f")102 mock_service.create_volume.assert_called_once_with(1, fakearg="f")103 scenario.sleep_between.assert_called_once_with(10, 20)104 mock_service.delete_volume.assert_called_once_with(105 mock_service.create_volume.return_value)106 def test_create_volume(self):107 mock_service = self.mock_cinder.return_value108 scenario = volumes.CreateVolume(self._get_context())109 scenario.run(1, fakearg="f")110 mock_service.create_volume.assert_called_once_with(1, fakearg="f")111 def test_create_volume_and_modify_metadata(self):112 mock_service = self.mock_cinder.return_value113 scenario = volumes.ModifyVolumeMetadata(self._get_context())114 scenario.run(sets=5, set_size=4, deletes=3, delete_size=2)115 mock_service.set_metadata.assert_called_once_with(116 "uuid", set_size=4, sets=5)117 mock_service.delete_metadata.assert_called_once_with(118 "uuid",119 keys=mock_service.set_metadata.return_value,120 deletes=3, delete_size=2)121 def test_create_and_extend_volume(self):122 mock_service = self.mock_cinder.return_value123 scenario = volumes.CreateAndExtendVolume(self._get_context())124 scenario.sleep_between = mock.MagicMock()125 scenario.run(1, 2, 10, 20, fakearg="f")126 mock_service.create_volume.assert_called_once_with(1, fakearg="f")127 mock_service.extend_volume.assert_called_once_with(128 mock_service.create_volume.return_value, new_size=2)129 scenario.sleep_between.assert_called_once_with(10, 20)130 mock_service.delete_volume.assert_called_once_with(131 mock_service.create_volume.return_value)132 def test_create_from_image_and_delete_volume(self):133 mock_service = self.mock_cinder.return_value134 scenario = volumes.CreateAndDeleteVolume(self._get_context())135 scenario.run(1, image="fake_image")136 mock_service.create_volume.assert_called_once_with(137 1, imageRef="fake_image")138 mock_service.delete_volume.assert_called_once_with(139 mock_service.create_volume.return_value)140 def test_create_volume_from_image(self):141 mock_service = self.mock_cinder.return_value142 scenario = volumes.CreateVolume(self._get_context())143 scenario.run(1, image="fake_image")144 mock_service.create_volume.assert_called_once_with(145 1, imageRef="fake_image")146 def test_create_volume_from_image_and_list(self):147 mock_service = self.mock_cinder.return_value148 scenario = volumes.CreateAndListVolume(self._get_context())149 scenario.run(1, True, "fake_image")150 mock_service.create_volume.assert_called_once_with(151 1, imageRef="fake_image")152 mock_service.list_volumes.assert_called_once_with(True)153 def test_create_from_volume_and_delete_volume(self):154 mock_service = self.mock_cinder.return_value155 vol_size = 1156 scenario = volumes.CreateFromVolumeAndDeleteVolume(self._get_context())157 scenario.run(vol_size)158 mock_service.create_volume.assert_called_once_with(159 1, source_volid="uuid")160 mock_service.delete_volume.assert_called_once_with(161 mock_service.create_volume.return_value)162 @mock.patch("%s.CreateAndDeleteSnapshot.sleep_between" % CINDER_VOLUMES)163 def test_create_and_delete_snapshot(self, mock_sleep_between):164 mock_service = self.mock_cinder.return_value165 scenario = volumes.CreateAndDeleteSnapshot(self._get_context())166 scenario.run(False, 10, 20, fakearg="f")167 mock_service.create_snapshot.assert_called_once_with("uuid",168 force=False,169 fakearg="f")170 mock_sleep_between.assert_called_once_with(10, 20)171 mock_service.delete_snapshot.assert_called_once_with(172 mock_service.create_snapshot.return_value)173 def test_create_and_list_snapshots(self):174 mock_service = self.mock_cinder.return_value175 scenario = volumes.CreateAndListSnapshots(self._get_context())176 scenario.run(False, True, fakearg="f")177 mock_service.create_snapshot.assert_called_once_with("uuid",178 force=False,179 fakearg="f")180 mock_service.list_snapshots.assert_called_once_with(True)181 def test_create_and_attach_volume(self):182 fake_server = mock.MagicMock()183 mock_service = self.mock_cinder.return_value184 scenario = volumes.CreateAndAttachVolume(self._get_context())185 scenario._boot_server = mock.MagicMock(return_value=fake_server)186 scenario._delete_server = mock.MagicMock()187 scenario._attach_volume = mock.MagicMock()188 scenario._detach_volume = mock.MagicMock()189 volume_args = {"some_key": "some_val"}190 vm_args = {"some_key": "some_val"}191 scenario.run(10, "img", "0",192 create_volume_params=volume_args,193 create_vm_params=vm_args)194 mock_service.create_volume.assert_called_once_with(195 10, **volume_args)196 scenario._attach_volume.assert_called_once_with(197 fake_server, mock_service.create_volume.return_value)198 scenario._detach_volume.assert_called_once_with(199 fake_server, mock_service.create_volume.return_value)200 mock_service.delete_volume.assert_called_once_with(201 mock_service.create_volume.return_value)202 scenario._delete_server.assert_called_once_with(fake_server)203 @mock.patch("rally_openstack.common.services.image.image.Image")204 def test_create_and_upload_volume_to_image(self, mock_image):205 mock_volume_service = self.mock_cinder.return_value206 mock_image_service = mock_image.return_value207 scenario = volumes.CreateAndUploadVolumeToImage(self._get_context())208 scenario.run(2, image="img", container_format="fake",209 disk_format="disk", do_delete=False, fakeargs="fakeargs")210 mock_volume_service.create_volume.assert_called_once_with(211 2, imageRef="img", fakeargs="fakeargs")212 mock_volume_service.upload_volume_to_image.assert_called_once_with(213 mock_volume_service.create_volume.return_value,214 container_format="fake", disk_format="disk", force=False)215 mock_volume_service.create_volume.reset_mock()216 mock_volume_service.upload_volume_to_image.reset_mock()217 scenario.run(1, image=None, do_delete=True, fakeargs="fakeargs")218 mock_volume_service.create_volume.assert_called_once_with(219 1, fakeargs="fakeargs")220 mock_volume_service.upload_volume_to_image.assert_called_once_with(221 mock_volume_service.create_volume.return_value,222 container_format="bare", disk_format="raw", force=False)223 mock_volume_service.delete_volume.assert_called_once_with(224 mock_volume_service.create_volume.return_value)225 mock_image_service.delete_image.assert_called_once_with(226 mock_volume_service.upload_volume_to_image.return_value.id)227 def test_create_snapshot_and_attach_volume(self):228 mock_service = self.mock_cinder.return_value229 scenario = volumes.CreateSnapshotAndAttachVolume(self._get_context())230 scenario._boot_server = mock.MagicMock()231 scenario._attach_volume = mock.MagicMock()232 scenario._detach_volume = mock.MagicMock()233 scenario.run("img", "flavor")234 self.assertTrue(mock_service.create_volume.called)235 volume = mock_service.create_volume.return_value236 snapshot = mock_service.create_snapshot.return_value237 mock_service.create_snapshot.assert_called_once_with(volume.id,238 force=False)239 mock_service.delete_snapshot.assert_called_once_with(snapshot)240 scenario._attach_volume.assert_called_once_with(241 scenario._boot_server.return_value, volume)242 scenario._detach_volume.assert_called_once_with(243 scenario._boot_server.return_value, volume)244 mock_service.delete_volume.assert_called_once_with(volume)245 @mock.patch("random.choice")246 def test_create_snapshot_and_attach_volume_use_volume_type_with_name(247 self, mock_choice):248 mock_service = self.mock_cinder.return_value249 scenario = volumes.CreateSnapshotAndAttachVolume(self._get_context())250 scenario._boot_server = mock.MagicMock()251 scenario._attach_volume = mock.MagicMock()252 scenario._detach_volume = mock.MagicMock()253 scenario.run("img", "flavor", volume_type="type")254 fake_volume = mock_service.create_volume.return_value255 fake_server = scenario._boot_server.return_value256 fake_snapshot = mock_service.create_snapshot.return_value257 mock_service.create_volume.assert_called_once_with(258 {"min": 1, "max": 5}, volume_type="type")259 mock_service.create_snapshot.assert_called_once_with(fake_volume.id,260 force=False)261 mock_service.delete_snapshot.assert_called_once_with(fake_snapshot)262 scenario._attach_volume.assert_called_once_with(fake_server,263 fake_volume)264 scenario._detach_volume.assert_called_once_with(fake_server,265 fake_volume)266 mock_service.delete_volume.assert_called_once_with(fake_volume)267 @mock.patch("random.randint")268 def test_create_nested_snapshots_and_attach_volume(self, mock_randint):269 mock_service = self.mock_cinder.return_value270 mock_randint.return_value = 2271 volume_kwargs = {"volume_type": "type1"}272 snapshot_kwargs = {"name": "snapshot1", "description": "snaphot one"}273 scenario = volumes.CreateNestedSnapshotsAndAttachVolume(274 context=self._get_context())275 scenario._boot_server = mock.MagicMock()276 scenario._attach_volume = mock.MagicMock()277 scenario._detach_volume = mock.MagicMock()278 scenario.run("img", "flavor", create_volume_kwargs=volume_kwargs,279 create_snapshot_kwargs=snapshot_kwargs)280 mock_service.create_volume.assert_called_once_with(281 mock_randint.return_value, **volume_kwargs)282 mock_service.create_snapshot.assert_called_once_with(283 mock_service.create_volume.return_value.id, force=False,284 **snapshot_kwargs)285 scenario._attach_volume(scenario._boot_server.return_value,286 mock_service.create_volume.return_value)287 mock_service.delete_volume.assert_called_once_with(288 mock_service.create_volume.return_value)289 mock_service.delete_snapshot.assert_called_once_with(290 mock_service.create_snapshot.return_value)291 scenario._detach_volume.assert_called_once_with(292 scenario._boot_server.return_value,293 mock_service.create_volume.return_value)294 @mock.patch("random.randint")295 def test_create_nested_snapshots_and_attach_volume_2(self, mock_randint):296 mock_service = self.mock_cinder.return_value297 mock_randint.return_value = 2298 nested_level = 3299 volume_size = mock_randint.return_value300 fake_volumes = [mock.Mock(size=volume_size)301 for i in range(nested_level)]302 fake_snapshots = [mock.Mock()303 for i in range(nested_level)]304 mock_service.create_volume.side_effect = fake_volumes305 mock_service.create_snapshot.side_effect = fake_snapshots306 scenario = volumes.CreateNestedSnapshotsAndAttachVolume(307 context=self._get_context())308 scenario._boot_server = mock.MagicMock()309 scenario._attach_volume = mock.MagicMock()310 scenario._detach_volume = mock.MagicMock()311 scenario.run("img", "flavor", nested_level=nested_level)312 expected_volumes = [mock.call(volume_size)]313 expected_snapshots = [mock.call(fake_volumes[0].id, force=False)]314 expected_attachs = [mock.call(scenario._boot_server.return_value,315 fake_volumes[0])]316 for i in range(nested_level - 1):317 expected_volumes.append(318 mock.call(volume_size, snapshot_id=fake_snapshots[i].id))319 expected_snapshots.append(320 mock.call(fake_volumes[i + 1].id, force=False))321 expected_attachs.append(322 mock.call(scenario._boot_server.return_value,323 fake_volumes[i + 1]))324 mock_service.create_volume.assert_has_calls(expected_volumes)325 mock_service.create_snapshot.assert_has_calls(expected_snapshots)326 scenario._attach_volume.assert_has_calls(expected_attachs)327 fake_volumes.reverse()328 fake_snapshots.reverse()329 mock_service.delete_volume.assert_has_calls(330 [mock.call(volume) for volume in fake_volumes])331 mock_service.delete_snapshot.assert_has_calls(332 [mock.call(snapshot) for snapshot in fake_snapshots])333 scenario._detach_volume.assert_has_calls(334 [mock.call(scenario._boot_server.return_value,335 fake_volumes[i])336 for i in range(len(fake_volumes))])337 def test_create_volume_backup(self):338 mock_service = self.mock_cinder.return_value339 scenario = volumes.CreateVolumeBackup(self._get_context())340 volume_kwargs = {"some_var": "zaq"}341 scenario.run(1, do_delete=True, create_volume_kwargs=volume_kwargs)342 mock_service.create_volume.assert_called_once_with(1, **volume_kwargs)343 mock_service.create_backup.assert_called_once_with(344 mock_service.create_volume.return_value.id)345 mock_service.delete_volume.assert_called_once_with(346 mock_service.create_volume.return_value)347 mock_service.delete_backup.assert_called_once_with(348 mock_service.create_backup.return_value)349 def test_create_volume_backup_no_delete(self):350 mock_service = self.mock_cinder.return_value351 scenario = volumes.CreateVolumeBackup(self._get_context())352 volume_kwargs = {"some_var": "zaq"}353 scenario.run(1, do_delete=False, create_volume_kwargs=volume_kwargs)354 mock_service.create_volume.assert_called_once_with(1, **volume_kwargs)355 mock_service.create_backup.assert_called_once_with(356 mock_service.create_volume.return_value.id)357 self.assertFalse(mock_service.delete_volume.called)358 self.assertFalse(mock_service.delete_backup.called)359 def test_create_and_restore_volume_backup(self):360 mock_service = self.mock_cinder.return_value361 volume_kwargs = {"some_var": "zaq"}362 scenario = volumes.CreateAndRestoreVolumeBackup(self._get_context())363 scenario.run(1, do_delete=True, create_volume_kwargs=volume_kwargs)364 fake_volume = mock_service.create_volume.return_value365 fake_backup = mock_service.create_backup.return_value366 mock_service.create_volume.assert_called_once_with(1, **volume_kwargs)367 mock_service.create_backup.assert_called_once_with(fake_volume.id)368 mock_service.restore_backup.assert_called_once_with(fake_backup.id)369 mock_service.delete_volume.assert_called_once_with(fake_volume)370 mock_service.delete_backup.assert_called_once_with(fake_backup)371 def test_create_and_restore_volume_backup_no_delete(self):372 mock_service = self.mock_cinder.return_value373 volume_kwargs = {"some_var": "zaq"}374 scenario = volumes.CreateAndRestoreVolumeBackup(self._get_context())375 scenario.run(1, do_delete=False, create_volume_kwargs=volume_kwargs)376 fake_volume = mock_service.create_volume.return_value377 fake_backup = mock_service.create_backup.return_value378 mock_service.create_volume.assert_called_once_with(1, **volume_kwargs)379 mock_service.create_backup.assert_called_once_with(fake_volume.id)380 mock_service.restore_backup.assert_called_once_with(fake_backup.id)381 self.assertFalse(mock_service.delete_volume.called)382 self.assertFalse(mock_service.delete_backup.called)383 def test_create_and_list_volume_backups(self):384 mock_service = self.mock_cinder.return_value385 volume_kwargs = {"some_var": "zaq"}386 scenario = volumes.CreateAndListVolumeBackups(self._get_context())387 scenario.run(1, detailed=True, do_delete=True,388 create_volume_kwargs=volume_kwargs)389 fake_volume = mock_service.create_volume.return_value390 fake_backup = mock_service.create_backup.return_value391 mock_service.create_volume.assert_called_once_with(1, **volume_kwargs)392 mock_service.create_backup.assert_called_once_with(fake_volume.id)393 mock_service.list_backups.assert_called_once_with(True)394 mock_service.delete_volume.assert_called_once_with(fake_volume)395 mock_service.delete_backup.assert_called_once_with(fake_backup)396 def test_create_and_list_volume_backups_no_delete(self):397 mock_service = self.mock_cinder.return_value398 volume_kwargs = {"some_var": "zaq"}399 scenario = volumes.CreateAndListVolumeBackups(self._get_context())400 scenario.run(1, detailed=True, do_delete=False,401 create_volume_kwargs=volume_kwargs)402 fake_volume = mock_service.create_volume.return_value403 mock_service.create_volume.assert_called_once_with(1, **volume_kwargs)404 mock_service.create_backup.assert_called_once_with(fake_volume.id)405 mock_service.list_backups.assert_called_once_with(True)406 self.assertFalse(mock_service.delete_volume.called)407 self.assertFalse(mock_service.delete_backup.called)408 @ddt.data({},409 {"nested_level": 2},410 {"image": "img"})411 @ddt.unpack412 def test_create_volume_and_clone(self, nested_level=1,413 image=None):414 create_volumes_count = nested_level + 1415 fake_volumes = [mock.Mock(size=1)416 for i in range(create_volumes_count)]417 mock_service = self.mock_cinder.return_value418 mock_service.create_volume.side_effect = fake_volumes419 scenario = volumes.CreateVolumeAndClone(self._get_context())420 scenario.run(1, image=image, nested_level=nested_level,421 fakearg="fake")422 expected = [mock.call(1, imageRef=image, fakearg="fake")423 if image else mock.call(1, fakearg="fake")]424 for i in range(nested_level):425 expected.append(mock.call(fake_volumes[i].size,426 source_volid=fake_volumes[i].id,427 fakearg="fake")428 )429 self._test_atomic_action_timer(scenario.atomic_actions(),430 "cinder.clone_volume",431 count=nested_level)432 mock_service.create_volume.assert_has_calls(expected)433 def test_create_volume_from_snapshot(self):434 mock_service = self.mock_cinder.return_value435 create_snapshot_args = {"force": False}436 scenario = volumes.CreateVolumeFromSnapshot(self._get_context())437 scenario.run(fakearg="f")438 fake_snapshot = mock_service.create_snapshot.return_value439 fake_volume = mock_service.create_volume.return_value440 mock_service.create_snapshot.assert_called_once_with("uuid")441 mock_service.create_volume.assert_called_once_with(442 1, snapshot_id=fake_snapshot.id, fakearg="f")443 mock_service.delete_snapshot.assert_called_once_with(fake_snapshot)444 mock_service.delete_volume.assert_called_once_with(fake_volume)445 mock_service.create_snapshot.reset_mock()446 mock_service.create_volume.reset_mock()447 mock_service.delete_snapshot.reset_mock()448 mock_service.delete_volume.reset_mock()449 scenario.run(do_delete=False,450 create_snapshot_kwargs=create_snapshot_args,451 fakearg="f")452 mock_service.create_snapshot.assert_called_once_with(453 "uuid", **create_snapshot_args)454 mock_service.create_volume.assert_called_once_with(455 1, snapshot_id=fake_snapshot.id, fakearg="f")456 self.assertFalse(mock_service.delete_snapshot.called)457 self.assertFalse(mock_service.delete_volume.called)458 @ddt.data({},459 {"image": "img"})460 @ddt.unpack461 def test_create_and_accept_transfer(self, image=None):462 mock_service = self.mock_cinder.return_value463 scenario = volumes.CreateAndAcceptTransfer(self._get_context())464 scenario.run(1, image=image, fakearg="fake")465 expected = [mock.call(1, imageRef=image, fakearg="fake")466 if image else mock.call(1, fakearg="fake")]467 mock_service.create_volume.assert_has_calls(expected)468 mock_service.transfer_create.assert_called_once_with(469 mock_service.create_volume.return_value.id)470 mock_service.transfer_accept.assert_called_once_with(471 mock_service.transfer_create.return_value.id,...

Full Screen

Full Screen

test_basic.py

Source:test_basic.py Github

copy

Full Screen

1# Copyright 2013: Mirantis Inc.2# All Rights Reserved.3#4# Licensed under the Apache License, Version 2.0 (the "License"); you may5# not use this file except in compliance with the License. You may obtain6# a copy of the License at7#8# http://www.apache.org/licenses/LICENSE-2.09#10# Unless required by applicable law or agreed to in writing, software11# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT12# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the13# License for the specific language governing permissions and limitations14# under the License.15from unittest import mock16import ddt17from rally import exceptions18from rally_openstack.task.scenarios.keystone import basic19from tests.unit import test20@ddt.ddt21class KeystoneBasicTestCase(test.ScenarioTestCase):22 def get_test_context(self):23 context = super(KeystoneBasicTestCase, self).get_test_context()24 context.update({25 "admin": {26 "id": "fake_user_id",27 "credential": mock.MagicMock()28 },29 "user": {30 "id": "fake_user_id",31 "credential": mock.MagicMock()32 },33 "tenant": {"id": "fake_tenant_id",34 "name": "fake_tenant_name"}35 })36 return context37 def setUp(self):38 super(KeystoneBasicTestCase, self).setUp()39 patch = mock.patch(40 "rally_openstack.common.services.identity.identity.Identity")41 self.addCleanup(patch.stop)42 self.mock_identity = patch.start()43 def test_create_user(self):44 scenario = basic.CreateUser(self.context)45 scenario.run(password="tttt", project_id="id")46 self.mock_identity.return_value.create_user.assert_called_once_with(47 password="tttt", project_id="id")48 def test_create_delete_user(self):49 identity_service = self.mock_identity.return_value50 fake_email = "abcd"51 fake_user = identity_service.create_user.return_value52 scenario = basic.CreateDeleteUser(self.context)53 scenario.run(email=fake_email, enabled=True)54 identity_service.create_user.assert_called_once_with(55 email=fake_email, enabled=True)56 identity_service.delete_user.assert_called_once_with(fake_user.id)57 def test_create_user_set_enabled_and_delete(self):58 identity_service = self.mock_identity.return_value59 scenario = basic.CreateUserSetEnabledAndDelete(self.context)60 fake_email = "abcd"61 fake_user = identity_service.create_user.return_value62 scenario.run(enabled=True, email=fake_email)63 identity_service.create_user.assert_called_once_with(64 email=fake_email, enabled=True)65 identity_service.update_user.assert_called_once_with(66 fake_user.id, enabled=False)67 identity_service.delete_user.assert_called_once_with(fake_user.id)68 def test_user_authenticate_and_validate_token(self):69 identity_service = self.mock_identity.return_value70 scenario = basic.AuthenticateUserAndValidateToken(self.context)71 fake_token = identity_service.fetch_token.return_value72 scenario.run()73 identity_service.fetch_token.assert_called_once_with()74 identity_service.validate_token.assert_called_once_with(fake_token)75 def test_create_tenant(self):76 scenario = basic.CreateTenant(self.context)77 scenario.run(enabled=True)78 self.mock_identity.return_value.create_project.assert_called_once_with(79 enabled=True)80 def test_create_tenant_with_users(self):81 identity_service = self.mock_identity.return_value82 fake_project = identity_service.create_project.return_value83 number_of_users = 184 scenario = basic.CreateTenantWithUsers(self.context)85 scenario.run(users_per_tenant=number_of_users, enabled=True)86 identity_service.create_project.assert_called_once_with(enabled=True)87 identity_service.create_users.assert_called_once_with(88 fake_project.id, number_of_users=number_of_users)89 def test_create_and_list_users(self):90 scenario = basic.CreateAndListUsers(self.context)91 passwd = "tttt"92 project_id = "id"93 scenario.run(password=passwd, project_id=project_id)94 self.mock_identity.return_value.create_user.assert_called_once_with(95 password=passwd, project_id=project_id)96 self.mock_identity.return_value.list_users.assert_called_once_with()97 def test_create_and_list_tenants(self):98 identity_service = self.mock_identity.return_value99 scenario = basic.CreateAndListTenants(self.context)100 scenario.run(enabled=True)101 identity_service.create_project.assert_called_once_with(enabled=True)102 identity_service.list_projects.assert_called_once_with()103 def test_assign_and_remove_user_role(self):104 fake_tenant = self.context["tenant"]["id"]105 fake_user = self.context["user"]["id"]106 fake_role = mock.MagicMock()107 self.mock_identity.return_value.create_role.return_value = fake_role108 scenario = basic.AddAndRemoveUserRole(self.context)109 scenario.run()110 self.mock_identity.return_value.create_role.assert_called_once_with()111 self.mock_identity.return_value.add_role.assert_called_once_with(112 role_id=fake_role.id, user_id=fake_user, project_id=fake_tenant)113 self.mock_identity.return_value.revoke_role.assert_called_once_with(114 fake_role.id, user_id=fake_user, project_id=fake_tenant)115 def test_create_and_delete_role(self):116 fake_role = mock.MagicMock()117 self.mock_identity.return_value.create_role.return_value = fake_role118 scenario = basic.CreateAndDeleteRole(self.context)119 scenario.run()120 self.mock_identity.return_value.create_role.assert_called_once_with()121 self.mock_identity.return_value.delete_role.assert_called_once_with(122 fake_role.id)123 def test_create_and_get_role(self):124 fake_role = mock.MagicMock()125 self.mock_identity.return_value.create_role.return_value = fake_role126 scenario = basic.CreateAndGetRole(self.context)127 scenario.run()128 self.mock_identity.return_value.create_role.assert_called_once_with()129 self.mock_identity.return_value.get_role.assert_called_once_with(130 fake_role.id)131 def test_create_and_list_user_roles(self):132 scenario = basic.CreateAddAndListUserRoles(self.context)133 fake_tenant = self.context["tenant"]["id"]134 fake_user = self.context["user"]["id"]135 fake_role = mock.MagicMock()136 self.mock_identity.return_value.create_role.return_value = fake_role137 scenario.run()138 self.mock_identity.return_value.create_role.assert_called_once_with()139 self.mock_identity.return_value.add_role.assert_called_once_with(140 user_id=fake_user, role_id=fake_role.id, project_id=fake_tenant)141 self.mock_identity.return_value.list_roles.assert_called_once_with(142 user_id=fake_user, project_id=fake_tenant)143 def test_create_and_list_roles(self):144 # Positive case145 scenario = basic.CreateAddListRoles(self.context)146 create_kwargs = {"fakewargs": "name"}147 list_kwargs = {"fakewargs": "f"}148 self.mock_identity.return_value.create_role = mock.Mock(149 return_value="role1")150 self.mock_identity.return_value.list_roles = mock.Mock(151 return_value=("role1", "role2"))152 scenario.run(create_role_kwargs=create_kwargs,153 list_role_kwargs=list_kwargs)154 self.mock_identity.return_value.create_role.assert_called_once_with(155 **create_kwargs)156 self.mock_identity.return_value.list_roles.assert_called_once_with(157 **list_kwargs)158 # Negative case 1: role isn't created159 self.mock_identity.return_value.create_role.return_value = None160 self.assertRaises(exceptions.RallyAssertionError,161 scenario.run, create_role_kwargs=create_kwargs,162 list_role_kwargs=list_kwargs)163 self.mock_identity.return_value.create_role.assert_called_with(164 **create_kwargs)165 # Negative case 2: role was created but included into list166 self.mock_identity.return_value.create_role.return_value = "role3"167 self.assertRaises(exceptions.RallyAssertionError,168 scenario.run, create_role_kwargs=create_kwargs,169 list_role_kwargs=list_kwargs)170 self.mock_identity.return_value.create_role.assert_called_with(171 **create_kwargs)172 self.mock_identity.return_value.list_roles.assert_called_with(173 **list_kwargs)174 @ddt.data(None, "keystone", "fooservice")175 def test_get_entities(self, service_name):176 identity_service = self.mock_identity.return_value177 fake_project = identity_service.create_project.return_value178 fake_user = identity_service.create_user.return_value179 fake_role = identity_service.create_role.return_value180 fake_service = identity_service.create_service.return_value181 scenario = basic.GetEntities(self.context)182 scenario.run(service_name)183 identity_service.create_project.assert_called_once_with()184 identity_service.create_user.assert_called_once_with(185 project_id=fake_project.id)186 identity_service.create_role.assert_called_once_with()187 identity_service.get_project.assert_called_once_with(fake_project.id)188 identity_service.get_user.assert_called_once_with(fake_user.id)189 identity_service.get_role.assert_called_once_with(fake_role.id)190 if service_name is None:191 identity_service.create_service.assert_called_once_with()192 self.assertFalse(identity_service.get_service_by_name.called)193 identity_service.get_service.assert_called_once_with(194 fake_service.id)195 else:196 identity_service.get_service_by_name.assert_called_once_with(197 service_name)198 self.assertFalse(identity_service.create_service.called)199 identity_service.get_service.assert_called_once_with(200 identity_service.get_service_by_name.return_value.id)201 def test_create_and_delete_service(self):202 identity_service = self.mock_identity.return_value203 scenario = basic.CreateAndDeleteService(self.context)204 service_type = "test_service_type"205 description = "test_description"206 fake_service = identity_service.create_service.return_value207 scenario.run(service_type=service_type, description=description)208 identity_service.create_service.assert_called_once_with(209 service_type=service_type, description=description)210 identity_service.delete_service.assert_called_once_with(211 fake_service.id)212 def test_create_update_and_delete_tenant(self):213 identity_service = self.mock_identity.return_value214 scenario = basic.CreateUpdateAndDeleteTenant(self.context)215 gen_name = mock.MagicMock()216 basic.CreateUpdateAndDeleteTenant.generate_random_name = gen_name217 fake_project = identity_service.create_project.return_value218 scenario.run()219 identity_service.create_project.assert_called_once_with()220 identity_service.update_project.assert_called_once_with(221 fake_project.id, description=gen_name.return_value,222 name=gen_name.return_value)223 identity_service.delete_project(fake_project.id)224 def test_create_user_update_password(self):225 identity_service = self.mock_identity.return_value226 scenario = basic.CreateUserUpdatePassword(self.context)227 fake_password = "pswd"228 fake_user = identity_service.create_user.return_value229 scenario.generate_random_name = mock.MagicMock(230 return_value=fake_password)231 scenario.run()232 scenario.generate_random_name.assert_called_once_with()233 identity_service.create_user.assert_called_once_with()234 identity_service.update_user.assert_called_once_with(235 fake_user.id, password=fake_password)236 def test_create_and_update_user(self):237 identity_service = self.mock_identity.return_value238 scenario = basic.CreateAndUpdateUser(self.context)239 scenario.admin_clients("keystone").users.get = mock.MagicMock()240 fake_user = identity_service.create_user.return_value241 create_args = {"fakearg1": "f"}242 update_args = {"fakearg1": "fakearg"}243 setattr(self.admin_clients("keystone").users.get.return_value,244 "fakearg1", "fakearg")245 scenario.run(create_user_kwargs=create_args,246 update_user_kwargs=update_args)247 identity_service.create_user.assert_called_once_with(**create_args)248 identity_service.update_user.assert_called_once_with(249 fake_user.id, **update_args)250 def test_create_and_list_services(self):251 identity_service = self.mock_identity.return_value252 scenario = basic.CreateAndListServices(self.context)253 service_type = "test_service_type"254 description = "test_description"255 scenario.run(service_type=service_type, description=description)256 identity_service.create_service.assert_called_once_with(257 service_type=service_type, description=description)258 identity_service.list_services.assert_called_once_with()259 def test_create_and_list_ec2credentials(self):260 identity_service = self.mock_identity.return_value261 scenario = basic.CreateAndListEc2Credentials(self.context)262 scenario.run()263 identity_service.create_ec2credentials.assert_called_once_with(264 self.context["user"]["id"],265 project_id=self.context["tenant"]["id"])266 identity_service.list_ec2credentials.assert_called_with(267 self.context["user"]["id"])268 def test_create_and_delete_ec2credential(self):269 identity_service = self.mock_identity.return_value270 fake_creds = identity_service.create_ec2credentials.return_value271 scenario = basic.CreateAndDeleteEc2Credential(self.context)272 scenario.run()273 identity_service.create_ec2credentials.assert_called_once_with(274 self.context["user"]["id"],275 project_id=self.context["tenant"]["id"])276 identity_service.delete_ec2credential.assert_called_once_with(277 self.context["user"]["id"], access=fake_creds.access)278 def test_add_and_remove_user_role(self):279 context = self.context280 tenant_id = context["tenant"]["id"]281 user_id = context["user"]["id"]282 fake_role = mock.MagicMock()283 self.mock_identity.return_value.create_role.return_value = fake_role284 scenario = basic.AddAndRemoveUserRole(context)285 scenario.run()286 self.mock_identity.return_value.create_role.assert_called_once_with()287 self.mock_identity.return_value.add_role.assert_called_once_with(288 role_id=fake_role.id, user_id=user_id, project_id=tenant_id)289 self.mock_identity.return_value.revoke_role.assert_called_once_with(...

Full Screen

Full Screen

test_init.py

Source:test_init.py Github

copy

Full Screen

...31 with patch_project, patch_at as mock_t, patch_tn as mock_tn, patch_l as mock_l:32 main(args_in=["init"])33 mock_tn.assert_called_once_with()34 mock_l.assert_called_once_with()35 mock_t.assert_called_once()36 mock_project.load_settings.assert_called_once_with()37 mock_project.init.assert_called_once_with(38 type_name,39 language,40 {41 "version": False,42 "subparser_name": None,43 "verbose": 0,44 "force": False,45 "type_name": None,46 "artifact_type": None,47 },48 )49 mock_project.generate.assert_called_once_with()50def test_init_module_method_interactive():51 type_name = object()52 language = object()53 mock_project, patch_project = get_mock_project()54 patch_tn = patch(55 "rpdk.core.module.init_module.input_typename", return_value=type_name56 )57 patch_l = patch(58 "rpdk.core.resource.init_resource.input_language", return_value=language59 )60 patch_at = patch("rpdk.core.init.init_artifact_type", return_value="MODULE")61 with patch_project, patch_tn as mock_tn, patch_l as mock_l, patch_at as mock_t:62 main(args_in=["init"])63 mock_tn.assert_called_once_with()64 mock_l.assert_not_called()65 mock_t.assert_called_once()66 mock_project.load_settings.assert_called_once_with()67 mock_project.init_module.assert_called_once_with(type_name)68 mock_project.generate.assert_not_called()69def test_init_resource_method_noninteractive():70 add_dummy_language_plugin()71 artifact_type = "RESOURCE"72 args = get_args("dummy", "Test::Test::Test", artifact_type)73 mock_project, patch_project = get_mock_project()74 patch_get_parser = patch(75 "rpdk.core.init.get_parsers", return_value={"dummy": dummy_parser}76 )77 # flake8: noqa: B95078 # pylint: disable=line-too-long79 with patch_project, patch_get_parser as mock_parser:80 main(81 args_in=[82 "init",83 "--type-name",84 args.type_name,85 "--artifact-type",86 args.artifact_type,87 args.language,88 "--dummy",89 ]90 )91 mock_parser.assert_called_once()92 mock_project.load_settings.assert_called_once_with()93 mock_project.init.assert_called_once_with(94 args.type_name,95 args.language,96 {97 "version": False,98 "subparser_name": args.language,99 "verbose": 0,100 "force": False,101 "type_name": args.type_name,102 "language": args.language,103 "dummy": True,104 "artifact_type": artifact_type,105 },106 )107 mock_project.generate.assert_called_once_with()108def test_init_resource_method_noninteractive_invalid_type_name():109 add_dummy_language_plugin()110 type_name = object()111 artifact_type = "RESOURCE"112 args = get_args("dummy", "invalid_type_name", "RESOURCE")113 mock_project, patch_project = get_mock_project()114 patch_tn = patch(115 "rpdk.core.resource.init_resource.input_typename", return_value=type_name116 )117 patch_t = patch("rpdk.core.init.init_artifact_type", return_value=artifact_type)118 patch_get_parser = patch(119 "rpdk.core.init.get_parsers", return_value={"dummy": dummy_parser}120 )121 # flake8: noqa: B950122 # pylint: disable=line-too-long123 with patch_project, patch_t, patch_tn as mock_tn, patch_get_parser as mock_parser:124 main(125 args_in=[126 "init",127 "-t",128 args.type_name,129 "-a",130 args.artifact_type,131 args.language,132 "--dummy",133 ]134 )135 mock_tn.assert_called_once_with()136 mock_parser.assert_called_once()137 mock_project.load_settings.assert_called_once_with()138 mock_project.init.assert_called_once_with(139 type_name,140 args.language,141 {142 "version": False,143 "subparser_name": args.language,144 "verbose": 0,145 "force": False,146 "type_name": args.type_name,147 "artifact_type": artifact_type,148 "dummy": True,149 "language": args.language,150 },151 )152 mock_project.generate.assert_called_once_with()153def test_input_with_validation_valid_first_try(capsys):154 sentinel1 = object()155 sentinel2 = object()156 validator = Mock(return_value=sentinel1)157 with patch("rpdk.core.init.input", return_value=sentinel2) as mock_input:158 ret = input_with_validation(PROMPT, validator)159 mock_input.assert_called_once_with()160 validator.assert_called_once_with(sentinel2)161 assert ret is sentinel1162 out, err = capsys.readouterr()163 assert not err164 assert PROMPT in out165def test_input_with_validation_valid_second_try(capsys):166 def mock_validator(value):167 if value == ERROR:168 raise WizardValidationError(ERROR)169 return value170 sentinel = object()171 with patch("rpdk.core.init.input", side_effect=(ERROR, sentinel)) as mock_input:172 ret = input_with_validation(PROMPT, mock_validator)173 assert mock_input.call_count == 2174 assert ret is sentinel175 out, err = capsys.readouterr()176 assert not err177 assert ERROR in out178def test_validate_type_name_valid():179 assert validate_type_name("AWS::Color::Red") == "AWS::Color::Red"180def test_validate_type_name_invalid():181 with pytest.raises(WizardValidationError):182 validate_type_name("AWS-Color-Red")183@pytest.mark.parametrize("value", ("y", "yes", "Y", "YES", "yEs", "YeS"))184def test_validate_yes_yes(value):185 assert validate_yes(value)186@pytest.mark.parametrize("value", ("n", "N", "no", "NO", "yesn't"))187def test_validate_yes_no(value):188 assert not validate_yes(value)189def test_validate_plugin_choice_not_an_int():190 validator = ValidatePluginChoice(["test"])191 with pytest.raises(WizardValidationError) as excinfo:192 validator("a")193 assert "integer" in str(excinfo.value)194def test_validate_plugin_choice_less_than_zero():195 validator = ValidatePluginChoice(["test"])196 with pytest.raises(WizardValidationError) as excinfo:197 validator("-1")198 assert "select" in str(excinfo.value)199def test_validate_plugin_choice_greater_than_choice():200 choices = range(3)201 validator = ValidatePluginChoice(choices)202 with pytest.raises(WizardValidationError) as excinfo:203 validator(str(len(choices) + 1)) # index is 1 based for input204 assert "select" in str(excinfo.value)205def test_validate_plugin_choice_valid():206 choices = ["1", PROMPT, "2"]207 validator = ValidatePluginChoice(choices)208 assert validator("2") == PROMPT209def test_init_module_method_noninteractive():210 add_dummy_language_plugin()211 artifact_type = "MODULE"212 args = get_args("dummy", "Test::Test::Test::MODULE", artifact_type)213 mock_project, patch_project = get_mock_project()214 patch_get_parser = patch(215 "rpdk.core.init.get_parsers", return_value={"dummy": dummy_parser}216 )217 # flake8: noqa: B950218 # pylint: disable=line-too-long219 with patch_project, patch_get_parser as mock_parser:220 main(221 args_in=[222 "init",223 "--type-name",224 args.type_name,225 "--artifact-type",226 args.artifact_type,227 args.language,228 "--dummy",229 ]230 )231 mock_parser.assert_called_once()232 mock_project.load_settings.assert_called_once_with()233 mock_project.init_module.assert_called_once_with(args.type_name)234 mock_project.generate.assert_not_called()235def test_check_for_existing_project_good_path():236 project = Mock(spec=Project)237 project.load_settings.side_effect = FileNotFoundError238 type(project).overwrite_enabled = mock_overwrite = PropertyMock()239 check_for_existing_project(project)240 project.load_settings.assert_called_once_with()241 mock_overwrite.assert_not_called()242def test_check_for_existing_project_bad_path_overwrite():243 project = Mock(spec=Project)244 project.overwrite_enabled = True245 project.settings_path = "" # not sure why this doesn't get specced?246 project.type_name = ""247 with patch("rpdk.core.init.input_with_validation", autospec=True) as mock_input:248 check_for_existing_project(project)249 mock_input.assert_not_called()250def test_check_for_existing_project_bad_path_ask_yes():251 project = Mock(spec=Project)252 project.overwrite_enabled = False253 project.settings_path = ""254 project.type_name = ""255 patch_input = patch(256 "rpdk.core.init.input_with_validation", autospec=True, return_value="m"257 )258 with patch_input as mock_input:259 check_for_existing_project(project)260 mock_input.assert_called_once_with(ANY, validate_yes, ANY)261 assert project.overwrite_enabled262def test_check_for_existing_project_bad_path_ask_no():263 project = Mock(spec=Project)264 project.overwrite_enabled = False265 project.settings_path = ""266 project.type_name = ""267 patch_input = patch(268 "rpdk.core.init.input_with_validation", autospec=True, return_value=False269 )270 with patch_input as mock_input:271 with pytest.raises(WizardAbortError):272 check_for_existing_project(project)273 mock_input.assert_called_once_with(ANY, validate_yes, ANY)274 assert not project.overwrite_enabled275def test_ignore_abort_ok():276 sentinel = object()277 function = Mock()278 wrapped = ignore_abort(function)279 wrapped(sentinel)280 function.assert_called_once_with(sentinel)281def test_ignore_abort_keyboard_interrupt():282 sentinel = object()283 function = Mock(side_effect=KeyboardInterrupt)284 wrapped = ignore_abort(function)285 with pytest.raises(SystemExit):286 wrapped(sentinel)287 function.assert_called_once_with(sentinel)288def test_ignore_abort_abort():289 sentinel = object()290 function = Mock(side_effect=WizardAbortError)291 wrapped = ignore_abort(function)292 with pytest.raises(SystemExit):293 wrapped(sentinel)294 function.assert_called_once_with(sentinel)295def test_input_typename_resource():296 type_name = "AWS::Color::Red"297 patch_input = patch("rpdk.core.utils.init_utils.input", return_value=type_name)298 with patch_input as mock_input:299 assert input_typename_resource() == type_name300 mock_input.assert_called_once()301def test_input_typename_module():302 type_name = "AWS::Color::Red::MODULE"303 patch_input = patch("rpdk.core.utils.init_utils.input", return_value=type_name)304 with patch_input as mock_input:305 assert input_typename_module() == type_name306 mock_input.assert_called_once()307def test_input_language_no_plugins():308 validator = ValidatePluginChoice([])309 with patch("rpdk.core.init.validate_plugin_choice", validator):310 with pytest.raises(WizardAbortError):311 input_language()312def test_input_language_one_plugin():313 validator = ValidatePluginChoice([PROMPT])314 with patch("rpdk.core.init.validate_plugin_choice", validator):315 assert input_language() == PROMPT316def test_input_language_several_plugins():317 validator = ValidatePluginChoice(["1", PROMPT, "2"])318 patch_validator = patch("rpdk.core.init.validate_plugin_choice", validator)319 patch_input = patch("rpdk.core.init.input", return_value="2")320 with patch_validator, patch_input as mock_input:321 assert input_language() == PROMPT...

Full Screen

Full Screen

fixtures.py

Source:fixtures.py Github

copy

Full Screen

1# testing/fixtures.py2# Copyright (C) 2005-2019 the SQLAlchemy authors and contributors3# <see AUTHORS file>4#5# This module is part of SQLAlchemy and is released under6# the MIT License: http://www.opensource.org/licenses/mit-license.php7import sys8import sqlalchemy as sa9from . import assertions10from . import config11from . import schema12from .engines import drop_all_tables13from .entities import BasicEntity14from .entities import ComparableEntity15from .util import adict16from .. import event17from .. import util18from ..ext.declarative import declarative_base19from ..ext.declarative import DeclarativeMeta20# whether or not we use unittest changes things dramatically,21# as far as how py.test collection works.22class TestBase(object):23 # A sequence of database names to always run, regardless of the24 # constraints below.25 __whitelist__ = ()26 # A sequence of requirement names matching testing.requires decorators27 __requires__ = ()28 # A sequence of dialect names to exclude from the test class.29 __unsupported_on__ = ()30 # If present, test class is only runnable for the *single* specified31 # dialect. If you need multiple, use __unsupported_on__ and invert.32 __only_on__ = None33 # A sequence of no-arg callables. If any are True, the entire testcase is34 # skipped.35 __skip_if__ = None36 def assert_(self, val, msg=None):37 assert val, msg38 # apparently a handful of tests are doing this....OK39 def setup(self):40 if hasattr(self, "setUp"):41 self.setUp()42 def teardown(self):43 if hasattr(self, "tearDown"):44 self.tearDown()45class TablesTest(TestBase):46 # 'once', None47 run_setup_bind = "once"48 # 'once', 'each', None49 run_define_tables = "once"50 # 'once', 'each', None51 run_create_tables = "once"52 # 'once', 'each', None53 run_inserts = "each"54 # 'each', None55 run_deletes = "each"56 # 'once', None57 run_dispose_bind = None58 bind = None59 metadata = None60 tables = None61 other = None62 @classmethod63 def setup_class(cls):64 cls._init_class()65 cls._setup_once_tables()66 cls._setup_once_inserts()67 @classmethod68 def _init_class(cls):69 if cls.run_define_tables == "each":70 if cls.run_create_tables == "once":71 cls.run_create_tables = "each"72 assert cls.run_inserts in ("each", None)73 cls.other = adict()74 cls.tables = adict()75 cls.bind = cls.setup_bind()76 cls.metadata = sa.MetaData()77 cls.metadata.bind = cls.bind78 @classmethod79 def _setup_once_inserts(cls):80 if cls.run_inserts == "once":81 cls._load_fixtures()82 cls.insert_data()83 @classmethod84 def _setup_once_tables(cls):85 if cls.run_define_tables == "once":86 cls.define_tables(cls.metadata)87 if cls.run_create_tables == "once":88 cls.metadata.create_all(cls.bind)89 cls.tables.update(cls.metadata.tables)90 def _setup_each_tables(self):91 if self.run_define_tables == "each":92 self.tables.clear()93 if self.run_create_tables == "each":94 drop_all_tables(self.metadata, self.bind)95 self.metadata.clear()96 self.define_tables(self.metadata)97 if self.run_create_tables == "each":98 self.metadata.create_all(self.bind)99 self.tables.update(self.metadata.tables)100 elif self.run_create_tables == "each":101 drop_all_tables(self.metadata, self.bind)102 self.metadata.create_all(self.bind)103 def _setup_each_inserts(self):104 if self.run_inserts == "each":105 self._load_fixtures()106 self.insert_data()107 def _teardown_each_tables(self):108 # no need to run deletes if tables are recreated on setup109 if self.run_define_tables != "each" and self.run_deletes == "each":110 with self.bind.connect() as conn:111 for table in reversed(self.metadata.sorted_tables):112 try:113 conn.execute(table.delete())114 except sa.exc.DBAPIError as ex:115 util.print_(116 ("Error emptying table %s: %r" % (table, ex)),117 file=sys.stderr,118 )119 def setup(self):120 self._setup_each_tables()121 self._setup_each_inserts()122 def teardown(self):123 self._teardown_each_tables()124 @classmethod125 def _teardown_once_metadata_bind(cls):126 if cls.run_create_tables:127 drop_all_tables(cls.metadata, cls.bind)128 if cls.run_dispose_bind == "once":129 cls.dispose_bind(cls.bind)130 cls.metadata.bind = None131 if cls.run_setup_bind is not None:132 cls.bind = None133 @classmethod134 def teardown_class(cls):135 cls._teardown_once_metadata_bind()136 @classmethod137 def setup_bind(cls):138 return config.db139 @classmethod140 def dispose_bind(cls, bind):141 if hasattr(bind, "dispose"):142 bind.dispose()143 elif hasattr(bind, "close"):144 bind.close()145 @classmethod146 def define_tables(cls, metadata):147 pass148 @classmethod149 def fixtures(cls):150 return {}151 @classmethod152 def insert_data(cls):153 pass154 def sql_count_(self, count, fn):155 self.assert_sql_count(self.bind, fn, count)156 def sql_eq_(self, callable_, statements):157 self.assert_sql(self.bind, callable_, statements)158 @classmethod159 def _load_fixtures(cls):160 """Insert rows as represented by the fixtures() method."""161 headers, rows = {}, {}162 for table, data in cls.fixtures().items():163 if len(data) < 2:164 continue165 if isinstance(table, util.string_types):166 table = cls.tables[table]167 headers[table] = data[0]168 rows[table] = data[1:]169 for table in cls.metadata.sorted_tables:170 if table not in headers:171 continue172 cls.bind.execute(173 table.insert(),174 [175 dict(zip(headers[table], column_values))176 for column_values in rows[table]177 ],178 )179class RemovesEvents(object):180 @util.memoized_property181 def _event_fns(self):182 return set()183 def event_listen(self, target, name, fn, **kw):184 self._event_fns.add((target, name, fn))185 event.listen(target, name, fn, **kw)186 def teardown(self):187 for key in self._event_fns:188 event.remove(*key)189 super_ = super(RemovesEvents, self)190 if hasattr(super_, "teardown"):191 super_.teardown()192class _ORMTest(object):193 @classmethod194 def teardown_class(cls):195 sa.orm.session.Session.close_all()196 sa.orm.clear_mappers()197class ORMTest(_ORMTest, TestBase):198 pass199class MappedTest(_ORMTest, TablesTest, assertions.AssertsExecutionResults):200 # 'once', 'each', None201 run_setup_classes = "once"202 # 'once', 'each', None203 run_setup_mappers = "each"204 classes = None205 @classmethod206 def setup_class(cls):207 cls._init_class()208 if cls.classes is None:209 cls.classes = adict()210 cls._setup_once_tables()211 cls._setup_once_classes()212 cls._setup_once_mappers()213 cls._setup_once_inserts()214 @classmethod215 def teardown_class(cls):216 cls._teardown_once_class()217 cls._teardown_once_metadata_bind()218 def setup(self):219 self._setup_each_tables()220 self._setup_each_classes()221 self._setup_each_mappers()222 self._setup_each_inserts()223 def teardown(self):224 sa.orm.session.Session.close_all()225 self._teardown_each_mappers()226 self._teardown_each_classes()227 self._teardown_each_tables()228 @classmethod229 def _teardown_once_class(cls):230 cls.classes.clear()231 _ORMTest.teardown_class()232 @classmethod233 def _setup_once_classes(cls):234 if cls.run_setup_classes == "once":235 cls._with_register_classes(cls.setup_classes)236 @classmethod237 def _setup_once_mappers(cls):238 if cls.run_setup_mappers == "once":239 cls._with_register_classes(cls.setup_mappers)240 def _setup_each_mappers(self):241 if self.run_setup_mappers == "each":242 self._with_register_classes(self.setup_mappers)243 def _setup_each_classes(self):244 if self.run_setup_classes == "each":245 self._with_register_classes(self.setup_classes)246 @classmethod247 def _with_register_classes(cls, fn):248 """Run a setup method, framing the operation with a Base class249 that will catch new subclasses to be established within250 the "classes" registry.251 """252 cls_registry = cls.classes253 class FindFixture(type):254 def __init__(cls, classname, bases, dict_):255 cls_registry[classname] = cls256 return type.__init__(cls, classname, bases, dict_)257 class _Base(util.with_metaclass(FindFixture, object)):258 pass259 class Basic(BasicEntity, _Base):260 pass261 class Comparable(ComparableEntity, _Base):262 pass263 cls.Basic = Basic264 cls.Comparable = Comparable265 fn()266 def _teardown_each_mappers(self):267 # some tests create mappers in the test bodies268 # and will define setup_mappers as None -269 # clear mappers in any case270 if self.run_setup_mappers != "once":271 sa.orm.clear_mappers()272 def _teardown_each_classes(self):273 if self.run_setup_classes != "once":274 self.classes.clear()275 @classmethod276 def setup_classes(cls):277 pass278 @classmethod279 def setup_mappers(cls):280 pass281class DeclarativeMappedTest(MappedTest):282 run_setup_classes = "once"283 run_setup_mappers = "once"284 @classmethod285 def _setup_once_tables(cls):286 pass287 @classmethod288 def _with_register_classes(cls, fn):289 cls_registry = cls.classes290 class FindFixtureDeclarative(DeclarativeMeta):291 def __init__(cls, classname, bases, dict_):292 cls_registry[classname] = cls293 return DeclarativeMeta.__init__(cls, classname, bases, dict_)294 class DeclarativeBasic(object):295 __table_cls__ = schema.Table296 _DeclBase = declarative_base(297 metadata=cls.metadata,298 metaclass=FindFixtureDeclarative,299 cls=DeclarativeBasic,300 )301 cls.DeclarativeBasic = _DeclBase302 fn()303 if cls.metadata.tables and cls.run_create_tables:...

Full Screen

Full Screen

test_artifact_bundles.py

Source:test_artifact_bundles.py Github

copy

Full Screen

...30 self.host = '127.0.0.1'31 self.connection = connection(self.host)32 self._client = ArtifactBundles(self.connection)33 @mock.patch.object(ResourceClient, 'get_all')34 def test_get_all_called_once(self, mock_get_all):35 filter = 'name=TestName'36 sort = 'name: ascending'37 self._client.get_all(2, 500, filter, sort)38 mock_get_all.assert_called_once_with(2, 500, filter=filter, sort=sort)39 @mock.patch.object(ResourceClient, 'get_all')40 def test_get_all_called_once_with_default_values(self, mock_get_all):41 self._client.get_all()42 mock_get_all.assert_called_once_with(0, -1, filter='', sort='')43 @mock.patch.object(ResourceClient, 'get')44 def test_get_called_once(self, mock_get):45 self._client.get('78836581-2b6f-4e26-9969-5667fb5837b4')46 mock_get.assert_called_once_with(47 '78836581-2b6f-4e26-9969-5667fb5837b4')48 @mock.patch.object(ResourceClient, 'get')49 def test_get_with_uri_called_once(self, mock_get):50 uri = '/rest/artifact-bundles/78836581-2b6f-4e26-9969-5667fb5837b4'51 self._client.get(uri)52 mock_get.assert_called_once_with(uri)53 @mock.patch.object(ResourceClient, 'get_by')54 def test_get_by_called_once(self, mock_get_by):55 self._client.get_by('name', 'ArtifactBundle')56 mock_get_by.assert_called_once_with(57 'name', 'ArtifactBundle')58 @mock.patch.object(ResourceClient, 'get_by_name')59 def test_get_by_name_called_once(self, mock_get_by):60 self._client.get_by_name('ArtifactBundle')61 mock_get_by.assert_called_once_with('ArtifactBundle')62 @mock.patch.object(ResourceClient, 'create')63 def test_create_called_once(self, mock_create):64 information = {65 "name": "ArtifactBundle"66 }67 mock_create.return_value = {}68 self._client.create(information)69 mock_create.assert_called_once_with(information.copy(), timeout=-1)70 @mock.patch.object(ResourceClient, 'update')71 def test_update_called_once(self, mock_update):72 information = {73 "name": "ArtifactBundleUpdate",74 "uri": "/rest/artifact-bundles/78836581-2b6f-4e26-9969-5667fb5837b4"75 }76 mock_update.return_value = {}77 self._client.update(information)78 default_values = {u'300': {u'type': u'ArtifactsBundle'}}79 mock_update.assert_called_once_with(information.copy(), default_values=default_values, timeout=-1)80 @mock.patch.object(ResourceClient, 'delete')81 def test_remove_called_once(self, mock_delete):82 id = '78836581-2b6f-4e26-9969-5667fb5837b4'83 self._client.delete(id)84 mock_delete.assert_called_once_with(id, timeout=-1)85 @mock.patch.object(ResourceClient, 'get_collection')86 def test_get_all_backups_called_once(self, mock_get_collection):87 uri = '/rest/artifact-bundles/backups'88 self._client.get_all_backups()89 mock_get_collection.assert_called_once_with(id_or_uri=uri)90 @mock.patch.object(ResourceClient, 'get')91 def test_get_backups_by_id_called_once(self, mock_get):92 uri = '/rest/artifact-bundles/backups'93 id = '78836581-2b6f-4e26-9969-5667fb5837b4'94 self._client.get_backup(id)95 mock_get.assert_called_once_with(id_or_uri=uri + '/' + id)96 @mock.patch.object(ResourceClient, 'download')97 def test_download_called_once_by_id(self, mock_download):98 destination = '~/image.zip'99 self._client.download_artifact_bundle('0ABDE00534F', destination)100 mock_download.assert_called_once_with('/rest/artifact-bundles/download/0ABDE00534F', destination)101 @mock.patch.object(ResourceClient, 'download')102 def test_download_called_once_by_uri(self, mock_download):103 uri = '/rest/artifact-bundles/0ABDE00534F'104 destination = '~/image.zip'105 self._client.download_artifact_bundle(uri, destination)106 mock_download.assert_called_once_with('/rest/artifact-bundles/download/0ABDE00534F', destination)107 @mock.patch.object(ResourceClient, 'download')108 def test_download_archive_artifact_bundle_by_id_called_once(self, mock_download):109 id = '78836581-2b6f-4e26-9969-5667fb5837b4'110 destination = '~/image.zip'111 self._client.download_archive_artifact_bundle(id, destination)112 expected_uri = '/rest/artifact-bundles/backups/archive/78836581-2b6f-4e26-9969-5667fb5837b4'113 mock_download.assert_called_once_with(expected_uri, destination)114 @mock.patch.object(ResourceClient, 'download')115 def test_download_archive_artifact_bundle_by_uri_called_once(self, mock_download):116 uri = '/rest/artifact-bundles/78836581-2b6f-4e26-9969-5667fb5837b4'117 destination = '~/image.zip'118 self._client.download_archive_artifact_bundle(uri, destination)119 expected_uri = '/rest/artifact-bundles/backups/archive/78836581-2b6f-4e26-9969-5667fb5837b4'120 mock_download.assert_called_once_with(expected_uri, destination)121 @mock.patch.object(ResourceClient, 'create')122 def test_create_backup_called_once(self, mock_create):123 information = {124 "deploymentGroupURI": "/rest/deployment-groups/00c1344d-e4dd-43c3-a733-1664e159a36f"125 }126 mock_create.return_value = {}127 self._client.create_backup(information)128 uri = '/rest/artifact-bundles/backups'129 mock_create.assert_called_once_with(information.copy(), uri=uri, timeout=-1)130 @mock.patch.object(ResourceClient, 'upload')131 def test_upload_artifact_bundle_called_once(self, mock_upload):132 filepath = "~/HPE-ImageStreamer-Developer-2016-09-12.zip"133 self._client.upload_bundle_from_file(filepath)134 mock_upload.assert_called_once_with(filepath)135 @mock.patch.object(ResourceClient, 'upload')136 def test_upload_backup_artifact_bundle_called_once(self, mock_upload):137 filepath = "~/HPE-ImageStreamer-Developer-2016-09-12.zip"138 deployment_groups = "/rest/deployment-groups/00c1344d-e4dd-43c3-a733-1664e159a36f"139 self._client.upload_backup_bundle_from_file(filepath, deployment_groups)140 expected_uri = '/rest/artifact-bundles/backups/archive?deploymentGrpUri=' + deployment_groups141 mock_upload.assert_called_once_with(filepath, expected_uri)142 @mock.patch.object(ResourceClient, 'upload')143 def test_upload_backup_artifact_bundle_called_once_with_id(self, mock_upload):144 filepath = "~/HPE-ImageStreamer-Developer-2016-09-12.zip"145 deployment_groups = "00c1344d-e4dd-43c3-a733-1664e159a36f"146 self._client.upload_backup_bundle_from_file(filepath, deployment_groups)147 expected_uri = '/rest/artifact-bundles/backups/archive?' \148 'deploymentGrpUri=/rest/deployment-groups/00c1344d-e4dd-43c3-a733-1664e159a36f'149 mock_upload.assert_called_once_with(filepath, expected_uri)150 @mock.patch.object(ResourceClient, 'update')151 def test_extract_called_once(self, mock_update):152 mock_update.return_value = {}153 uri = {'uri': '/rest/artifact-bundles/78836581-2b6f-4e26-9969-5667fb5837b4'}154 custom_headers = {'Content-Type': 'text/plain'}155 self._client.extract_bundle(uri)156 mock_update.assert_called_once_with(uri, custom_headers=custom_headers, timeout=-1)157 @mock.patch.object(ResourceClient, 'update')158 def test_extract_backup_bundle_called_once(self, mock_update):159 mock_update.return_value = {}160 data = {161 'deploymentGroupURI': '/rest/deployment-groups/00c1344d-e4dd-43c3-a733-1664e159a36f'162 }163 uri = '/rest/artifact-bundles/backups/archive'164 self._client.extract_backup_bundle(data)165 mock_update.assert_called_once_with(data, uri=uri, timeout=-1)166 @mock.patch.object(ResourceClient, 'update')167 def test_stop_creation_called_once(self, mock_update):168 mock_update.return_value = {}169 artifact_uri = "/rest/artifact-bundles/04939e89-bcb0-49fc-814f-1a6bc0a2f63c"170 task_uri = "/rest/tasks/A15F9270-46FC-48DF-94A9-D11EDB52877E"171 self._client.stop_artifact_creation(artifact_uri, task_uri)172 uri = artifact_uri + '/stopArtifactCreate'173 task_uri = {174 'taskUri': task_uri175 }...

Full Screen

Full Screen

test_power_devices.py

Source:test_power_devices.py Github

copy

Full Screen

...46 def test_get_utilization_with_defaults(self, mock_get):47 self._power_devices.get_utilization('35323930-4936-4450-5531-303153474820')48 mock_get.assert_called_once_with('35323930-4936-4450-5531-303153474820', None, None, False, None)49 @mock.patch.object(ResourceClient, 'get_all')50 def test_get_all_called_once(self, mock_get_all):51 filter = 'name=TestName'52 sort = 'name:ascending'53 self._power_devices.get_all(2, 500, filter=filter, sort=sort)54 mock_get_all.assert_called_once_with(2, 500, filter=filter, sort=sort, query='')55 @mock.patch.object(ResourceClient, 'get_all')56 def test_get_all_called_once_with_default(self, mock_get_all):57 self._power_devices.get_all()58 mock_get_all.assert_called_once_with(0, -1, filter='', sort='', query='')59 @mock.patch.object(ResourceClient, 'get')60 def test_get_by_id_called_once(self, mock_get):61 rack_id = "f0a0a113-ec97-41b4-83ce-d7c92b900e7c"62 self._power_devices.get(rack_id)63 mock_get.assert_called_once_with(rack_id)64 @mock.patch.object(ResourceClient, 'get')65 def test_get_by_uri_called_once(self, mock_get):66 rack_uri = "/rest/racks/f0a0a113-ec97-41b4-83ce-d7c92b900e7c"67 self._power_devices.get(rack_uri)68 mock_get.assert_called_once_with(rack_uri)69 @mock.patch.object(ResourceClient, 'get_by')70 def test_get_by_called_once(self, mock_get_by):71 self._power_devices.get_by("name", "test name")72 mock_get_by.assert_called_once_with("name", "test name")73 @mock.patch.object(ResourceClient, 'create')74 def test_add_called_once_with_defaults(self, mock_create):75 rack = {76 "name": "MyRack"77 }78 self._power_devices.add(rack)79 mock_create.assert_called_once_with(rack, timeout=-1)80 @mock.patch.object(ResourceClient, 'create')81 def test_create_called_once(self, mock_create):82 rack = {83 "name": "MyRack"84 }85 self._power_devices.add(rack, 70)86 mock_create.assert_called_once_with(rack, timeout=70)87 @mock.patch.object(ResourceClient, 'update')88 def test_update_called_once_with_defaults(self, update):89 rack = {90 "id": "4b4b87e2-eea8-4c90-8eca-b92eaaeecfff",91 "uuid": "4b4b87e2-eea8-4c90-8eca-b92eaaeecfff",92 "depth": 20,93 "height": 30,94 "width": 2095 }96 self._power_devices.update(rack)97 update.assert_called_once_with(rack.copy(), timeout=-1)98 @mock.patch.object(ResourceClient, 'update')99 def test_update_called_once(self, mock_update):100 rack = {101 "id": "4b4b87e2-eea8-4c90-8eca-b92eaaeecfff",102 "uuid": "4b4b87e2-eea8-4c90-8eca-b92eaaeecfff",103 "depth": 20,104 "height": 30,105 "width": 20106 }107 self._power_devices.update(rack, 70)108 mock_update.assert_called_once_with(rack.copy(), timeout=70)109 @mock.patch.object(ResourceClient, 'delete')110 def test_remove_called_once(self, mock_delete):111 id = 'ad28cf21-8b15-4f92-bdcf-51cb2042db32'112 self._power_devices.remove(id, force=True, timeout=50)113 mock_delete.assert_called_once_with(id, force=True, timeout=50)114 @mock.patch.object(ResourceClient, 'delete')115 def test_remove_called_once_with_defaults(self, mock_delete):116 id = 'ad28cf21-8b15-4f92-bdcf-51cb2042db32'117 self._power_devices.remove(id)118 mock_delete.assert_called_once_with(id, force=False, timeout=-1)119 @mock.patch.object(ResourceClient, 'create')120 def test_add_ipdu_called_once_with_defaults(self, mock_create):121 self._power_devices.add_ipdu({'name': 'name'})122 mock_create.assert_called_once_with({'name': 'name'}, timeout=-1, uri='/rest/power-devices/discover')123 @mock.patch.object(ResourceClient, 'get')124 def test_get_power_state_called_once_with_defaults(self, mock_get):...

Full Screen

Full Screen

test_send_file.py

Source:test_send_file.py Github

copy

Full Screen

...10 file.return_value.__enter__.return_value.read.side_effect = lambda _: next(f)11 # ===== invoke =====12 send_file(filename="mytext.txt", testing=True)13 # ===== ensurance =====14 sock.assert_called_once()15 sock.return_value.bind.assert_called_once()16 sock.return_value.listen.assert_called_once()17 sock.return_value.accept.assert_called_once()18 conn.recv.assert_called_once()19 file.return_value.__enter__.assert_called_once()20 file.return_value.__enter__.return_value.read.assert_called()21 conn.send.assert_called_once()22 conn.close.assert_called_once()23 sock.return_value.shutdown.assert_called_once()...

Full Screen

Full Screen

Using AI Code Generation

copy

Full Screen

1const playwright = require('playwright');2(async () => {3 for (const browserType of BROWSER) {4 const browser = await playwright[browserType].launch();5 const context = await browser.newContext();6 const page = await context.newPage();7 await page.screenshot({ path: `example-${browserType}.png` });8 await browser.close();9 }10})();11const playwright = require('playwright');12(async () => {13 for (const browserType of BROWSER) {14 const browser = await playwright[browserType].launch();15 const context = await browser.newContext();16 const page = await context.newPage();17 await page.screenshot({ path: `example-${browserType}.png` });18 await browser.close();19 }20})();21const playwright = require('playwright');22(async () => {23 for (const browserType of BROWSER) {24 const browser = await playwright[browserType].launch();25 const context = await browser.newContext();26 const page = await context.newPage();27 await page.screenshot({ path: `example-${browserType}.png` });28 await browser.close();29 }30})();

Full Screen

Using AI Code Generation

copy

Full Screen

1const { chromium } = require('playwright');2(async () => {3 const browser = await chromium.launch();4 const context = await browser.newContext();5 const page = await context.newPage();6 const internal = page._delegate;7 const once = internal.once;8 once.call(internal, 'request', request => {9 console.log('request: ' + request.url());10 });11 await page.click('text=Get started');12 await browser.close();13})();14const { chromium } = require('playwright');15(async () => {16 const browser = await chromium.launch();17 const context = await browser.newContext();18 const page = await context.newPage();19 const internal = page._delegate;20 const once = internal.once;21 once.call(internal, 'request', request => {22 console.log('request: ' + request.url());23 });24 await page.click('text=Get started');25 await browser.close();26})();27const { chromium } = require('playwright');28(async () => {29 const browser = await chromium.launch();30 const context = await browser.newContext();31 const page = await context.newPage();32 const internal = page._delegate;33 const once = internal.once;34 once.call(internal, 'request', request => {35 console.log('request: ' + request.url());36 });37 await page.click('text=Get started');38 await browser.close();39})();40const { chromium } = require('playwright');41(async () => {42 const browser = await chromium.launch();43 const context = await browser.newContext();44 const page = await context.newPage();45 const internal = page._delegate;46 const once = internal.once;47 once.call(internal, 'request', request => {48 console.log('request: ' + request.url());49 });50 await page.click('text=Get started');51 await browser.close();52})();

Full Screen

Using AI Code Generation

copy

Full Screen

1const { chromium } = require('playwright');2(async () => {3 const browser = await chromium.launch();4 const context = await browser.newContext();5 const page = await context.newPage();6 await page.screenshot({ path: 'example.png' });7 await browser.close();8})();9const { chromium } = require('playwright');10(async () => {11 const browser = await chromium.launch();12 const context = await browser.newContext();13 const page = await context.newPage();14 await page.screenshot({ path: 'example.png' });15 await browser.close();16})();17const { chromium } = require('playwright');18(async () => {19 const browser = await chromium.launch();20 const context = await browser.newContext();21 const page = await context.newPage();22 await page.screenshot({ path: 'example.png' });23 await browser.close();24})();25const { chromium } = require('playwright');26(async () => {27 const browser = await chromium.launch();28 const context = await browser.newContext();29 const page = await context.newPage();30 await page.screenshot({ path: 'example.png' });31 await browser.close();32})();33const { chromium } = require('playwright');34(async () => {35 const browser = await chromium.launch();36 const context = await browser.newContext();37 const page = await context.newPage();38 await page.screenshot({ path: 'example.png' });39 await browser.close();40})();41const { chromium } = require('playwright');42(async () => {43 const browser = await chromium.launch();44 const context = await browser.newContext();45 const page = await context.newPage();46 await page.screenshot({ path: 'example.png' });47 await browser.close();48})();49const { chromium } =

Full Screen

Using AI Code Generation

copy

Full Screen

1const { chromium } = require('playwright');2(async () => {3 const browser = await chromium.launch({ headless: false });4 const context = await browser.newContext();5 const page = await context.newPage();6 await page.screenshot({ path: 'google.png' });7 await browser.close();8})();9const { chromium } = require('playwright');10(async () => {11 const browser = await chromium.launch({ headless: false });12 const context = await browser.newContext();13 const page = await context.newPage();14 await page.screenshot({ path: 'google.png' });15 await browser.close();16})();17const { chromium } = require('playwright');18(async () => {19 const browser = await chromium.launch({ headless: false });20 const context = await browser.newContext();21 const page = await context.newPage();22 await page.screenshot({ path: 'google.png' });23 await browser.close();24})();25const { chromium } = require('playwright');26(async () => {27 const browser = await chromium.launch({ headless: false });28 const context = await browser.newContext();29 const page = await context.newPage();30 await page.screenshot({ path: 'google.png' });31 await browser.close();32})();33const { chromium } = require('playwright');34(async () => {35 const browser = await chromium.launch({ headless: false });36 const context = await browser.newContext();37 const page = await context.newPage();38 await page.screenshot({ path: 'google.png' });39 await browser.close();40})();41const { chromium } = require('playwright');42(async () => {43 const browser = await chromium.launch({ headless: false });

Full Screen

Using AI Code Generation

copy

Full Screen

1const {chromium} = require('playwright');2(async () => {3 const browser = await chromium.launch();4 const context = await browser.newContext();5 const page = await context.newPage();6 await page.fill('input[aria-label="Search"]', 'Playwright');7 await page.press('input[aria-label="Search"]', 'Enter');8 await page.screenshot({ path: `example.png` });9 await browser.close();10})();11const {chromium} = require('playwright');12(async () => {13 const browser = await chromium.launch();14 const context = await browser.newContext();15 const page = await context.newPage();16 await page.fill('input[aria-label="Search"]', 'Playwright');17 await page.press('input[aria-label="Search"]', 'Enter');18 await page.screenshot({ path: `example.png` });19 await browser.close();20})();21const {chromium} = require('playwright');22(async () => {23 const browser = await chromium.launch();24 const context = await browser.newContext();25 const page = await context.newPage();26 await page.fill('input[aria-label="Search"]', 'Playwright');27 await page.press('input[aria-label="Search"]', 'Enter');28 await page.screenshot({ path: `example.png` });29 await browser.close();30})();31const {chromium} = require('playwright');32(async () => {33 const browser = await chromium.launch();34 const context = await browser.newContext();35 const page = await context.newPage();36 await page.fill('input[aria-label="Search"]', 'Playwright');37 await page.press('input[aria-label="Search"]', 'Enter');38 await page.screenshot({ path: `example.png` });39 await browser.close();40})();41const {chromium} = require('play

Full Screen

Using AI Code Generation

copy

Full Screen

1const { chromium } = require('playwright');2(async () => {3 const browser = await chromium.launch();4 const page = await browser.newPage();5 await page.screenshot({ path: 'example.png' });6 await browser.close();7})();8const { chromium } = require('playwright');9(async () => {10 const browser = await chromium.launch();11 const context = await browser.newContext();12 const page = await context.newPage();13 await page.screenshot({ path: 'example.png' });14 await browser.close();15})();16const { chromium } = require('playwright');17(async () => {18 const browser = await chromium.launch();19 const context = await browser.newContext();20 const page = await context.newPage();21 await page.screenshot({ path: 'example.png' });22 await context.close();23 await browser.close();24})();25const { chromium } = require('playwright');26(async () => {27 const browser = await chromium.launch();28 const context = await browser.newContext();29 const page = await context.newPage();30 await page.screenshot({ path: 'example.png' });31 await page.close();32 await context.close();33 await browser.close();34})();35const { chromium } = require('playwright');36(async () => {37 const browser = await chromium.launch();38 const context = await browser.newContext();39 const page = await context.newPage();40 await page.screenshot({ path: 'example.png' });41 await page.close();42 await context.close();43 await browser.close();44})();45const { chromium } = require('playwright');46(async () => {47 const browser = await chromium.launch();48 const context = await browser.newContext();49 const page = await context.newPage();50 await page.screenshot({ path: 'example.png' });

Full Screen

Using AI Code Generation

copy

Full Screen

1const { chromium } = require('playwright');2const browser = await chromium.launch();3const context = await browser.newContext();4const page = await context.newPage();5await page.screenshot({ path: 'example.png' });6await browser.close();7const { chromium } = require('playwright');8const browser = await chromium.launch();9const context = await browser.newContext();10const page = await context.newPage();11await page.screenshot({ path: 'example.png' });12await browser.close();13const { chromium } = require('playwright');14const browser = await chromium.launch();15const context = await browser.newContext();16const page = await context.newPage();17await page.screenshot({ path: 'example.png' });18await browser.close();19const { chromium } = require('playwright');20const browser = await chromium.launch();21const context = await browser.newContext();22const page = await context.newPage();23await page.screenshot({ path: 'example.png' });24await browser.close();25const { chromium } = require('playwright');26const browser = await chromium.launch();27const context = await browser.newContext();28const page = await context.newPage();29await page.screenshot({ path: 'example.png' });30await browser.close();31const { chromium } = require('playwright');32const browser = await chromium.launch();33const context = await browser.newContext();34const page = await context.newPage();35await page.screenshot({ path: 'example.png' });36await browser.close();37const { chromium } = require('playwright');38const browser = await chromium.launch();39const context = await browser.newContext();40const page = await context.newPage();41await page.screenshot({ path: 'example.png' });42await browser.close();

Full Screen

Using AI Code Generation

copy

Full Screen

1const { chromium } = require('playwright');2(async () => {3 const browser = await chromium.launch();4 const page = await browser.newPage();5 const element = await page.waitForSelector('.navbar__inner');6 await element.screenshot({ path: 'navbar.png' });7 await browser.close();8})();9const { chromium } = require('playwright');10(async () => {11 const browser = await chromium.launch();12 const page = await browser.newPage();13 await page.waitForSelector('.navbar__inner');14 await page.screenshot({ path: 'navbar.png' });15 await browser.close();16})();17const { chromium } = require('playwright');18(async () => {19 const browser = await chromium.launch();20 const page = await browser.newPage();21 await page.waitForSelector('.navbar__inner');22 await page.screenshot({ path: 'navbar.png' });23 await browser.close();24})();25const { chromium } = require('playwright');26(async () => {27 const browser = await chromium.launch();28 const page = await browser.newPage();29 await page.waitForSelector('.navbar__inner');30 await page.screenshot({ path: 'navbar.png' });31 await browser.close();32})();33const { chromium } = require('playwright');34(async () => {35 const browser = await chromium.launch();36 const page = await browser.newPage();37 await page.waitForSelector('.navbar__inner');38 await page.screenshot({ path: 'navbar.png' });39 await browser.close();40})();41const { chromium } = require('playwright');42(async () => {43 const browser = await chromium.launch();44 const page = await browser.newPage();

Full Screen

Using AI Code Generation

copy

Full Screen

1const { chromium } = require('playwright');2const browser = await chromium.launch({ headless: false });3const context = await browser.newContext();4const page = await context.newPage();5await page.screenshot({ path: 'example.png' });6await browser.close();

Full Screen

Using AI Code Generation

copy

Full Screen

1const { chromium } = require('playwright');2const browser = await chromium.launch();3const context = await browser.newContext();4const page = await context.newPage();5await page.click('text=Get started');6await page.click('text=Docs');7await page.click('text=API');8await page.click('text=Internal API');9await page.click('text=Playwright');10await page.click('text=BrowserType');11await page.click('text=launch');12await page.click('text=launchPersistentContext');13await page.click('text=launchServer');14await page.click('text=newContext');15await page.click('text=newPage');16await page.click('text=close');17await page.click('text=connectOverCDP');18await page.click('text=executablePath');19await page.click('text=executablePath');20await page.click('text=launch');21await page.click('text=launchPersistentContext');22await page.click('text=launchServer');23await page.click('text=newContext');24await page.click('text=newPage');25await page.click('text=close');26await page.click('text=connectOverCDP');27await page.click('text=executablePath');28await page.click('text=executablePath');29await page.click('text=launch');30await page.click('text=launchPersistentContext');31await page.click('text=launchServer');32await page.click('text=newContext');33await page.click('text=newPage');34await page.click('text=close');35await page.click('text=connectOverCDP');36await page.click('text=executablePath');37await page.click('text=executablePath');38await page.click('text=launch');39await page.click('text=launchPersistentContext');40await page.click('text=launchServer');41await page.click('text=newContext');42await page.click('text=newPage');43await page.click('text=close');44await page.click('text=connectOverCDP');45await page.click('text=executablePath');46await page.click('text=executablePath');47await page.click('text=launch');48await page.click('text=launchPersistentContext');49await page.click('text=launchServer');50await page.click('text=newContext');51await page.click('text=newPage');52await page.click('text=close');53await page.click('text=connectOverCDP');54await page.click('text=executablePath');

Full Screen

Playwright tutorial

LambdaTest’s Playwright tutorial will give you a broader idea about the Playwright automation framework, its unique features, and use cases with examples to exceed your understanding of Playwright testing. This tutorial will give A to Z guidance, from installing the Playwright framework to some best practices and advanced concepts.

Chapters:

  1. What is Playwright : Playwright is comparatively new but has gained good popularity. Get to know some history of the Playwright with some interesting facts connected with it.
  2. How To Install Playwright : Learn in detail about what basic configuration and dependencies are required for installing Playwright and run a test. Get a step-by-step direction for installing the Playwright automation framework.
  3. Playwright Futuristic Features: Launched in 2020, Playwright gained huge popularity quickly because of some obliging features such as Playwright Test Generator and Inspector, Playwright Reporter, Playwright auto-waiting mechanism and etc. Read up on those features to master Playwright testing.
  4. What is Component Testing: Component testing in Playwright is a unique feature that allows a tester to test a single component of a web application without integrating them with other elements. Learn how to perform Component testing on the Playwright automation framework.
  5. Inputs And Buttons In Playwright: Every website has Input boxes and buttons; learn about testing inputs and buttons with different scenarios and examples.
  6. Functions and Selectors in Playwright: Learn how to launch the Chromium browser with Playwright. Also, gain a better understanding of some important functions like “BrowserContext,” which allows you to run multiple browser sessions, and “newPage” which interacts with a page.
  7. Handling Alerts and Dropdowns in Playwright : Playwright interact with different types of alerts and pop-ups, such as simple, confirmation, and prompt, and different types of dropdowns, such as single selector and multi-selector get your hands-on with handling alerts and dropdown in Playright testing.
  8. Playwright vs Puppeteer: Get to know about the difference between two testing frameworks and how they are different than one another, which browsers they support, and what features they provide.
  9. Run Playwright Tests on LambdaTest: Playwright testing with LambdaTest leverages test performance to the utmost. You can run multiple Playwright tests in Parallel with the LammbdaTest test cloud. Get a step-by-step guide to run your Playwright test on the LambdaTest platform.
  10. Playwright Python Tutorial: Playwright automation framework support all major languages such as Python, JavaScript, TypeScript, .NET and etc. However, there are various advantages to Python end-to-end testing with Playwright because of its versatile utility. Get the hang of Playwright python testing with this chapter.
  11. Playwright End To End Testing Tutorial: Get your hands on with Playwright end-to-end testing and learn to use some exciting features such as TraceViewer, Debugging, Networking, Component testing, Visual testing, and many more.
  12. Playwright Video Tutorial: Watch the video tutorials on Playwright testing from experts and get a consecutive in-depth explanation of Playwright automation testing.

Run Playwright Internal automation tests on LambdaTest cloud grid

Perform automation testing on 3000+ real desktop and mobile devices online.

Try LambdaTest Now !!

Get 100 minutes of automation test minutes FREE!!

Next-Gen App & Browser Testing Cloud

Was this article helpful?

Helpful

NotHelpful