578 lines
		
	
	
		
			28 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			578 lines
		
	
	
		
			28 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
| from __future__ import unicode_literals
 | |
| 
 | |
| import boto.rds2
 | |
| import boto.vpc
 | |
| from boto.exception import BotoServerError
 | |
| import sure  # noqa
 | |
| from moto import mock_ec2, mock_rds2
 | |
| from tests.helpers import disable_on_py3
 | |
| 
 | |
| 
 | |
| @disable_on_py3()
 | |
| @mock_rds2
 | |
| def test_create_database():
 | |
|     conn = boto.rds2.connect_to_region("us-west-2")
 | |
|     database = conn.create_db_instance(db_instance_identifier='db-master-1',
 | |
|                                        allocated_storage=10,
 | |
|                                        engine='postgres',
 | |
|                                        db_instance_class='db.m1.small',
 | |
|                                        master_username='root',
 | |
|                                        master_user_password='hunter2',
 | |
|                                        db_security_groups=["my_sg"])
 | |
|     database['CreateDBInstanceResponse']['CreateDBInstanceResult']['DBInstance']['DBInstanceStatus'].should.equal('available')
 | |
|     database['CreateDBInstanceResponse']['CreateDBInstanceResult']['DBInstance']['DBInstanceIdentifier'].should.equal("db-master-1")
 | |
|     database['CreateDBInstanceResponse']['CreateDBInstanceResult']['DBInstance']['AllocatedStorage'].should.equal('10')
 | |
|     database['CreateDBInstanceResponse']['CreateDBInstanceResult']['DBInstance']['DBInstanceClass'].should.equal("db.m1.small")
 | |
|     database['CreateDBInstanceResponse']['CreateDBInstanceResult']['DBInstance']['MasterUsername'].should.equal("root")
 | |
|     database['CreateDBInstanceResponse']['CreateDBInstanceResult']['DBInstance']['DBSecurityGroups'][0]['DBSecurityGroup']['DBSecurityGroupName'].should.equal('my_sg')
 | |
| 
 | |
| 
 | |
| @disable_on_py3()
 | |
| @mock_rds2
 | |
| def test_get_databases():
 | |
|     conn = boto.rds2.connect_to_region("us-west-2")
 | |
| 
 | |
|     instances = conn.describe_db_instances()
 | |
|     list(instances['DescribeDBInstancesResponse']['DescribeDBInstancesResult']['DBInstances']).should.have.length_of(0)
 | |
| 
 | |
|     conn.create_db_instance(db_instance_identifier='db-master-1',
 | |
|                             allocated_storage=10,
 | |
|                             engine='postgres',
 | |
|                             db_instance_class='db.m1.small',
 | |
|                             master_username='root',
 | |
|                             master_user_password='hunter2',
 | |
|                             db_security_groups=["my_sg"])
 | |
|     conn.create_db_instance(db_instance_identifier='db-master-2',
 | |
|                             allocated_storage=10,
 | |
|                             engine='postgres',
 | |
|                             db_instance_class='db.m1.small',
 | |
|                             master_username='root',
 | |
|                             master_user_password='hunter2',
 | |
|                             db_security_groups=["my_sg"])
 | |
|     instances = conn.describe_db_instances()
 | |
|     list(instances['DescribeDBInstancesResponse']['DescribeDBInstancesResult']['DBInstances']).should.have.length_of(2)
 | |
| 
 | |
|     instances = conn.describe_db_instances("db-master-1")
 | |
|     list(instances['DescribeDBInstancesResponse']['DescribeDBInstancesResult']['DBInstances']).should.have.length_of(1)
 | |
|     instances['DescribeDBInstancesResponse']['DescribeDBInstancesResult']['DBInstances'][0]['DBInstanceIdentifier'].should.equal("db-master-1")
 | |
| 
 | |
| 
 | |
| @disable_on_py3()
 | |
| @mock_rds2
 | |
| def test_describe_non_existant_database():
 | |
|     conn = boto.rds2.connect_to_region("us-west-2")
 | |
|     conn.describe_db_instances.when.called_with("not-a-db").should.throw(BotoServerError)
 | |
| 
 | |
| 
 | |
| @disable_on_py3()
 | |
| @mock_rds2
 | |
| def test_modify_db_instance():
 | |
|     conn = boto.rds2.connect_to_region("us-west-2")
 | |
|     database = conn.create_db_instance(db_instance_identifier='db-master-1',
 | |
|                                        allocated_storage=10,
 | |
|                                        engine='postgres',
 | |
|                                        db_instance_class='db.m1.small',
 | |
|                                        master_username='root',
 | |
|                                        master_user_password='hunter2',
 | |
|                                        db_security_groups=["my_sg"])
 | |
|     instances = conn.describe_db_instances('db-master-1')
 | |
