Add unit tests for vpc_security_groups when creating RDS database ins…

…tances.
boto · Aug 22, 2013 · 0a90627 · 0a90627
1 parent 7841230
commit 0a90627
Showing 1 changed file with 61 additions and 0 deletions.
diff --git a/tests/unit/rds/test_connection.py b/tests/unit/rds/test_connection.py
@@ -24,6 +24,7 @@
 from tests.unit import unittest
 from tests.unit import AWSMockServiceTestCase
 
+from boto.ec2.securitygroup import SecurityGroup
 from boto.rds import RDSConnection
 from boto.rds.parametergroup import ParameterGroup
 
@@ -387,6 +388,66 @@ def test_restore_dbinstance_from_point_in_time__db_subnet_group_name(self):
             'DBSubnetGroupName': 'dbsubnetgroup',
         }, ignore_params_values=['Version'])
 
+    def test_create_db_instance_vpc_sg_str(self):
+        self.set_http_response(status_code=200)
+        db = self.service_connection.create_dbinstance(
+            'SimCoProd01',
+            10,
+            'db.m1.large',
+            'master',
+            'Password01',
+            param_group='default.mysql5.1',
+            db_subnet_group_name='dbSubnetgroup01',
+            vpc_security_groups=['sg-1', 'sg-2'])
+
+        self.assert_request_parameters({
+            'Action': 'CreateDBInstance',
+            'AllocatedStorage': 10,
+            'AutoMinorVersionUpgrade': 'true',
+            'DBInstanceClass': 'db.m1.large',
+            'DBInstanceIdentifier': 'SimCoProd01',
+            'DBParameterGroupName': 'default.mysql5.1',
+            'DBSubnetGroupName': 'dbSubnetgroup01',
+            'Engine': 'MySQL5.1',
+            'MasterUsername': 'master',
+            'MasterUserPassword': 'Password01',
+            'Port': 3306,
+            'VpcSecurityGroupIds.member.1': 'sg-1',
+            'VpcSecurityGroupIds.member.2': 'sg-2'
+        }, ignore_params_values=['Version'])
+
+    def test_create_db_instance_vpc_sg_obj(self):
+        self.set_http_response(status_code=200)
+
+        sg1 = SecurityGroup(name='sg-1')
+        sg2 = SecurityGroup(name='sg-2')
+
+        db = self.service_connection.create_dbinstance(
+            'SimCoProd01',
+            10,
+            'db.m1.large',
+            'master',
+            'Password01',
+            param_group='default.mysql5.1',
+            db_subnet_group_name='dbSubnetgroup01',
+            vpc_security_groups=[sg1, sg2])
+
+        self.assert_request_parameters({
+            'Action': 'CreateDBInstance',
+            'AllocatedStorage': 10,
+            'AutoMinorVersionUpgrade': 'true',
+            'DBInstanceClass': 'db.m1.large',
+            'DBInstanceIdentifier': 'SimCoProd01',
+            'DBParameterGroupName': 'default.mysql5.1',
+            'DBSubnetGroupName': 'dbSubnetgroup01',
+            'Engine': 'MySQL5.1',
+            'MasterUsername': 'master',
+            'MasterUserPassword': 'Password01',
+            'Port': 3306,
+            'VpcSecurityGroupIds.member.1': 'sg-1',
+            'VpcSecurityGroupIds.member.2': 'sg-2'
+        }, ignore_params_values=['Version'])
+
 
 class TestRDSOptionGroups(AWSMockServiceTestCase):
     connection_class = RDSConnection