def test_overwrite_update_shard_count_on_error(self):
update_shard_count_header = {
'X-Amz-Target': 'Kinesis_20131202.UpdateShardCount'
}
request_data = '{"StreamName": "TestStream", "TargetShardCount": 2, "ScalingType": "UNIFORM_SCALING"}'
error_response = Response()
error_response.status_code = 400
response = UPDATE_KINESIS.return_response('POST', '/', request_data,
update_shard_count_header,
error_response)
self.assertEqual(response.status_code, 200)
resp_json = json.loads(to_str(response.content))
self.assertEqual(resp_json['StreamName'], 'TestStream')
self.assertEqual(resp_json['CurrentShardCount'], 1)
self.assertEqual(resp_json['TargetShardCount'], 2)
def test_overwrite_update_shard_count_on_error(self):
if config.KINESIS_PROVIDER == "kinesalite":
update_shard_count_header = {
"X-Amz-Target": "Kinesis_20131202.UpdateShardCount"
}
request_data = '{"StreamName": "TestStream", "TargetShardCount": 2, "ScalingType": "UNIFORM_SCALING"}'
error_response = Response()
error_response.status_code = 400
response = UPDATE_KINESIS.return_response(
"POST", "/", request_data, update_shard_count_header,
error_response)
self.assertEqual(200, response.status_code)
resp_json = json.loads(to_str(response.content))
self.assertEqual("TestStream", resp_json["StreamName"])
self.assertEqual(1, resp_json["CurrentShardCount"])
self.assertEqual(2, resp_json["TargetShardCount"])
else:
self.assertTrue(True)