def setUp(self): self.s3_service = Mock() self.s3_session = Mock() self.ec2_session = Mock() self.bucket_name = Mock() self.reservation_id = "id" self.key_pair_serv = KeyPairService(self.s3_service)
def test_load_key_pair_by_name_not_found(self): s3_se = Mock() s3_se.get_key = Mock(return_value=None) key_pair_serv = KeyPairService(s3_se) key = \ key_pair_serv.load_key_pair_by_name(self.s3_session, self.bucket_name, self.reservation_id) self.assertIsNone(key)
def setUp(self): self.s3_service = Mock() self.s3_session = Mock() self.ec2_session = Mock() self.bucket_name = Mock() self.reservation_id = 'id' self.key_pair_serv = KeyPairService(self.s3_service)
def __init__(self): self.image_waiter = AMIWaiter() self.command_result_parser = CommandResultsParser() self.cancellation_service = CommandCancellationService() self.client_err_wrapper = ClientErrorWrapper() self.tag_service = TagService( client_err_wrapper=self.client_err_wrapper) self.ec2_instance_waiter = InstanceWaiter( cancellation_service=self.cancellation_service) self.instance_service = InstanceService(self.tag_service, self.ec2_instance_waiter) self.ec2_storage_service = EC2StorageService() self.model_parser = AWSModelsParser() self.cloudshell_session_helper = CloudshellDriverHelper() self.aws_session_manager = AWSSessionProvider() self.password_waiter = PasswordWaiter(self.cancellation_service) self.vm_custom_params_extractor = VmCustomParamsExtractor() self.ami_credentials_service = InstanceCredentialsService( self.password_waiter) self.security_group_service = SecurityGroupService(self.tag_service) self.subnet_waiter = SubnetWaiter() self.subnet_service = SubnetService(self.tag_service, self.subnet_waiter) self.s3_service = S3BucketService() self.vpc_peering_waiter = VpcPeeringConnectionWaiter() self.key_pair_service = KeyPairService(self.s3_service) self.vpc_waiter = VPCWaiter() self.route_tables_service = RouteTablesService(self.tag_service) self.network_interface_service = NetworkInterfaceService( subnet_service=self.subnet_service, security_group_service=self.security_group_service, ) self.elastic_ip_service = ElasticIpService() self.vm_details_provider = VmDetailsProvider() self.session_number_service = SessionNumberService() self.traffic_mirror_service = TrafficMirrorService() self.request_parser = DriverRequestParser() self.vpc_service = VPCService( tag_service=self.tag_service, subnet_service=self.subnet_service, instance_service=self.instance_service, vpc_waiter=self.vpc_waiter, vpc_peering_waiter=self.vpc_peering_waiter, sg_service=self.security_group_service, route_table_service=self.route_tables_service, traffic_mirror_service=self.traffic_mirror_service, ) self.prepare_connectivity_operation = PrepareSandboxInfraOperation( vpc_service=self.vpc_service, security_group_service=self.security_group_service, key_pair_service=self.key_pair_service, tag_service=self.tag_service, route_table_service=self.route_tables_service, cancellation_service=self.cancellation_service, subnet_service=self.subnet_service, subnet_waiter=self.subnet_waiter, ) self.deploy_ami_operation = DeployAMIOperation( instance_service=self.instance_service, ami_credential_service=self.ami_credentials_service, security_group_service=self.security_group_service, tag_service=self.tag_service, vpc_service=self.vpc_service, key_pair_service=self.key_pair_service, subnet_service=self.subnet_service, elastic_ip_service=self.elastic_ip_service, network_interface_service=self.network_interface_service, cancellation_service=self.cancellation_service, device_index_strategy=AllocateMissingValuesDeviceIndexStrategy(), vm_details_provider=self.vm_details_provider, ) self.refresh_ip_operation = RefreshIpOperation( instance_service=self.instance_service) self.power_management_operation = PowerOperation( instance_service=self.instance_service, instance_waiter=self.ec2_instance_waiter, ) self.delete_ami_operation = DeleteAMIOperation( instance_service=self.instance_service, ec2_storage_service=self.ec2_storage_service, security_group_service=self.security_group_service, tag_service=self.tag_service, elastic_ip_service=self.elastic_ip_service, ) self.clean_up_operation = CleanupSandboxInfraOperation( vpc_service=self.vpc_service, key_pair_service=self.key_pair_service, route_table_service=self.route_tables_service, traffic_mirror_service=self.traffic_mirror_service, ) self.deployed_app_ports_operation = DeployedAppPortsOperation( self.vm_custom_params_extractor, security_group_service=self.security_group_service, instance_service=self.instance_service, ) self.access_key_operation = GetAccessKeyOperation( key_pair_service=self.key_pair_service) self.set_app_security_groups_operation = SetAppSecurityGroupsOperation( instance_service=self.instance_service, tag_service=self.tag_service, security_group_service=self.security_group_service, ) self.vm_details_operation = VmDetailsOperation( instance_service=self.instance_service, vm_details_provider=self.vm_details_provider, ) self.autoload_operation = AutoloadOperation() self.snapshot_operation = SnapshotOperation(self.instance_service, self.image_waiter) self.traffic_mirroring_operation = TrafficMirrorOperation( tag_service=self.tag_service, session_number_service=self.session_number_service, traffic_mirror_service=self.traffic_mirror_service, cancellation_service=self.cancellation_service, )
class TestKeyPair(TestCase): def setUp(self): self.s3_service = Mock() self.s3_session = Mock() self.ec2_session = Mock() self.bucket_name = Mock() self.reservation_id = 'id' self.key_pair_serv = KeyPairService(self.s3_service) def test_create_key_pair(self): key = \ self.key_pair_serv.create_key_pair(self.ec2_session, self.s3_session, self.bucket_name, self.reservation_id) self.assertTrue(self.ec2_session.create_key_pair.called_with( self.key_pair_serv.get_reservation_key_name(self.reservation_id))) self.assertTrue( self.s3_service.put_key.called_with(self.s3_session, self.bucket_name, self.key_pair_serv._get_s3_key_location(self.reservation_id), self.ec2_session.create_key_pair())) self.assertEqual(key, self.ec2_session.create_key_pair()) def test_get_key_for_reservation(self): key = \ self.key_pair_serv.get_key_for_reservation(self.s3_session, self.bucket_name, self.reservation_id) self.assertTrue( self.s3_service.get_key.called_with(self.s3_session, self.bucket_name, self.key_pair_serv._get_s3_key_location(self.reservation_id))) self.assertEqual(key, self.key_pair_serv._get_s3_key_location(self.reservation_id)) def test_get_key_for_reservation_not_found(self): s3_se = Mock() s3_se.get_key = Mock(return_value=None) key_pair_serv = KeyPairService(s3_se) key = \ key_pair_serv.get_key_for_reservation(self.s3_session, self.bucket_name, self.reservation_id) self.assertIsNone(key) def test_load_key_pair_by_name(self): key = \ self.key_pair_serv.load_key_pair_by_name(self.s3_session, self.bucket_name, self.reservation_id) self.assertTrue(self.s3_service.get_key.called_with( self.s3_session, self.bucket_name, self.key_pair_serv._get_s3_key_location(self.reservation_id))) self.assertTrue(self.s3_service.get_body_of_object.called_with( self.s3_service.get_key)) self.assertEqual(key, self.s3_service.get_body_of_object()) def test_load_key_pair_by_name_not_found(self): s3_se = Mock() s3_se.get_key = Mock(return_value=None) key_pair_serv = KeyPairService(s3_se) key = \ key_pair_serv.load_key_pair_by_name(self.s3_session, self.bucket_name, self.reservation_id) self.assertIsNone(key)
class TestKeyPair(TestCase): def setUp(self): self.s3_service = Mock() self.s3_session = Mock() self.ec2_session = Mock() self.bucket_name = Mock() self.reservation_id = "id" self.key_pair_serv = KeyPairService(self.s3_service) def test_create_key_pair(self): key = self.key_pair_serv.create_key_pair(self.ec2_session, self.s3_session, self.bucket_name, self.reservation_id) self.assertTrue( self.ec2_session.create_key_pair.called_with( self.key_pair_serv.get_reservation_key_name( self.reservation_id))) self.assertTrue( self.s3_service.put_key.called_with( self.s3_session, self.bucket_name, self.key_pair_serv._get_s3_key_location(self.reservation_id), self.ec2_session.create_key_pair(), )) self.assertEqual(key, self.ec2_session.create_key_pair()) def test_get_key_for_reservation(self): key = self.key_pair_serv.get_key_for_reservation( self.s3_session, self.bucket_name, self.reservation_id) self.assertTrue( self.s3_service.get_key.called_with( self.s3_session, self.bucket_name, self.key_pair_serv._get_s3_key_location(self.reservation_id), )) self.assertEqual( key, self.key_pair_serv._get_s3_key_location(self.reservation_id)) def test_get_key_for_reservation_not_found(self): s3_se = Mock() s3_se.get_key = Mock(return_value=None) key_pair_serv = KeyPairService(s3_se) key = key_pair_serv.get_key_for_reservation(self.s3_session, self.bucket_name, self.reservation_id) self.assertIsNone(key) def test_load_key_pair_by_name(self): key = self.key_pair_serv.load_key_pair_by_name(self.s3_session, self.bucket_name, self.reservation_id) self.assertTrue( self.s3_service.get_key.called_with( self.s3_session, self.bucket_name, self.key_pair_serv._get_s3_key_location(self.reservation_id), )) self.assertTrue( self.s3_service.get_body_of_object.called_with( self.s3_service.get_key)) self.assertEqual(key, self.s3_service.get_body_of_object()) def test_load_key_pair_by_name_not_found(self): s3_se = Mock() s3_se.get_key = Mock(return_value=None) key_pair_serv = KeyPairService(s3_se) key = key_pair_serv.load_key_pair_by_name(self.s3_session, self.bucket_name, self.reservation_id) self.assertIsNone(key)