From fe8ff37f66ef96fc0f9c93e398898f95893e5af9 Mon Sep 17 00:00:00 2001
From: ThetaDev <t.testboy@gmail.com>
Date: Thu, 24 Nov 2022 20:03:38 +0100
Subject: [PATCH] feat(codegen): add A/B test 3: channel handles in search
 results

---
 codegen/src/abtest.rs | 33 +++++++++++++++++++++++++++++----
 codegen/src/main.rs   |  2 +-
 2 files changed, 30 insertions(+), 5 deletions(-)

diff --git a/codegen/src/abtest.rs b/codegen/src/abtest.rs
index ae7a000..1ec83c6 100644
--- a/codegen/src/abtest.rs
+++ b/codegen/src/abtest.rs
@@ -3,6 +3,8 @@ use futures::{stream, StreamExt};
 use indicatif::{ProgressBar, ProgressStyle};
 use num_enum::TryFromPrimitive;
 use rustypipe::client::{ClientType, RustyPipe, YTContext};
+use rustypipe::model::YouTubeItem;
+use rustypipe::param::search_filter::{Entity, SearchFilter};
 use serde::{Deserialize, Serialize};
 
 #[derive(
@@ -12,9 +14,13 @@ use serde::{Deserialize, Serialize};
 pub enum ABTest {
     AttributedTextDescription = 1,
     ThreeTabChannelLayout = 2,
+    ChannelHandlesInSearchResults = 3,
 }
 
-const N_TESTS: u16 = 2;
+const TESTS_TO_RUN: [ABTest; 2] = [
+    ABTest::AttributedTextDescription,
+    ABTest::ChannelHandlesInSearchResults,
+];
 
 #[derive(Debug, Serialize, Deserialize)]
 pub struct ABTestRes {
@@ -63,6 +69,9 @@ pub async fn run_test(
                     ABTest::ThreeTabChannelLayout => {
                         three_tab_channel_layout(&rp, &visitor_data).await
                     }
+                    ABTest::ChannelHandlesInSearchResults => {
+                        channel_handles_in_search_results(&rp, &visitor_data).await
+                    }
                 }
                 .unwrap();
                 pb.inc(1);
@@ -103,11 +112,10 @@ async fn get_visitor_data(http: &reqwest::Client) -> String {
 pub async fn run_all_tests(n: usize, concurrency: usize) -> Vec<ABTestRes> {
     let mut results = Vec::new();
 
-    for id in 1..=N_TESTS {
-        let ab = ABTest::try_from(id).unwrap();
+    for ab in TESTS_TO_RUN {
         let (occurrences, _, _) = run_test(ab, n, concurrency).await;
         results.push(ABTestRes {
-            id,
+            id: ab as u16,
             name: ab,
             tests: n,
             occurrences,
@@ -145,3 +153,20 @@ pub async fn three_tab_channel_layout(rp: &RustyPipe, visitor_data: &str) -> Res
         .unwrap();
     Ok(channel.has_live || channel.has_shorts)
 }
+
+pub async fn channel_handles_in_search_results(rp: &RustyPipe, visitor_data: &str) -> Result<bool> {
+    let search = rp
+        .query()
+        .visitor_data(visitor_data)
+        .search_filter("rust", &SearchFilter::new().entity(Entity::Channel))
+        .await
+        .unwrap();
+
+    Ok(search.items.items.iter().any(|itm| match itm {
+        YouTubeItem::Channel(channel) => channel
+            .subscriber_count
+            .map(|sc| sc > 100 && channel.video_count.is_none())
+            .unwrap_or_default(),
+        _ => false,
+    }))
+}
diff --git a/codegen/src/main.rs b/codegen/src/main.rs
index a0f83bf..66a02c0 100644
--- a/codegen/src/main.rs
+++ b/codegen/src/main.rs
@@ -86,7 +86,7 @@ async fn main() {
                 }
                 None => {
                     let res = abtest::run_all_tests(n, cli.concurrency).await;
-                    println!("{}", serde_json::to_string(&res).unwrap())
+                    println!("{}", serde_json::to_string_pretty(&res).unwrap())
                 }
             };
         }