|     instances['DescribeDBInstancesResponse']['DescribeDBInstancesResult']['DBInstances'][0]['AllocatedStorage'].should.equal('10')
 | |
|     conn.modify_db_instance(db_instance_identifier='db-master-1', allocated_storage=20, apply_immediately=True)
 | |
|     instances = conn.describe_db_instances('db-master-1')
 | |
|     instances['DescribeDBInstancesResponse']['DescribeDBInstancesResult']['DBInstances'][0]['AllocatedStorage'].should.equal('20')
 | |
| 
 | |
| 
 | |
| @disable_on_py3()
 | |
| @mock_rds2
 | |
| def test_modify_non_existant_database():
 | |
|     conn = boto.rds2.connect_to_region("us-west-2")
 | |
|     conn.modify_db_instance.when.called_with(db_instance_identifier='not-a-db',
 | |
|                                              allocated_storage=20,
 | |
|                                              apply_immediately=True).should.throw(BotoServerError)
 | |
| 
 | |
| @disable_on_py3()
 | |
| @mock_rds2
 | |
| def test_reboot_db_instance():
 | |
|     conn = boto.rds2.connect_to_region("us-west-2")
 | |
|     conn.create_db_instance(db_instance_identifier='db-master-1',
 | |
|                             allocated_storage=10,
 | |
|                             engine='postgres',
 | |
|                             db_instance_class='db.m1.small',
 | |
|                             master_username='root',
 | |
|                             master_user_password='hunter2',
 | |
|                             db_security_groups=["my_sg"])
 | |
|     database = conn.reboot_db_instance('db-master-1')
 | |
|     database['RebootDBInstanceResponse']['RebootDBInstanceResult']['DBInstance']['DBInstanceIdentifier'].should.equal("db-master-1")
 | |
| 
 | |
| 
 | |
| @disable_on_py3()
 | |
| @mock_rds2
 | |
| def test_reboot_non_existant_database():
 | |
|     conn = boto.rds2.connect_to_region("us-west-2")
 | |
|     conn.reboot_db_instance.when.called_with("not-a-db").should.throw(BotoServerError)
 | |
| 
 | |
| 
 | |
| @disable_on_py3()
 | |
| @mock_rds2
 | |
| def test_delete_database():
 | |
|     conn = boto.rds2.connect_to_region("us-west-2")
 | |
|     instances = conn.describe_db_instances()
 | |
|     list(instances['DescribeDBInstancesResponse']['DescribeDBInstancesResult']['DBInstances']).should.have.length_of(0)
 | |
|     conn.create_db_instance(db_instance_identifier='db-master-1',
 | |
|                             allocated_storage=10,
 | |
|                             engine='postgres',
 | |
|                             db_instance_class='db.m1.small',
 | |
|                             master_username='root',
 | |
|                             master_user_password='hunter2',
 | |
|                             db_security_groups=["my_sg"])
 | |
|     instances = conn.describe_db_instances()
 | |
|     list(instances['DescribeDBInstancesResponse']['DescribeDBInstancesResult']['DBInstances']).should.have.length_of(1)
 | |
| 
 | |
|     conn.delete_db_instance("db-master-1")
 | |
|     instances = conn.describe_db_instances()
 | |
|     list(instances['DescribeDBInstancesResponse']['DescribeDBInstancesResult']['DBInstances']).should.have.length_of(0)
 | |
| 
 | |
| 
 | |
| @disable_on_py3()
 | |
| @mock_rds2
 | |
| def test_delete_non_existant_database():
 | |
|     conn = boto.rds2.connect_to_region("us-west-2")
 | |
|     conn.delete_db_instance.when.called_with("not-a-db").should.throw(BotoServerError)
 | |
| 
 | |
| 
 | |
| @disable_on_py3()
 | |
| @mock_rds2
 | |
| def test_create_option_group():
 | |
|     conn = boto.rds2.connect_to_region("us-west-2")
 | |
|     option_group = conn.create_option_group('test', 'mysql', '5.6', 'test option group')
 | |
|     option_group['CreateOptionGroupResponse']['CreateOptionGroupResult']['OptionGroup']['OptionGroupName'].should.equal('test')
 | |
|     option_group['CreateOptionGroupResponse']['CreateOptionGroupResult']['OptionGroup']['EngineName'].should.equal('mysql')
 | |
|     option_group['CreateOptionGroupResponse']['CreateOptionGroupResult']['OptionGroup']['OptionGroupDescription'].should.equal('test option group')
 | |
|     option_group['CreateOptionGroupResponse']['CreateOptionGroupResult']['OptionGroup']['MajorEngineVersion'].should.equal('5.6')
 | |
| 
 | |
| 
 | |
| @disable_on_py3()
 | |
| @mock_rds2
 | |
