Add more unit test for KeyAssembler.