예제 #1
0
  def testGetSuggestedInstancesTwoDifferentSize(self, getRDSInstancesMock):
    region = "us-west-2"
    # Instance 1
    instanceMock1 = Mock(spec="boto.rds.dbinstance.DBInstance")
    instanceMock1.status = "available"
    instanceMock1.allocated_storage = 64.0
    instanceMock1.id = "testId1"
    # Instance 2
    instanceMock2 = Mock(spec="boto.rds.dbinstance.DBInstance")
    instanceMock2.status = "available"
    instanceMock2.allocated_storage = 65.0
    instanceMock2.id = "testId2"

    getRDSInstancesMock.return_value = [
        instanceMock1,
        instanceMock2,
    ]

    suggestions = rds_utils.getSuggestedInstances(region)
    self.assertIsInstance(suggestions, types.GeneratorType)
    suggestions = list(suggestions)

    self.assertSequenceEqual(suggestions, [
        {"id": "testId2", "name": "testId2", "namespace": "AWS/RDS",
         "region": region},
        {"id": "testId1", "name": "testId1", "namespace": "AWS/RDS",
         "region": region},
    ])
    getRDSInstancesMock.assert_call_once_with(region)
예제 #2
0
  def testGetSuggestedInstancesNone(self, getRDSInstancesMock):
    getRDSInstancesMock.return_value = []

    suggestions = rds_utils.getSuggestedInstances("dummy-region")
    self.assertIsInstance(suggestions, types.GeneratorType)
    suggestions = list(suggestions)

    self.assertSequenceEqual(suggestions, [])
    getRDSInstancesMock.assert_call_once_with("dummy-region")
예제 #3
0
    def testGetSuggestedInstancesNone(self, getRDSInstancesMock):
        getRDSInstancesMock.return_value = []

        suggestions = rds_utils.getSuggestedInstances("dummy-region")
        self.assertIsInstance(suggestions, types.GeneratorType)
        suggestions = list(suggestions)

        self.assertSequenceEqual(suggestions, [])
        getRDSInstancesMock.assert_call_once_with("dummy-region")
예제 #4
0
  def testGetSuggestedInstancesNoAvailable(self, getRDSInstancesMock):
    instanceMock1 = Mock(spec="boto.rds.dbinstance.DBInstance")
    instanceMock1.status = "not-available"
    getRDSInstancesMock.return_value = [
        instanceMock1,
    ]

    suggestions = rds_utils.getSuggestedInstances("dummy-region")
    self.assertIsInstance(suggestions, types.GeneratorType)
    suggestions = list(suggestions)

    self.assertSequenceEqual(suggestions, [])
    getRDSInstancesMock.assert_call_once_with("dummy-region")
예제 #5
0
    def testGetSuggestedInstancesNoAvailable(self, getRDSInstancesMock):
        instanceMock1 = Mock(spec="boto.rds.dbinstance.DBInstance")
        instanceMock1.status = "not-available"
        getRDSInstancesMock.return_value = [
            instanceMock1,
        ]

        suggestions = rds_utils.getSuggestedInstances("dummy-region")
        self.assertIsInstance(suggestions, types.GeneratorType)
        suggestions = list(suggestions)

        self.assertSequenceEqual(suggestions, [])
        getRDSInstancesMock.assert_call_once_with("dummy-region")
예제 #6
0
    def testGetSuggestedInstancesTwoDifferentSize(self, getRDSInstancesMock):
        region = "us-west-2"
        # Instance 1
        instanceMock1 = Mock(spec="boto.rds.dbinstance.DBInstance")
        instanceMock1.status = "available"
        instanceMock1.allocated_storage = 64.0
        instanceMock1.id = "testId1"
        # Instance 2
        instanceMock2 = Mock(spec="boto.rds.dbinstance.DBInstance")
        instanceMock2.status = "available"
        instanceMock2.allocated_storage = 65.0
        instanceMock2.id = "testId2"

        getRDSInstancesMock.return_value = [
            instanceMock1,
            instanceMock2,
        ]

        suggestions = rds_utils.getSuggestedInstances(region)
        self.assertIsInstance(suggestions, types.GeneratorType)
        suggestions = list(suggestions)

        self.assertSequenceEqual(suggestions, [
            {
                "id": "testId2",
                "name": "testId2",
                "namespace": "AWS/RDS",
                "region": region
            },
            {
                "id": "testId1",
                "name": "testId1",
                "namespace": "AWS/RDS",
                "region": region
            },
        ])
        getRDSInstancesMock.assert_call_once_with(region)