| def test_create_option_group_bad_engine_name():
 | |
|     conn = boto.rds2.connect_to_region("us-west-2")
 | |
|     conn.create_option_group.when.called_with('test', 'invalid_engine', '5.6', 'test invalid engine').should.throw(BotoServerError)
 | |
| 
 | |
| 
 | |
| @disable_on_py3()
 | |
| @mock_rds2
 | |
| def test_create_option_group_bad_engine_major_version():
 | |
|     conn = boto.rds2.connect_to_region("us-west-2")
 | |
|     conn.create_option_group.when.called_with('test', 'mysql', '6.6.6', 'test invalid engine version').should.throw(BotoServerError)
 | |
| 
 | |
| 
 | |
| @disable_on_py3()
 | |
| @mock_rds2
 | |
| def test_create_option_group_empty_description():
 | |
|     conn = boto.rds2.connect_to_region("us-west-2")
 | |
|     conn.create_option_group.when.called_with('test', 'mysql', '5.6', '').should.throw(BotoServerError)
 | |
| 
 | |
| 
 | |
| @disable_on_py3()
 | |
| @mock_rds2
 | |
| def test_create_option_group_duplicate():
 | |
|     conn = boto.rds2.connect_to_region("us-west-2")
 | |
|     conn.create_option_group('test', 'mysql', '5.6', 'test option group')
 | |
|     conn.create_option_group.when.called_with('test', 'mysql', '5.6', 'foo').should.throw(BotoServerError)
 | |
| 
 | |
| 
 | |
| @disable_on_py3()
 | |
| @mock_rds2
 | |
| def test_describe_option_group():
 | |
|     conn = boto.rds2.connect_to_region("us-west-2")
 | |
|     conn.create_option_group('test', 'mysql', '5.6', 'test option group')
 | |
|     option_groups = conn.describe_option_groups('test')
 | |
|     option_groups['DescribeOptionGroupsResponse']['DescribeOptionGroupsResult']['OptionGroupsList'][0]['OptionGroupName'].should.equal('test')
 | |
| 
 | |
| 
 | |
| @disable_on_py3()
 | |
| @mock_rds2
 | |
| def test_describe_non_existant_option_group():
 | |
|     conn = boto.rds2.connect_to_region("us-west-2")
 | |
|     conn.describe_option_groups.when.called_with("not-a-option-group").should.throw(BotoServerError)
 | |
| 
 | |
| 
 | |
| @disable_on_py3()
 | |
| @mock_rds2
 | |
| def test_delete_option_group():
 | |
|     conn = boto.rds2.connect_to_region("us-west-2")
 | |
|     conn.create_option_group('test', 'mysql', '5.6', 'test option group')
 | |
|     option_groups = conn.describe_option_groups('test')
 | |
|     option_groups['DescribeOptionGroupsResponse']['DescribeOptionGroupsResult']['OptionGroupsList'][0]['OptionGroupName'].should.equal('test')
 | |
|     conn.delete_option_group('test')
 | |
|     conn.describe_option_groups.when.called_with('test').should.throw(BotoServerError)
 | |
| 
 | |
| 
 | |
| @disable_on_py3()
 | |
| @mock_rds2
 | |
| def test_delete_non_existant_option_group():
 | |
|     conn = boto.rds2.connect_to_region("us-west-2")
 | |
|     conn.delete_option_group.when.called_with('non-existant').should.throw(BotoServerError)
 | |
| 
 | |
| 
 | |
| @disable_on_py3()
 | |
| @mock_rds2
 | |
| def test_describe_option_group_options():
 | |
|     conn = boto.rds2.connect_to_region("us-west-2")
 | |
|     option_group_options = conn.describe_option_group_options('sqlserver-ee')
 | |
|     len(option_group_options['DescribeOptionGroupOptionsResponse']['DescribeOptionGroupOptionsResult']['OptionGroupOptions']).should.equal(4)
 | |
|     option_group_options = conn.describe_option_group_options('sqlserver-ee', '11.00')
 | |
|     len(option_group_options['DescribeOptionGroupOptionsResponse']['DescribeOptionGroupOptionsResult']['OptionGroupOptions']).should.equal(2)
 | |
|     option_group_options = conn.describe_option_group_options('mysql', '5.6')
 | |
|     len(option_group_options['DescribeOptionGroupOptionsResponse']['DescribeOptionGroupOptionsResult']['OptionGroupOptions']).should.equal(1)
 | |
|     conn.describe_option_group_options.when.called_with('non-existent').should.throw(BotoServerError)
 | |
|     conn.describe_option_group_options.when.called_with('mysql', 'non-existent').should.throw(BotoServerError)
 | |
| 
 | |
| 
 | |
| @disable_on_py3()
 | |
| @mock_rds2
 | |
