Yasuke/moto
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
moto/mediaconnect/responses.py
View File

@ -18,6 +18,9 @@ class MediaConnectResponse(BaseResponse):
entitlements = self._get_param("entitlements") entitlements = self._get_param("entitlements")
name = self._get_param("name") name = self._get_param("name")
outputs = self._get_param("outputs") 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 = self._get_param("source")
source_failover_config = self._get_param("sourceFailoverConfig") source_failover_config = self._get_param("sourceFailoverConfig")
sources = self._get_param("sources") sources = self._get_param("sources")

12
tests/test_mediaconnect/test_mediaconnect.py
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 = kwargs.get(
"source", "source",
{ {
@ -59,6 +66,9 @@ def test_create_flow_succeeds():
response["Flow"]["FlowArn"][:26].should.equal("arn:aws:mediaconnect:flow:") response["Flow"]["FlowArn"][:26].should.equal("arn:aws:mediaconnect:flow:")
response["Flow"]["Name"].should.equal("test Flow 1") response["Flow"]["Name"].should.equal("test Flow 1")
response["Flow"]["Status"].should.equal("STANDBY") 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][ response["Flow"]["Sources"][0][
"SourceArn" "SourceArn"
] == "arn:aws:mediaconnect:source:Source A" ] == "arn:aws:mediaconnect:source:Source A"