Add tests for get data