| def test_modify_option_group():
 | |
|     conn = boto.rds2.connect_to_region("us-west-2")
 | |
|     conn.create_option_group('test', 'mysql', '5.6', 'test option group')
 | |
|     # TODO: create option and validate before deleting.
 | |
|     # if Someone can tell me how the hell to use this function
 | |
|     # to add options to an option_group, I can finish coding this.
 | |
|     result = conn.modify_option_group('test', [], ['MEMCACHED'], True)
 | |
|     result['ModifyOptionGroupResponse']['ModifyOptionGroupResult']['EngineName'].should.equal('mysql')
 | |
|     result['ModifyOptionGroupResponse']['ModifyOptionGroupResult']['Options'].should.equal([])
 | |
|     result['ModifyOptionGroupResponse']['ModifyOptionGroupResult']['OptionGroupName'].should.equal('test')
 | |
| 
 | |
| 
 | |
| @disable_on_py3()
 | |
| @mock_rds2
 | |
| def test_modify_option_group_no_options():
 | |
|     conn = boto.rds2.connect_to_region("us-west-2")
 | |
|     conn.create_option_group('test', 'mysql', '5.6', 'test option group')
 | |
|     conn.modify_option_group.when.called_with('test').should.throw(BotoServerError)
 | |
| 
 | |
| 
 | |
| @disable_on_py3()
 | |
| @mock_rds2
 | |
| def test_modify_non_existant_option_group():
 | |
|     conn = boto.rds2.connect_to_region("us-west-2")
 | |
|     conn.modify_option_group.when.called_with('non-existant', [('OptionName', 'Port', 'DBSecurityGroupMemberships', 'VpcSecurityGroupMemberships', 'OptionSettings')]).should.throw(BotoServerError)
 | |
| 
 | |
| 
 | |
| @disable_on_py3()
 | |
| @mock_rds2
 | |
| def test_delete_non_existant_database():
 | |
|     conn = boto.rds2.connect_to_region("us-west-2")
 | |
|     conn.delete_db_instance.when.called_with("not-a-db").should.throw(BotoServerError)
 | |
| 
 | |
| 
 | |
| @disable_on_py3()
 | |
| @mock_rds2
 | |
| def test_list_tags_invalid_arn():
 | |
|     conn = boto.rds2.connect_to_region("us-west-2")
 | |
|     conn.list_tags_for_resource.when.called_with('arn:aws:rds:bad-arn').should.throw(BotoServerError)
 | |
| 
 | |
| 
 | |
| @disable_on_py3()
 | |
| @mock_rds2
 | |
| def test_list_tags_db():
 | |
|     conn = boto.rds2.connect_to_region("us-west-2")
 | |
|     result = conn.list_tags_for_resource('arn:aws:rds:us-west-2:1234567890:db:foo')
 | |
|     result['ListTagsForResourceResponse']['ListTagsForResourceResult']['TagList'].should.equal([])
 | |
|     conn.create_db_instance(db_instance_identifier='db-with-tags',
 | |
|                             allocated_storage=10,
 | |
|                             engine='postgres',
 | |
|                             db_instance_class='db.m1.small',
 | |
|                             master_username='root',
 | |
|                             master_user_password='hunter2',
 | |
|                             db_security_groups=["my_sg"],
 | |
|                             tags=[('foo', 'bar'), ('foo1', 'bar1')])
 | |
|     result = conn.list_tags_for_resource('arn:aws:rds:us-west-2:1234567890:db:db-with-tags')
 | |
|     result['ListTagsForResourceResponse']['ListTagsForResourceResult']['TagList'].should.equal([{'Value': 'bar',
 | |
|                                                                                                  'Key': 'foo'},
 | |
|                                                                                                 {'Value': 'bar1',
 | |
|                                                                                                  'Key': 'foo1'}])
 | |
| 
 | |
| 
 | |
| @disable_on_py3()
 | |
| @mock_rds2
 | |
| def test_add_tags_db():
 | |
|     conn = boto.rds2.connect_to_region("us-west-2")
 | |
|     conn.create_db_instance(db_instance_identifier='db-without-tags',
 | |
|                             allocated_storage=10,
 | |
|                             engine='postgres',
 | |
|                             db_instance_class='db.m1.small',
 | |
|                             master_username='root',
 | |
|                             master_user_password='hunter2',
 | |
|                             db_security_groups=["my_sg"],
 | |
|                             tags=[('foo', 'bar'), ('foo1', 'bar1')])
 | |
|     result = conn.list_tags_for_resource('arn:aws:rds:us-west-2:1234567890:db:db-without-tags')
 | |
|     list(result['ListTagsForResourceResponse']['ListTagsForResourceResult']['TagList']).should.have.length_of(2)
 | |
