Update UI tests

StanfordBDHG · Apr 4, 2024 · ad5196c · ad5196c
1 parent 9a30837
commit ad5196c
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/HealthGPT/Onboarding/LLMSource.swift b/HealthGPT/Onboarding/LLMSource.swift
@@ -7,7 +7,6 @@
 //
 
 import Foundation
-import SpeziLLMOpenAI
 
 
 enum LLMSource: String, CaseIterable, Identifiable, Codable {

diff --git a/HealthGPTUITests/OnboardingUITests.swift b/HealthGPTUITests/OnboardingUITests.swift
@@ -73,8 +73,16 @@ extension XCUIApplication {
 
         let picker = pickers["llmSourcePicker"]
         let optionToSelect = picker.pickerWheels.element(boundBy: 0)
-        optionToSelect.adjust(toPickerWheelValue: "Open AI LLM")
+        optionToSelect.adjust(toPickerWheelValue: "On-device LLM")
+
+        XCTAssertTrue(buttons["Save Choice"].waitForExistence(timeout: 5))
+        buttons["Save Choice"].tap()
+
+        XCTAssertTrue(staticTexts["LLM Download"].waitForExistence(timeout: 5))
+        XCTAssertTrue(buttons["Back"].waitForExistence(timeout: 2))
+        buttons["Back"].tap()
 
+        optionToSelect.adjust(toPickerWheelValue: "Open AI LLM")
         XCTAssertTrue(buttons["Save Choice"].waitForExistence(timeout: 5))
         buttons["Save Choice"].tap()
     }
-Original file line number
+Diff line change
@@ Expand Up / @@ -7,7 +7,6 @@ @@
     //
     import Foundation
-    import SpeziLLMOpenAI
     enum LLMSource: String, CaseIterable, Identifiable, Codable {
@@ Expand Down @@