Add tests for link model