def test_add_find_board(self):
"""
Test board object is added and found.
:return:
"""
mapper = TrololoIdMapper("/tmp")
mapper.add_board(
TrololoBoard.load(None, {
"id": "han_solo",
"name": "Millennium Falcon"
}))
s_res = mapper.get_id_by_name("Millennium Falcon")
assert s_res[TrololoIdMapper.S_BOARD].pop() == "han_solo"
def test_save_dump(self):
"""
Test save.
:return:
"""
_name = "Loop in redundant loopback"
_id = "networking"
mapper = TrololoIdMapper("/tmp")
mapper.add_board(TrololoBoard.load(None, {"id": _id, "name": _name}))
dumper = MagicMock()
with patch("pickle.dump", dumper):
mapper.save()
assert dumper.called
assert dumper.call_args[0][0][TrololoIdMapper.S_BOARD].get(
_name).pop() == _id