def test_add_to_registry_legacy_sig(fake_dev):
reg = DeviceRegistry()
quirk_list = mock.MagicMock()
model_dict = mock.MagicMock(spec_set=dict)
model_dict.__getitem__.return_value = quirk_list
manuf_dict = mock.MagicMock()
manuf_dict.__getitem__.return_value = model_dict
reg._registry = manuf_dict
fake_dev.signature = {
1: {},
2: {},
3: {
"manufacturer": mock.sentinel.legacy_manufacturer,
"model": mock.sentinel.legacy_model,
},
}
reg.add_to_registry(fake_dev)
assert manuf_dict.__getitem__.call_count == 1
assert manuf_dict.__getitem__.call_args[0][
0] is mock.sentinel.legacy_manufacturer
assert model_dict.__getitem__.call_count == 1
assert model_dict.__getitem__.call_args[0][0] is mock.sentinel.legacy_model
assert quirk_list.append.call_count == 1
assert quirk_list.append.call_args[0][0] is fake_dev
def test_remove_new_sig(fake_dev):
fake_dev.signature = {
1: {},
2: {},
3: {
'manufacturer': mock.sentinel.legacy_manufacturer,
'model': mock.sentinel.legacy_model,
},
'endpoints': {
1: {
'manufacturer': mock.sentinel.manufacturer,
'model': mock.sentinel.model,
}
},
'manufacturer': mock.sentinel.dev_manufacturer,
'model': mock.sentinel.dev_model,
}
reg = DeviceRegistry()
quirk_list = mock.MagicMock()
model_dict = mock.MagicMock(spec_set=dict)
model_dict.__getitem__.return_value = quirk_list
manuf_dict = mock.MagicMock()
manuf_dict.__getitem__.return_value = model_dict
reg._registry = manuf_dict
reg.remove(fake_dev)
assert manuf_dict.__getitem__.call_count == 1
assert manuf_dict.__getitem__.call_args[0][0] is mock.sentinel.dev_manufacturer
assert model_dict.__getitem__.call_count == 1
assert model_dict.__getitem__.call_args[0][0] is mock.sentinel.dev_model
assert quirk_list.append.call_count == 0
assert quirk_list.remove.call_count == 1
assert quirk_list.remove.call_args[0][0] is fake_dev
def test_add_to_registry_models_info(fake_dev):
fake_dev.signature = {
1: {},
2: {},
3: {
"manufacturer": mock.sentinel.legacy_manufacturer,
"model": mock.sentinel.legacy_model,
},
"endpoints": {
1: {
"manufacturer": mock.sentinel.manufacturer,
"model": mock.sentinel.model,
}
},
SIG_MODELS_INFO: [
(mock.sentinel.manuf_1, mock.sentinel.model_1),
(mock.sentinel.manuf_2, mock.sentinel.model_2),
],
}
reg = DeviceRegistry()
quirk_list = mock.MagicMock()
model_dict = mock.MagicMock(spec_set=dict)
model_dict.__getitem__.return_value = quirk_list
manuf_dict = mock.MagicMock()
manuf_dict.__getitem__.return_value = model_dict
reg._registry = manuf_dict
reg.add_to_registry(fake_dev)
assert manuf_dict.__getitem__.call_count == 2
assert manuf_dict.__getitem__.call_args_list[0][0][
0] is mock.sentinel.manuf_1
assert manuf_dict.__getitem__.call_args_list[1][0][
0] is mock.sentinel.manuf_2
assert model_dict.__getitem__.call_count == 2
assert model_dict.__getitem__.call_args_list[0][0][
0] is mock.sentinel.model_1
assert model_dict.__getitem__.call_args_list[1][0][
0] is mock.sentinel.model_2
assert quirk_list.append.call_count == 2
assert quirk_list.append.call_args_list[0][0][0] is fake_dev
assert quirk_list.append.call_args_list[1][0][0] is fake_dev
quirk_list.reset_mock()
model_dict.reset_mock()
manuf_dict.reset_mock()
