Functional view controller testing

YamamotoDesu · Mar 28, 2023 · 5ddbcb4 · 5ddbcb4
1 parent d24240b
commit 5ddbcb4
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/04-test-expressions/projects/starter/FitNess/FitNess/UI Layer/StepCountController.swift b/04-test-expressions/projects/starter/FitNess/FitNess/UI Layer/StepCountController.swift
@@ -101,6 +101,7 @@ class StepCountController: UIViewController {
 extension StepCountController {
   func updateGoal(newGoal: Int) {
     // update this function
+    AppModel.instance.dataModel.goal = newGoal
   }
 
   private func showNeedGoalAlert() {

diff --git a/...sions/projects/starter/FitNess/FitNessTests/Cases/UI Layer/StepCountControllerTests.swift b/...sions/projects/starter/FitNess/FitNessTests/Cases/UI Layer/StepCountControllerTests.swift
@@ -44,6 +44,7 @@ class StepCountControllerTests: XCTestCase {
 
   override func tearDownWithError() throws {
     sut = nil
+    AppModel.instance.dataModel.goal = nil
     try super.tearDownWithError()
   }
 
@@ -88,4 +89,11 @@ class StepCountControllerTests: XCTestCase {
   }
 
   // MARK: - Chase View
+  func testDataModel_whenGoalUpdate_updatesToNewGoal() {
+    // when
+    sut.updateGoal(newGoal: 50)
+
+    // then
+    XCTAssertEqual(AppModel.instance.dataModel.goal, 50)
+  }
 }