[ticket/1] Add tests