Add shortcode tests