Added support for ListenerAddress based on srt or zixi-pull protocols. (#4961)

This commit is contained in:
SundeepK 2022-03-22 20:11:22 +00:00 committed by GitHub
parent 26397e9c6d
commit 75ecc31ec4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 14 additions and 1 deletions

View File

@ -18,6 +18,9 @@ class MediaConnectResponse(BaseResponse):
entitlements = self._get_param("entitlements")
name = self._get_param("name")
outputs = self._get_param("outputs")
for index, output in enumerate(outputs):
if output.get("protocol") in ["srt-listener", "zixi-pull"]:
output["listenerAddress"] = f"{index}.0.0.0"
source = self._get_param("source")
source_failover_config = self._get_param("sourceFailoverConfig")
sources = self._get_param("sources")

View File

@ -23,7 +23,14 @@ def _create_flow_config(name, **kwargs):
}
],
)
outputs = kwargs.get("outputs", [{"Name": "Output 1", "Protocol": "zixi-push"}])
outputs = kwargs.get(
"outputs",
[
{"Name": "Output 1", "Protocol": "zixi-push"},
{"Name": "Output 2", "Protocol": "zixi-pull"},
{"Name": "Output 3", "Protocol": "srt-listener"},
],
)
source = kwargs.get(
"source",
{
@ -59,6 +66,9 @@ def test_create_flow_succeeds():
response["Flow"]["FlowArn"][:26].should.equal("arn:aws:mediaconnect:flow:")
response["Flow"]["Name"].should.equal("test Flow 1")
response["Flow"]["Status"].should.equal("STANDBY")
response["Flow"]["Outputs"][0].should.equal({"Name": "Output 1"})
response["Flow"]["Outputs"][1]["ListenerAddress"].should.equal("1.0.0.0")
response["Flow"]["Outputs"][2]["ListenerAddress"].should.equal("2.0.0.0")
response["Flow"]["Sources"][0][
"SourceArn"
] == "arn:aws:mediaconnect:source:Source A"