|     conn.add_tags_to_resource('arn:aws:rds:us-west-2:1234567890:db:db-without-tags',
 | |
|                               [('foo', 'fish'), ('foo2', 'bar2')])
 | |
|     result = conn.list_tags_for_resource('arn:aws:rds:us-west-2:1234567890:db:db-without-tags')
 | |
|     list(result['ListTagsForResourceResponse']['ListTagsForResourceResult']['TagList']).should.have.length_of(3)
 | |
| 
 | |
| 
 | |
| @disable_on_py3()
 | |
| @mock_rds2
 | |
| def test_remove_tags_db():
 | |
|     conn = boto.rds2.connect_to_region("us-west-2")
 | |
|     conn.create_db_instance(db_instance_identifier='db-with-tags',
 | |
|                             allocated_storage=10,
 | |
|                             engine='postgres',
 | |
|                             db_instance_class='db.m1.small',
 | |
|                             master_username='root',
 | |
|                             master_user_password='hunter2',
 | |
|                             db_security_groups=["my_sg"],
 | |
|                             tags=[('foo', 'bar'), ('foo1', 'bar1')])
 | |
|     result = conn.list_tags_for_resource('arn:aws:rds:us-west-2:1234567890:db:db-with-tags')
 | |
|     len(result['ListTagsForResourceResponse']['ListTagsForResourceResult']['TagList']).should.equal(2)
 | |
|     conn.remove_tags_from_resource('arn:aws:rds:us-west-2:1234567890:db:db-with-tags', ['foo'])
 | |
|     result = conn.list_tags_for_resource('arn:aws:rds:us-west-2:1234567890:db:db-with-tags')
 | |
|     len(result['ListTagsForResourceResponse']['ListTagsForResourceResult']['TagList']).should.equal(1)
 | |
| 
 | |
| 
 | |
| @disable_on_py3()
 | |
| @mock_rds2
 | |
| def test_add_tags_option_group():
 | |
|     conn = boto.rds2.connect_to_region("us-west-2")
 | |
|     conn.create_option_group('test', 'mysql', '5.6', 'test option group')
 | |
|     result = conn.list_tags_for_resource('arn:aws:rds:us-west-2:1234567890:og:test')
 | |
|     list(result['ListTagsForResourceResponse']['ListTagsForResourceResult']['TagList']).should.have.length_of(0)
 | |
|     conn.add_tags_to_resource('arn:aws:rds:us-west-2:1234567890:og:test',
 | |
|                                        [('foo', 'fish'), ('foo2', 'bar2')])
 | |
|     result = conn.list_tags_for_resource('arn:aws:rds:us-west-2:1234567890:og:test')
 | |
|     list(result['ListTagsForResourceResponse']['ListTagsForResourceResult']['TagList']).should.have.length_of(2)
 | |
| 
 | |
| 
 | |
| @disable_on_py3()
 | |
| @mock_rds2
 | |
| def test_remove_tags_option_group():
 | |
|     conn = boto.rds2.connect_to_region("us-west-2")
 | |
|     conn.create_option_group('test', 'mysql', '5.6', 'test option group')
 | |
|     conn.add_tags_to_resource('arn:aws:rds:us-west-2:1234567890:og:test',
 | |
|                                        [('foo', 'fish'), ('foo2', 'bar2')])
 | |
|     result = conn.list_tags_for_resource('arn:aws:rds:us-west-2:1234567890:og:test')
 | |
|     list(result['ListTagsForResourceResponse']['ListTagsForResourceResult']['TagList']).should.have.length_of(2)
 | |
|     conn.remove_tags_from_resource('arn:aws:rds:us-west-2:1234567890:og:test',
 | |
|                                  ['foo'])
 | |
|     result = conn.list_tags_for_resource('arn:aws:rds:us-west-2:1234567890:og:test')
 | |
|     list(result['ListTagsForResourceResponse']['ListTagsForResourceResult']['TagList']).should.have.length_of(1)
 | |
| 
 | |
| 
 | |
| @disable_on_py3()
 | |
| @mock_rds2
 | |
| def test_create_database_security_group():
 | |
|     conn = boto.rds2.connect_to_region("us-west-2")
 | |
| 
 | |
|     result = conn.create_db_security_group('db_sg', 'DB Security Group')
 | |
|     result['CreateDBSecurityGroupResponse']['CreateDBSecurityGroupResult']['DBSecurityGroup']['DBSecurityGroupName'].should.equal("db_sg")
 | |
|     result['CreateDBSecurityGroupResponse']['CreateDBSecurityGroupResult']['DBSecurityGroup']['DBSecurityGroupDescription'].should.equal("DB Security Group")
 | |
|     result['CreateDBSecurityGroupResponse']['CreateDBSecurityGroupResult']['DBSecurityGroup']['IPRanges'].should.equal([])
 | |
