Add test for get_connection