Add tests for links