def test_hls_stream(self): url = "http://test.se/stream.m3u8" master = "http://test.se/master.m3u8" stream = HLSStream(self.session, url, headers={"User-Agent": "Test"}) self.assertEqual( { "type": "hls", "url": url, "headers": { "User-Agent": "Test", "Accept": "*/*", "Accept-Encoding": DEFAULT_ACCEPT_ENCODING, "Connection": "keep-alive", } }, stream.__json__() ) stream = HLSStream(self.session, url, master, headers={"User-Agent": "Test"}) self.assertEqual( { "type": "hls", "url": url, "headers": { "User-Agent": "Test", "Accept": "*/*", "Accept-Encoding": DEFAULT_ACCEPT_ENCODING, "Connection": "keep-alive", }, "master": master }, stream.__json__() )
def test_hls_stream(session, common_args, expected_headers): stream = HLSStream(session, "http://host/stream.m3u8?foo=bar", **common_args) assert stream.__json__() == { "type": "hls", "url": "http://host/stream.m3u8?foo=bar&sessionqueryparamkey=sessionqueryparamval&queryparamkey=queryparamval", "headers": expected_headers, }