| 
 | |
| 
 | |
| @disable_on_py3()
 | |
| @mock_rds2
 | |
| def test_get_security_groups():
 | |
|     conn = boto.rds2.connect_to_region("us-west-2")
 | |
| 
 | |
|     result = conn.describe_db_security_groups()
 | |
|     result['DescribeDBSecurityGroupsResponse']['DescribeDBSecurityGroupsResult']['DBSecurityGroups'].should.have.length_of(0)
 | |
| 
 | |
|     conn.create_db_security_group('db_sg1', 'DB Security Group')
 | |
|     conn.create_db_security_group('db_sg2', 'DB Security Group')
 | |
| 
 | |
|     result = conn.describe_db_security_groups()
 | |
|     result['DescribeDBSecurityGroupsResponse']['DescribeDBSecurityGroupsResult']['DBSecurityGroups'].should.have.length_of(2)
 | |
| 
 | |
|     result = conn.describe_db_security_groups("db_sg1")
 | |
|     result['DescribeDBSecurityGroupsResponse']['DescribeDBSecurityGroupsResult']['DBSecurityGroups'].should.have.length_of(1)
 | |
|     result['DescribeDBSecurityGroupsResponse']['DescribeDBSecurityGroupsResult']['DBSecurityGroups'][0]['DBSecurityGroupName'].should.equal("db_sg1")
 | |
| 
 | |
| 
 | |
| @disable_on_py3()
 | |
| @mock_rds2
 | |
| def test_get_non_existant_security_group():
 | |
|     conn = boto.rds2.connect_to_region("us-west-2")
 | |
|     conn.describe_db_security_groups.when.called_with("not-a-sg").should.throw(BotoServerError)
 | |
| 
 | |
| 
 | |
| @disable_on_py3()
 | |
| @mock_rds2
 | |
| def test_delete_database_security_group():
 | |
|     conn = boto.rds2.connect_to_region("us-west-2")
 | |
|     conn.create_db_security_group('db_sg', 'DB Security Group')
 | |
| 
 | |
|     result = conn.describe_db_security_groups()
 | |
|     result['DescribeDBSecurityGroupsResponse']['DescribeDBSecurityGroupsResult']['DBSecurityGroups'].should.have.length_of(1)
 | |
| 
 | |
|     conn.delete_db_security_group("db_sg")
 | |
|     result = conn.describe_db_security_groups()
 | |
|     result['DescribeDBSecurityGroupsResponse']['DescribeDBSecurityGroupsResult']['DBSecurityGroups'].should.have.length_of(0)
 | |
| 
 | |
| 
 | |
| @disable_on_py3()
 | |
| @mock_rds2
 | |
| def test_delete_non_existant_security_group():
 | |
|     conn = boto.rds2.connect_to_region("us-west-2")
 | |
|     conn.delete_db_security_group.when.called_with("not-a-db").should.throw(BotoServerError)
 | |
| 
 | |
| 
 | |
| @disable_on_py3()
 | |
| @mock_rds2
 | |
| def test_security_group_authorize():
 | |
|     conn = boto.rds2.connect_to_region("us-west-2")
 | |
|     security_group = conn.create_db_security_group('db_sg', 'DB Security Group')
 | |
|     security_group['CreateDBSecurityGroupResponse']['CreateDBSecurityGroupResult']['DBSecurityGroup']['IPRanges'].should.equal([])
 | |
| 
 | |
| 
 | |
|     conn.authorize_db_security_group_ingress(db_security_group_name='db_sg',
 | |
|                                              cidrip='10.3.2.45/32')
 | |
| 
 | |
|     result = conn.describe_db_security_groups("db_sg")
 | |
|     result['DescribeDBSecurityGroupsResponse']['DescribeDBSecurityGroupsResult']['DBSecurityGroups'][0]['IPRanges'].should.have.length_of(1)
 | |
|     result['DescribeDBSecurityGroupsResponse']['DescribeDBSecurityGroupsResult']['DBSecurityGroups'][0]['IPRanges'].should.equal(['10.3.2.45/32'])
 | |
| 
 | |
|     conn.authorize_db_security_group_ingress(db_security_group_name='db_sg',
 | |
|                                              cidrip='10.3.2.46/32')
 | |
|     result = conn.describe_db_security_groups("db_sg")
 | |
|     result['DescribeDBSecurityGroupsResponse']['DescribeDBSecurityGroupsResult']['DBSecurityGroups'][0]['IPRanges'].should.have.length_of(2)
 | |
|     result['DescribeDBSecurityGroupsResponse']['DescribeDBSecurityGroupsResult']['DBSecurityGroups'][0]['IPRanges'].should.equal(['10.3.2.45/32', '10.3.2.46/32'])
 | |
