def test_port_multiplex(self):
self.config.multiplex = True
main = Main(self.config)
port = main._port(0)
self.assertEqual(port, None)
port = main._port(1)
self.assertEqual(port, None)
def test_login_unsuccessfull(self, sys: MagicMock):
with patch('locast2dvr.main.LocastService') as service:
main = Main(self.config)
service.login.side_effect = Exception("oops!")
main._login()
service.login.assert_called_once_with('foo', 'secret')
sys.exit.assert_called_once_with(1)
def test_startup_order(self, ssdp_server: MagicMock):
with patch.multiple(
'locast2dvr.main.Main',
_login=MagicMock(),
_init_geos=MagicMock(),
_init_multiplexer=MagicMock(),
_init_tuners=MagicMock(),
_check_ffmpeg=MagicMock(),
_report=MagicMock(),
_generate_or_load_uid=MagicMock(),
):
ssdp_instance = MagicMock()
ssdp_server.return_value = ssdp_instance
main = Main(self.config)
tuner1 = MagicMock()
tuner2 = MagicMock()
main.tuners = [tuner1, tuner2]
main.start()
main._login.assert_called_once()
main._init_geos.assert_called_once()
main._init_tuners.assert_called_once()
main._check_ffmpeg.assert_called_once()
main._report.assert_called_once()
main._generate_or_load_uid.assert_called_once()
tuner1.start.assert_called_once()
tuner2.start.assert_called_once()
ssdp_server.assert_called()
ssdp_instance.start.assert_called()
def test_multiplex_debug(self, multiplexer: MagicMock):
self.config.multiplex = True
self.config.multiplex_debug = True
main = Main(self.config)
main.geos = [Geo()]
main._init_multiplexer()
multiplexer.assert_called_once_with(self.config, 6078, main.ssdp)
def test_port_multiplex_debug(self):
self.config.multiplex = True
self.config.multiplex_debug = True
main = Main(self.config)
port = main._port(0)
self.assertEqual(port, 6077)
port = main._port(1)
self.assertEqual(port, 6078)
def test_override_locations(self):
self.config.override_location = '1.99,2.33'
self.config.override_zipcodes = None
main = Main(self.config)
main._init_geos()
geo = Geo(coords={'latitude': '1.99', 'longitude': '2.33'})
self.assertEqual(len(main.geos), 1)
self.assertEqual(main.geos[0], geo)
def test_override_none(self):
self.config.override_location = None
self.config.override_zipcodes = None
main = Main(self.config)
main._init_geos()
self.assertEqual(len(main.geos), 1)
self.assertEqual(main.geos[0], Geo())
def test_tuners(self, tuner: MagicMock):
main = Main(self.config)
g1 = Geo()
g2 = Geo()
main.geos = [g1, g2]
main._init_tuners()
self.assertEqual(len(main.tuners), 2)
tuner.assert_any_call(g1, main.config, main.ssdp, port=6077)
tuner.assert_any_call(g1, main.config, main.ssdp, port=6078)
def test_direct(self):
self.config.direct = True
self.config.ffmpeg = None
with patch('locast2dvr.main.distutils.spawn.find_executable') as f:
main = Main(self.config)
main.log = MagicMock()
main._check_ffmpeg()
self.assertEqual(main.config.ffmpeg, None)
f.assert_not_called()
def test_ffmpeg_missing(self):
self.config.ffmpeg = None
with patch('locast2dvr.main.distutils.spawn.find_executable') as f:
f.return_value = None
main = Main(self.config)
main.log = MagicMock()
main._check_ffmpeg()
self.assertEqual(main.config.ffmpeg, None)
f.assert_called_once_with('ffmpeg')
def test_ffmpeg_from_config(self):
self.config.ffmpeg = '/usr/bin/ffmpeg-test'
with patch('locast2dvr.main.distutils.spawn.find_executable') as f:
f.return_value = '/usr/bin/ffmpeg-test'
main = Main(self.config)
main.log = MagicMock()
main._check_ffmpeg()
self.assertEqual(main.config.ffmpeg, '/usr/bin/ffmpeg-test')
f.assert_called_once_with('/usr/bin/ffmpeg-test')
def test_tuners_multiplex(self, tuner: MagicMock):
self.config.multiplex = True
main = Main(self.config)
g1 = Geo()
g2 = Geo()
main.geos = [g1, g2]
main._init_tuners()
self.assertEqual(len(main.tuners), 2)
tuner.assert_any_call(g1, main.config, main.ssdp, port=None)
tuner.assert_any_call(g2, main.config, main.ssdp, port=None)
def test_override_zipcodes(self):
self.config.override_location = None
self.config.override_zipcodes = '90210,11011'
main = Main(self.config)
main._init_geos()
self.assertEqual(len(main.geos), 2)
self.assertEqual(main.geos[0], Geo('90210'))
self.assertEqual(main.geos[1], Geo('11011'))
def test_main(self):
main = Main(self.config)
self.assertEqual(main.config, self.config)
self.assertEqual(main.geos, [])
self.assertEqual(main.tuners, [])
self.assertEqual(main.multiplexer, None)
self.assertEqual(main.ssdp, None)
def test_report_with_multiplexer(self):
main = Main(self.config)
main.multiplexer = MagicMock()
tuner1 = MagicMock(spec=Tuner)
tuner1.city = "TestTown"
tuner1.zipcode = "111111"
tuner1.dma = "373"
tuner1.uid = "TEST_0"
tuner1.url = None
tuner2 = MagicMock(spec=Tuner)
tuner2.city = "TestTown2"
tuner2.zipcode = "111112"
tuner2.dma = "372"
tuner2.uid = "TEST_1"
tuner2.url = "http://localhost:6789"
main.tuners = [tuner1, tuner2]
main.log = MagicMock()
main.multiplexer.url = "http://localhost:7890"
main.multiplexer.uid = "MULTI"
main._report()
self.assertEqual(len(main.log.info.mock_calls), 10)
def test_startup_with_multiplexer(self, *args):
with patch.multiple('locast2dvr.main.Main',
_login=MagicMock(return_value='New_Key'),
_init_geos=MagicMock(),
_init_multiplexer=MagicMock(),
_init_tuners=MagicMock(),
_check_ffmpeg=MagicMock(),
_report=MagicMock()):
main = Main(self.config)
tuners = [MagicMock(), MagicMock()]
main.tuners = tuners
main.multiplexer = MagicMock()
main.start()
def test_startup_no_ssdp(self, ssdp_server: MagicMock):
self.config.ssdp = False
with patch.multiple('locast2dvr.main.Main',
_login=MagicMock(return_value='New_Key'),
_init_geos=MagicMock(),
_init_multiplexer=MagicMock(),
_init_tuners=MagicMock(),
_check_ffmpeg=MagicMock(),
_report=MagicMock()):
main = Main(self.config)
tuners = [MagicMock(), MagicMock()]
main.tuners = tuners
main.multiplexer = MagicMock()
ssdp_server.return_value = ssdp_instance = MagicMock()
main.start()
ssdp_instance.start.assert_not_called()
def test_multiplex_none(self):
main = Main(self.config)
main._init_multiplexer()
self.assertIsNone(main.multiplexer)
def test_port(self):
main = Main(self.config)
port = main._port(0)
self.assertEqual(port, 6077)
port = main._port(1)
self.assertEqual(port, 6078)
def test_login_successfull(self):
with patch('locast2dvr.main.LocastService') as service:
main = Main(self.config)
main._login()
service.login.assert_called_once_with('foo', 'secret')