def test_url(): command = Command('method', ['path'], {}) url = command.url('host') assert url == 'coaps://host:5684/path' command2 = Command('method', ['path1', 'path2'], {}) url = command2.url('host') assert url == 'coaps://host:5684/path1/path2'
def test_url(): command = Command("method", ["path"], {}) url = command.url("host") assert url == "coaps://host:5684/path" command2 = Command("method", ["path1", "path2"], {}) url = command2.url("host") assert url == "coaps://host:5684/path1/path2"
def test_combining_dict_keys(): DATA_EMPTY_DICT = {'key_dict': {}} DATA_DICT_INT = {'key_dict': {'key_int': 0}} DATA_DICT_STRING = {'key_dict': {'key_string': 'a'}} DATA_DICT_STRING2 = {'key_dict': {'key_string': 'b'}} DATA_DICT_INTSTRING = {'key_dict': {'key_int': 0, 'key_string': 'a'}} command1 = Command('method', 'path', DATA_EMPTY_DICT) command2 = Command('method', 'path', DATA_DICT_INT) combined = command1 + command2 assert combined._data == DATA_DICT_INT command1 = Command('method', 'path', DATA_DICT_INT) command2 = Command('method', 'path', DATA_DICT_STRING) combined = command1 + command2 assert combined._data == DATA_DICT_INTSTRING command1 = Command('method', 'path', DATA_DICT_STRING) command2 = Command('method', 'path', DATA_DICT_STRING2) combined = command1 + command2 assert combined._data == DATA_DICT_STRING2 command1 = Command('method', 'path', DATA_DICT_INT) command2 = Command('method', 'path', DATA_DICT_STRING2) command3 = Command('method', 'path', DATA_DICT_STRING) combined = command1 + command2 + command3 assert combined._data == DATA_DICT_INTSTRING
def test_combining_dict_keys(): DATA_EMPTY_DICT = {"key_dict": {}} DATA_DICT_INT = {"key_dict": {"key_int": 0}} DATA_DICT_STRING = {"key_dict": {"key_string": "a"}} DATA_DICT_STRING2 = {"key_dict": {"key_string": "b"}} DATA_DICT_INTSTRING = {"key_dict": {"key_int": 0, "key_string": "a"}} command1 = Command("method", "path", DATA_EMPTY_DICT) command2 = Command("method", "path", DATA_DICT_INT) combined = command1 + command2 assert combined._data == DATA_DICT_INT command1 = Command("method", "path", DATA_DICT_INT) command2 = Command("method", "path", DATA_DICT_STRING) combined = command1 + command2 assert combined._data == DATA_DICT_INTSTRING command1 = Command("method", "path", DATA_DICT_STRING) command2 = Command("method", "path", DATA_DICT_STRING2) combined = command1 + command2 assert combined._data == DATA_DICT_STRING2 command1 = Command("method", "path", DATA_DICT_INT) command2 = Command("method", "path", DATA_DICT_STRING2) command3 = Command("method", "path", DATA_DICT_STRING) combined = command1 + command2 + command3 assert combined._data == DATA_DICT_INTSTRING
def test_combining_list_keys(): DATA_EMPTY_LIST = {'key_list': []} DATA_INT_LIST1 = {'key_list': [0, 1, 2]} DATA_INT_LIST2 = {'key_list': [10, 11, 12]} command1 = Command('method', 'path', DATA_EMPTY_LIST) command2 = Command('method', 'path', DATA_INT_LIST1) combined = command1 + command2 assert combined._data == DATA_INT_LIST1 # Duplicated keys are replaced if not dicts command1 = Command('method', 'path', DATA_INT_LIST1) command2 = Command('method', 'path', DATA_INT_LIST2) combined = command1 + command2 assert combined._data == DATA_INT_LIST2
def test_combining_listed_dict_keys(): DATA_EMPTY_DICT = {"key_ldict": [{}]} DATA_DICT_INT = {"key_ldict": [{"key_int": 0}]} DATA_DICT_STRING = {"key_ldict": [{"key_string": "a"}]} DATA_DICT_INTSTRING = {"key_ldict": [{"key_int": 0, "key_string": "a"}]} command1 = Command("method", "path", DATA_EMPTY_DICT) command2 = Command("method", "path", DATA_DICT_INT) combined = command1 + command2 assert combined._data == DATA_DICT_INT command1 = Command("method", "path", DATA_DICT_INT) command2 = Command("method", "path", DATA_DICT_STRING) combined = command1 + command2 assert combined._data == DATA_DICT_INTSTRING
def test_combining_listed_dict_keys(): DATA_EMPTY_DICT = {'key_ldict': [{}]} DATA_DICT_INT = {'key_ldict': [{'key_int': 0}]} DATA_DICT_STRING = {'key_ldict': [{'key_string': 'a'}]} DATA_DICT_INTSTRING = {'key_ldict': [{'key_int': 0, 'key_string': 'a'}]} command1 = Command('method', 'path', DATA_EMPTY_DICT) command2 = Command('method', 'path', DATA_DICT_INT) combined = command1 + command2 assert combined._data == DATA_DICT_INT command1 = Command('method', 'path', DATA_DICT_INT) command2 = Command('method', 'path', DATA_DICT_STRING) combined = command1 + command2 assert combined._data == DATA_DICT_INTSTRING
def test_combining_integer_keys(): DATA_INT = {'key_int': 0} DATA_INT_SAME_KEY = {'key_int': 1} DATA_INT2 = {'key_int_2': 1} COMBINED_INT = {'key_int': 0, 'key_int_2': 1} command1 = Command('method', 'path', DATA_INT) command2 = Command('method', 'path', DATA_INT2) combined = command1 + command2 assert combined._data == COMBINED_INT command1 = Command('method', 'path', DATA_INT) command2 = Command('method', 'path', DATA_INT_SAME_KEY) # We should always take the last key if we can't merge combined = command1 + command2 assert combined._data == DATA_INT_SAME_KEY
def test_property_access(): def pr(): pass def ec(): pass command = Command( method="method", path="path", data="data", parse_json=True, observe=False, observe_duration=0, process_result=pr, err_callback=ec, ) assert command.method == "method" assert command.path == "path" assert command.parse_json is True assert command.observe is False assert command.observe_duration == 0 assert command.process_result == pr assert command.err_callback == ec
def test_combining_string_keys(): DATA_STRING = {"key_string": "a"} DATA_STRING_SAME_KEY = {"key_string": "same"} DATA_STRING2 = {"key_string_2": "b"} COMBINED_STRING = {"key_string": "a", "key_string_2": "b"} command1 = Command("method", "path", DATA_STRING) command2 = Command("method", "path", DATA_STRING2) combined = command1 + command2 assert combined._data == COMBINED_STRING command1 = Command("method", "path", DATA_STRING) command2 = Command("method", "path", DATA_STRING_SAME_KEY) # We should always take the last key if we can't merge combined = command1 + command2 assert combined._data == DATA_STRING_SAME_KEY
def test_combining_string_keys(): DATA_STRING = {'key_string': 'a'} DATA_STRING_SAME_KEY = {'key_string': 'same'} DATA_STRING2 = {'key_string_2': 'b'} COMBINED_STRING = {'key_string': 'a', 'key_string_2': 'b'} command1 = Command('method', 'path', DATA_STRING) command2 = Command('method', 'path', DATA_STRING2) combined = command1 + command2 assert combined._data == COMBINED_STRING command1 = Command('method', 'path', DATA_STRING) command2 = Command('method', 'path', DATA_STRING_SAME_KEY) # We should always take the last key if we can't merge combined = command1 + command2 assert combined._data == DATA_STRING_SAME_KEY
def set_values(self, values, *, index=0): """ Set values on socket control. Returns a Command. """ assert len(self.raw) == 1, "Only devices with 1 socket supported" return Command("put", self._device.path, {ATTR_SWITCH_PLUG: [values]})
def set_values(self, values, *, index=0): """ Set values on light control. Returns a Command. """ assert len(self.raw) == 1, "Only devices with 1 light supported" return Command("put", self._device.path, {ATTR_LIGHT_CONTROL: [values]})
def test_combining_mutates(): DATA_INT = {"key_int": 0} DATA_INT2 = {"key_int_2": 1} COMBINED_INT = {"key_int": 0, "key_int_2": 1} command1 = Command("method", "path", DATA_INT) command2 = Command("method", "path", DATA_INT2) combined = command1 + command2 # Adding shouldn't mutate the original commands assert command1._data == DATA_INT assert command2._data == DATA_INT2 assert combined._data == COMBINED_INT # Combining should mutate the original command command1.combine_data(command2) assert command1._data == COMBINED_INT assert command2._data == DATA_INT2
def test_combining_mutates(): DATA_INT = {'key_int': 0} DATA_INT2 = {'key_int_2': 1} COMBINED_INT = {'key_int': 0, 'key_int_2': 1} command1 = Command('method', 'path', DATA_INT) command2 = Command('method', 'path', DATA_INT2) combined = command1 + command2 # Adding shouldn't mutate the original commands assert command1._data == DATA_INT assert command2._data == DATA_INT2 assert combined._data == COMBINED_INT # Combining should mutate the original command command1.combine_data(command2) assert command1._data == COMBINED_INT assert command2._data == DATA_INT2
def test_combining_with_none(): DATA_INT = {"key_int": 0} command1 = Command("method", "path", DATA_INT) combined = command1 + None assert combined._data == DATA_INT # Combining should mutate the original command command1.combine_data(None) assert command1._data == DATA_INT
async def test_request_returns_list(monkeypatch): monkeypatch.setattr("aiocoap.Context.create_client_context", mock_create_context) api = (await APIFactory.init("127.0.0.1")).request command = Command("", "") response = await api([command, command, command]) assert type(response) == list
async def test_request_returns_list(monkeypatch): monkeypatch.setattr('aiocoap.Context.create_client_context', mock_create_context) api = APIFactory('127.0.0.1').request command = Command('', '') response = await api([command, command, command]) assert type(response) == list
def test_combining_with_none(): DATA_INT = {'key_int': 0} command1 = Command('method', 'path', DATA_INT) combined = command1 + None assert combined._data == DATA_INT # Combining should mutate the original command command1.combine_data(None) assert command1._data == DATA_INT
def test_request_returns_single(monkeypatch): monkeypatch.setattr('aiocoap.Context.create_client_context', mock_create_context) api = APIFactory('127.0.0.1').request command = Command('', '') response = yield from api(command) assert type(response) != list
def test_result(): def pr(value): return value + 1 command = Command('method', 'path', {}, process_result=pr) assert command.result is None assert command.raw_result is None command.result = 0 assert command.result == 1 assert command.raw_result == 0
def test_result(): """Test callback process_result.""" def process_result(value): return value + 1 command = Command("method", "path", {}, process_result=process_result) assert command.result is None assert command.raw_result is None command.process_result(0) assert command.result == 1 assert command.raw_result == 0
def dump_all(): endpoints = api(gateway.get_endpoints()) for endpoint in endpoints: parts = endpoint[1:].split('/') if not all(part.isdigit() for part in parts): continue pprint(api(Command('get', parts))) print() print()
async def test_request_returns_single(monkeypatch): """Test return single object.""" monkeypatch.setattr("aiocoap.Context.create_client_context", mock_create_context) api = (await APIFactory.init("127.0.0.1")).request command = Command("", "") response = await api(command) assert not isinstance(response, list)
def set_value(self, value): """ Set values on blind control. Returns a Command. """ return Command('put', self._device.path, { ATTR_START_BLINDS: [ { ATTR_BLIND_CURRENT_POSITION: value } ] })
def fetch(self): try: logger.info("Force fetch state for {}".format(self.device.name)) def process_result(result): return Device(result) self.device = self.api( Command('get', [ROOT_DEVICES, self.device.id], process_result=process_result)) time.sleep(0.1) except RequestTimeout as e: self.dirty = True logger.info("There was timeout for actions, retry in some time")
def test_property_access(): """Test property access in Command.""" def error_callback(): pass command = Command( method="method", path="path", data="data", parse_json=True, observe=False, observe_duration=0, err_callback=error_callback, ) assert command.method == "method" assert command.path == "path" assert command.parse_json is True assert command.observe is False assert command.observe_duration == 0 assert command.err_callback is error_callback
def test_property_access(): def pr(): pass def ec(): pass command = Command(method='method', path='path', data='data', parse_json=True, observe=False, observe_duration=0, process_result=pr, err_callback=ec) assert command.method == 'method' assert command.path == 'path' assert command.parse_json is True assert command.observe is False assert command.observe_duration == 0 assert command.process_result == pr assert command.err_callback == ec
async def run(shutdown): # Assign configuration variables. # The configuration check takes care they are present. conf = load_json(CONFIG_FILE) try: identity = conf[args.host].get('identity') psk = conf[args.host].get('key') api_factory = APIFactory(host=args.host, psk_id=identity, psk=psk) except KeyError: identity = uuid.uuid4().hex api_factory = APIFactory(host=args.host, psk_id=identity) try: psk = await api_factory.generate_psk(args.key) print('Generated PSK: ', psk) conf[args.host] = {'identity': identity, 'key': psk} save_json(CONFIG_FILE, conf) except AttributeError: raise PytradfriError("Please provide the 'Security Code' on the " "back of your Tradfri gateway using the " "-K flag.") api = api_factory.request gateway = Gateway() # end copy/pasted # # set and regularly renew the commissioning timeout, remove when done # async def keep_commissioning_alive(readiness): try: while True: await api(gateway.set_commissioning_timeout(60)) if readiness is not None: readiness() readiness = None await asyncio.sleep(45) finally: await api(gateway.set_commissioning_timeout(00)) commissioning_ready = asyncio.Future() commissioning = asyncio.Task( keep_commissioning_alive(lambda: commissioning_ready.set_result(None))) # # monitor the device list and give instructions # last_devices = None def devices_updated(result): nonlocal last_devices if last_devices is None: print("Originally, %s device(s) are known" % len(result)) else: for r in result: if r not in last_devices: asyncio.Task(new_device(r)) last_devices = result async def new_device(devno): nonlocal commissioning print("New device, fetching details...", end="", flush=True) device_command = gateway.get_device(devno) device = await api(device_command) print() print(" New device description: %s" % (device, )) if commissioning: if device.has_light_control: print("That was not in the expected sequence: This device was" " a light and not a controller. You can still pair" " another controller device.") else: print("Found a controller. You can now go ahead and add light" " bulbs by pairing them to the switch as you would do" " without a gateway. Press Ctrl-C when done.") commissioning.cancel() commissioning = None # if you wanted to implemente infinite-commissioning mode, you # should cancel or restart keep_commissioning_alive in a way # that resets the timeout, because the timeout will have gone # to 0 the moment the device was added. else: if not device.has_light_control: print("That was unexpected: A controller showed up even though" " the gateway was not in pairing mode any more.") else: print("You can still add more light bulbs; press Ctrl-C when" " done.") observe_devices = Command('get', [ROOT_DEVICES], observe=True, process_result=devices_updated) await api(observe_devices) await commissioning_ready print("Ready to start: Gateway is in commissioning mode.") print("Pressing the pairing button on a switch, dimmer or motion detector" " for 10s near the gateway until the gateway blinks fast. A few" " seconds later, it the new device shows up here. You may need to" " switch off light bulbs in the immediate vicinity (?).") # # run until the outer loop says not to any more # await api_factory.shutdown() await shutdown if commissioning is not None: print("Please allow for the commissioning mode to be disabled") commissioning.cancel()
def set_value(self, value): """ Set values on blind control. Returns a Command. """ return Command('put', self._device.path, {ATTR_START_BLINDS: [value]})