| 
 | |
| 
 | |
| @disable_on_py3()
 | |
| @mock_rds2
 | |
| def test_add_security_group_to_database():
 | |
|     conn = boto.rds2.connect_to_region("us-west-2")
 | |
| 
 | |
|     conn.create_db_instance(db_instance_identifier='db-master-1',
 | |
|                             allocated_storage=10,
 | |
|                             engine='postgres',
 | |
|                             db_instance_class='db.m1.small',
 | |
|                             master_username='root',
 | |
|                             master_user_password='hunter2')
 | |
|     result = conn.describe_db_instances()
 | |
|     result['DescribeDBInstancesResponse']['DescribeDBInstancesResult']['DBInstances'][0]['DBSecurityGroups'].should.equal([])
 | |
|     conn.create_db_security_group('db_sg', 'DB Security Group')
 | |
|     conn.modify_db_instance(db_instance_identifier='db-master-1',
 | |
|                             db_security_groups=['db_sg'])
 | |
|     result = conn.describe_db_instances()
 | |
|     result['DescribeDBInstancesResponse']['DescribeDBInstancesResult']['DBInstances'][0]['DBSecurityGroups'][0]['DBSecurityGroup']['DBSecurityGroupName'].should.equal('db_sg')
 | |
| 
 | |
| 
 | |
| @disable_on_py3()
 | |
| @mock_ec2
 | |
| @mock_rds2
 | |
| def test_create_database_subnet_group():
 | |
|     vpc_conn = boto.vpc.connect_to_region("us-west-2")
 | |
|     vpc = vpc_conn.create_vpc("10.0.0.0/16")
 | |
|     subnet1 = vpc_conn.create_subnet(vpc.id, "10.1.0.0/24")
 | |
|     subnet2 = vpc_conn.create_subnet(vpc.id, "10.2.0.0/24")
 | |
| 
 | |
|     subnet_ids = [subnet1.id, subnet2.id]
 | |
|     conn = boto.rds2.connect_to_region("us-west-2")
 | |
|     result = conn.create_db_subnet_group("db_subnet", "my db subnet", subnet_ids)
 | |
|     result['CreateDBSubnetGroupResponse']['CreateDBSubnetGroupResult']['DBSubnetGroup']['DBSubnetGroupName'].should.equal("db_subnet")
 | |
|     result['CreateDBSubnetGroupResponse']['CreateDBSubnetGroupResult']['DBSubnetGroup']['DBSubnetGroupDescription'].should.equal("my db subnet")
 | |
|     subnets = result['CreateDBSubnetGroupResponse']['CreateDBSubnetGroupResult']['DBSubnetGroup']['Subnets']
 | |
|     subnet_group_ids = [subnets['Subnet'][0]['SubnetIdentifier'], subnets['Subnet'][1]['SubnetIdentifier']]
 | |
|     list(subnet_group_ids).should.equal(subnet_ids)
 | |
| 
 | |
| 
 | |
| @disable_on_py3()
 | |
| @mock_ec2
 | |
| @mock_rds2
 | |
| def test_create_database_in_subnet_group():
 | |
|     vpc_conn = boto.vpc.connect_to_region("us-west-2")
 | |
|     vpc = vpc_conn.create_vpc("10.0.0.0/16")
 | |
|     subnet = vpc_conn.create_subnet(vpc.id, "10.1.0.0/24")
 | |
| 
 | |
|     conn = boto.rds2.connect_to_region("us-west-2")
 | |
|     conn.create_db_subnet_group("db_subnet1", "my db subnet", [subnet.id])
 | |
|     conn.create_db_instance(db_instance_identifier='db-master-1',
 | |
|                             allocated_storage=10,
 | |
|                             engine='postgres',
 | |
|                             db_instance_class='db.m1.small',
 | |
|                             master_username='root',
 | |
|                             master_user_password='hunter2',
 | |
|                             db_subnet_group_name='db_subnet1')
 | |
|     result = conn.describe_db_instances("db-master-1")
 | |
|     result['DescribeDBInstancesResponse']['DescribeDBInstancesResult']['DBInstances'][0]['DBSubnetGroup']['DBSubnetGroupName'].should.equal("db_subnet1")
 | |
| 
 | |
| 
 | |
| @disable_on_py3()
 | |
| @mock_ec2
 | |
| @mock_rds2
 | |
| def test_describe_database_subnet_group():
 | |
|    vpc_conn = boto.vpc.connect_to_region("us-west-2")
 | |
|    vpc = vpc_conn.create_vpc("10.0.0.0/16")
 | |
|    subnet = vpc_conn.create_subnet(vpc.id, "10.1.0.0/24")
 | |
| 
 | |
|    conn = boto.rds2.connect_to_region("us-west-2")
 | |
