IDM-55 Add tests