def test_success(self): """ The top one will have to spill over into the second network """ expected = { "status": "SUCCESS", "reason": None, "data": { "CidrBlock1": "10.0.1.128/25", "CidrBlock2": "10.0.128.0/17", "CidrBlock3": "10.1.128.0/17", }, } request = { "RequestType": "Create", "ResourceProperties": { "VpcId": "", "Sizes": ["25", "17", "17"], }, } handler(request, {}, responder=self.__responder, client=MockEc2()) self.assertEqual(self.response, expected)
def test_delete(self): expected = { "status": "SUCCESS", "reason": None, "data": {}, } handler({"RequestType": "Delete"}, {}, responder=self.__responder) self.assertEqual(self.response, expected)
def test_missing_both(self): expected = { "status": "FAILED", "reason": "Missing parameter(s): VpcId, Sizes", "data": {}, } request = { "RequestType": "Create", "ResourceProperties": {}, } handler(request, {}, responder=self.__responder) self.assertEqual(self.response, expected)
def test_request_too_large(self): expected = { "status": "FAILED", "reason": "Not enough space for the requested CIDR blocks", "data": {}, } request = { "RequestType": "Create", "ResourceProperties": { "VpcId": "", "Sizes": ["16"], }, } handler(request, {}, responder=self.__responder, client=MockEc2()) self.assertEqual(self.response, expected)
def test_bad_sizes(self): expected = { "status": "FAILED", "reason": "An invalid subnet size was specified: 23, camel, 25", "data": {}, } request = { "RequestType": "Create", "ResourceProperties": { "VpcId": "", "Sizes": ["23", "camel", "25"], }, } handler(request, {}, responder=self.__responder) self.assertEqual(self.response, expected)