Source code for amazon_dash.tests.test_discovery

from __future__ import print_function
import unittest
from amazon_dash.tests._compat import patch

from amazon_dash.discovery import BANNED_DEVICES, pkt_text, AMAZON_DEVICES, mac_id_list, discovery_print, discover
from amazon_dash.listener import Device



[docs]class TestDiscovery(unittest.TestCase):

[docs]    def test_banned_devices(self):
        banned = Device(BANNED_DEVICES[0])
        self.assertFalse(pkt_text(banned))



[docs]    def test_amazon_device(self):
        device = Device(AMAZON_DEVICES[0] + ':00:00:00')
        self.assertIn('Amazon', pkt_text(device))



[docs]    def test_device(self):
        device = Device('11:22:33:44:55:66')
        self.assertIn(device.src, pkt_text(device))



[docs]    @patch('sys.stdout')
    def test_discovery_print(self, write_patch):
        mac = '11:22:33:44:55:66'
        with patch('click.echo') as echo:
            discovery_print(Device(mac))
            discovery_print(Device(mac))
            echo.assert_called_once()
        mac_id_list.remove(mac)