|    conn.create_db_subnet_group("db_subnet1", "my db subnet", [subnet.id])
 | |
|    conn.create_db_subnet_group("db_subnet2", "my db subnet", [subnet.id])
 | |
| 
 | |
|    resp = conn.describe_db_subnet_groups()
 | |
|    groups_resp = resp['DescribeDBSubnetGroupsResponse']
 | |
| 
 | |
|    subnet_groups = groups_resp['DescribeDBSubnetGroupsResult']['DBSubnetGroups']
 | |
|    subnet_groups.should.have.length_of(2)
 | |
| 
 | |
|    subnets = groups_resp['DescribeDBSubnetGroupsResult']['DBSubnetGroups'][0]['DBSubnetGroup']['Subnets']
 | |
|    subnets.should.have.length_of(1)
 | |
| 
 | |
|    list(resp).should.have.length_of(1)
 | |
|    list(groups_resp).should.have.length_of(2)
 | |
|    list(conn.describe_db_subnet_groups("db_subnet1")).should.have.length_of(1)
 | |
| 
 | |
|    conn.describe_db_subnet_groups.when.called_with("not-a-subnet").should.throw(BotoServerError)
 | |
| 
 | |
| 
 | |
| @disable_on_py3()
 | |
| @mock_ec2
 | |
| @mock_rds2
 | |
| def test_delete_database_subnet_group():
 | |
|     vpc_conn = boto.vpc.connect_to_region("us-west-2")
 | |
|     vpc = vpc_conn.create_vpc("10.0.0.0/16")
 | |
|     subnet = vpc_conn.create_subnet(vpc.id, "10.1.0.0/24")
 | |
| 
 | |
|     conn = boto.rds2.connect_to_region("us-west-2")
 | |
|     result = conn.describe_db_subnet_groups()
 | |
|     result['DescribeDBSubnetGroupsResponse']['DescribeDBSubnetGroupsResult']['DBSubnetGroups'].should.have.length_of(0)
 | |
| 
 | |
|     conn.create_db_subnet_group("db_subnet1", "my db subnet", [subnet.id])
 | |
|     result = conn.describe_db_subnet_groups()
 | |
|     result['DescribeDBSubnetGroupsResponse']['DescribeDBSubnetGroupsResult']['DBSubnetGroups'].should.have.length_of(1)
 | |
| 
 | |
|     conn.delete_db_subnet_group("db_subnet1")
 | |
|     result = conn.describe_db_subnet_groups()
 | |
|     result['DescribeDBSubnetGroupsResponse']['DescribeDBSubnetGroupsResult']['DBSubnetGroups'].should.have.length_of(0)
 | |
| 
 | |
|     conn.delete_db_subnet_group.when.called_with("db_subnet1").should.throw(BotoServerError)
 | |
| 
 | |
| 
 | |
| @disable_on_py3()
 | |
| @mock_rds2
 | |
| def test_create_database_replica():
 | |
|     conn = boto.rds2.connect_to_region("us-west-2")
 | |
| 
 | |
|     conn.create_db_instance(db_instance_identifier='db-master-1',
 | |
|                             allocated_storage=10,
 | |
|                             engine='postgres',
 | |
|                             db_instance_class='db.m1.small',
 | |
|                             master_username='root',
 | |
|                             master_user_password='hunter2',
 | |
|                             db_security_groups=["my_sg"])
 | |
| 
 | |
|     replica = conn.create_db_instance_read_replica("db-replica-1", "db-master-1", "db.m1.small")
 | |
|     replica['CreateDBInstanceReadReplicaResponse']['CreateDBInstanceReadReplicaResult']['DBInstance']['ReadReplicaSourceDBInstanceIdentifier'].should.equal('db-master-1')
 | |
|     replica['CreateDBInstanceReadReplicaResponse']['CreateDBInstanceReadReplicaResult']['DBInstance']['DBInstanceClass'].should.equal('db.m1.small')
 | |
|     replica['CreateDBInstanceReadReplicaResponse']['CreateDBInstanceReadReplicaResult']['DBInstance']['DBInstanceIdentifier'].should.equal('db-replica-1')
 | |
| 
 | |
|     master = conn.describe_db_instances("db-master-1")
 | |
|     master['DescribeDBInstancesResponse']['DescribeDBInstancesResult']['DBInstances'][0]['ReadReplicaDBInstanceIdentifiers'].should.equal(['db-replica-1'])
 | |
| 
 | |
|     conn.delete_db_instance("db-replica-1")
 | |
| 
 | |
|     master = conn.describe_db_instances("db-master-1")
 | |
|     master['DescribeDBInstancesResponse']['DescribeDBInstancesResult']['DBInstances'][0]['ReadReplicaDBInstanceIdentifiers'].should.equal([])
 |