Log experiment assignment upon launch

WMF Framework/Event Platform/EventPlatformClient.swift

@@ -151,7 +151,7 @@ import WMFData
         case talkPages = "/analytics/mobile_apps/ios_talk_page_interaction/2.1.0"
         case readingLists = "/analytics/mobile_apps/ios_reading_lists/2.2.0"
         case userHistory = "/analytics/mobile_apps/ios_user_history/2.0.0"
-        case search = "/analytics/mobile_apps/ios_search/2.2.0"
+        case search = "/analytics/mobile_apps/ios_search/2.3.1"
         case sessions = "/analytics/mobile_apps/app_session/1.1.0"
         case settings = "/analytics/mobile_apps/ios_setting_action/1.1.0"
         case login = "/analytics/mobile_apps/ios_login_action/1.1.0"

WMFData/Sources/WMFData/Data Controllers/Navigation/WMFNavigationExperimentsDataController.swift

@@ -26,7 +26,7 @@ public final class WMFNavigationExperimentsDataController {
         self.experimentsDataController = WMFExperimentsDataController(store: experimentStore)
     }
 
-    public func assignArticleSearchBarExperiment(project: WMFProject) throws {
+    public func assignArticleSearchBarExperiment(project: WMFProject) throws -> ArticleSearchBarExperimentAssignment {
 
         guard project.qualifiesForNavigationV2Experiment() else {
             throw CustomError.invalidProject
@@ -40,7 +40,16 @@ public final class WMFNavigationExperimentsDataController {
             throw CustomError.alreadyAssignedBucket
         }
 
-        try experimentsDataController.determineBucketForExperiment(.articleSearchBar, withPercentage: articleSearchBarExperimentPercentage)
+        let bucketValue = try experimentsDataController.determineBucketForExperiment(.articleSearchBar, withPercentage: articleSearchBarExperimentPercentage)
+
+        switch bucketValue {
+        case .articleSearchBarTest:
+            return ArticleSearchBarExperimentAssignment.test
+        case .articleSearchBarControl:
+            return ArticleSearchBarExperimentAssignment.control
+        default:
+            throw CustomError.unexpectedAssignment
+        }
     }
 
     public func articleSearchBarExperimentAssignment() throws -> ArticleSearchBarExperimentAssignment {

Wikipedia/Code/SearchFunnel.swift

@@ -1,3 +1,5 @@
+import WMFData
+
 @objc public final class SearchFunnel: NSObject {
     @objc static let shared = SearchFunnel()
 
@@ -18,11 +20,13 @@
         case click
         case cancel
         case langSwitch = "langswitch"
+        case launch
     }
 
     public struct Event: EventInterface {
         public static let schema: EventPlatformClient.Schema = .search
         let action: Action
+        let actionData: [String: String]?
         let source: String
         let position: Int?
         let search_type: String?
@@ -38,9 +42,9 @@
         return lang
     }
 
-    func logEvent(action: Action, source: String, position: Int? = nil, searchType: WMFSearchType? = nil, numberOfResults: Int? = nil, timeToDisplay: Int? = nil, wikiId: String?) {
+    func logEvent(action: Action, actionData: [String: String]? = nil, source: String, position: Int? = nil, searchType: WMFSearchType? = nil, numberOfResults: Int? = nil, timeToDisplay: Int? = nil, wikiId: String?) {
         guard let searchSessionToken else { return }
-        let event = Event(action: action, source: source, position: position, search_type: searchType?.rawValue, number_of_results: numberOfResults, time_to_display_results: timeToDisplay, session_token: searchSessionToken, wiki_id: wikiId)
+        let event = Event(action: action, actionData: actionData, source: source, position: position, search_type: searchType?.rawValue, number_of_results: numberOfResults, time_to_display_results: timeToDisplay, session_token: searchSessionToken, wiki_id: wikiId)
         EventPlatformClient.shared.submit(stream: .search, event: event)
     }
 
@@ -72,6 +76,15 @@
     func logShowSearchError(with type: WMFSearchType, elapsedTime: Double, source: String) {
         logEvent(action: .error, source: source, searchType: type, timeToDisplay: Int(elapsedTime * 1000), wikiId: searchLanguage)
     }
+
+    func logDidAssignArticleSearchExperiment(assignment: WMFNavigationExperimentsDataController.ArticleSearchBarExperimentAssignment) {
+        let group: String
+        switch assignment {
+        case .control: group = "a"
+        case .test: group = "b"
+        }
+        logEvent(action: .launch, actionData: ["group": group], source: "unknown", wikiId: nil)
+    }
 
 }
 

Wikipedia/Code/WMFAppViewController+Extensions.swift

@@ -105,11 +105,11 @@ extension WMFAppViewController {
         }
 
         do {
-            try dataController.assignArticleSearchBarExperiment(project: project)
+            let assignment = try dataController.assignArticleSearchBarExperiment(project: project)
+            SearchFunnel.shared.logDidAssignArticleSearchExperiment(assignment: assignment)
         } catch {
             DDLogWarn("Error assigning article search experiment: \(error)")
         }
-
     }
 
 }