🐛 fix: 기수 변경 취소 후 내 정보가 표시되지 않는 문제 수정

Projects/App/Sources/Navigation/Profile/ProfileCoordinator.swift

@@ -7,6 +7,7 @@
 
 import Foundation
 
+import AuthDomainInterface
 import DDDSharedUI
 import OnBoarding
 import Profile
@@ -22,6 +23,7 @@ public struct ProfileCoordinator: Sendable {
   @ObservableState
   public struct State: Equatable {
     var routes: [Route<ProfileScreen.State>]
+    @Shared(.isEditingGeneration) var editGeneration
 
     public init() {
       routes = [.root(.profile(.init()), embedInNavigationView: true)]
@@ -51,14 +53,20 @@ public struct ProfileCoordinator: Sendable {
 
   }
 
-  public enum NavigationAction: Equatable {
+  public enum NavigationAction: Equatable, Sendable {
     case presentLogin
     case presentRoot
     case presentStaff
     case presentMember
   }
 
   func handleRoute(state: inout State, action: Action) -> Effect<Action> {
+    defer {
+      // 버튼뿐 아니라 시스템 뒤로가기(updateRoutes)로 편집 화면이 닫혀도 조회를 재개한다.
+      if state.editGeneration, !state.routes.contains(where: { $0.screen.onBoarding != nil }) {
+        state.$editGeneration.withLock { $0 = false }
+      }
+    }
     switch action {
       case .router(let action):
         return routerAction(state: &state, action: action)
@@ -177,6 +185,7 @@ extension ProfileCoordinator {
     state: inout State,
     action: NavigationAction
   ) -> Effect<Action> {
+    state.$editGeneration.withLock { $0 = false }
     switch action {
     case .presentLogin:
       return .none

Projects/App/Sources/Reducer/AppReducer.swift

@@ -80,6 +80,7 @@ public struct AppReducer: Sendable {
   @Dependency(\.continuousClock) var clock
   @Shared(.userSession) var userSession
   @Shared(.staffRole) var staffRole
+  @Shared(.isEditingGeneration) var editGeneration
 
   private enum CancelID: Hashable {
     case transition
@@ -176,6 +177,7 @@ public struct AppReducer: Sendable {
       // 로그아웃·토큰 만료 이후 새 계정의 프로필 조회 전까지 이전 ID가 남지 않도록 한다.
       $userSession.withLock { $0 = .empty }
       $staffRole.withLock { $0 = nil }
+      $editGeneration.withLock { $0 = false }
       state = .auth(.init())
       return .none
 

Projects/App/Tests/Sources/LogoutNavigationTests.swift

@@ -12,6 +12,8 @@ import Testing
 @testable import DDDStorage
 import AuthDomainInterface
 import DDDStorageInterface
+@testable import Profile
+import ProfileDomainInterface
 
 import ComposableArchitecture
 import SQLiteData
@@ -64,16 +66,19 @@ struct LogoutNavigationTests {
     } operation: {
       @Shared(.userSession) var session
       @Shared(.staffRole) var role
+      @Shared(.isEditingGeneration) var editing
       $session.withLock {
         $0 = UserSession(userID: 42, name: "이전 계정", userRole: .manager, accessToken: "old-token")
       }
       $role.withLock { $0 = .manager }
+      $editing.withLock { $0 = true }
       let store = Store(initialState: AppReducer.State.member(.init())) { AppReducer() }
 
       await store.send(expired ? .async(.refreshTokenExpired) : .view(.presentAuth)).finish()
 
       #expect(session == .empty)
       #expect(role == nil)
+      #expect(!editing)
       guard case .auth = store.state else {
         Issue.record("세션을 비운 뒤 인증 화면으로 전환해야 한다")
         return
@@ -84,6 +89,74 @@ struct LogoutNavigationTests {
     }
   }
 
+  @Test("기수 변경을 버튼이나 시스템 뒤로가기로 취소하면 내 정보 재조회가 가능하다", arguments: [false, true])
+  func generationEditCancellationRestoresProfileLoading(systemBack: Bool) async {
+    await withDependencies {
+      $0.defaultInMemoryStorage = InMemoryStorage()
+      $0.profileUseCase = MockProfileRepository()
+    } operation: {
+      let store = Store(initialState: ProfileCoordinator.State()) { ProfileCoordinator() }
+      await store.send(.router(.routeAction(
+        id: 0, action: .profile(.delegate(.presentEditGeneration))
+      ))).finish()
+      #expect(store.editGeneration)
+      #expect(store.routes.count == 2)
+
+      // 편집 도중의 단계 이동/뒤로가기는 편집 모드를 유지한다.
+      await store.send(.router(.routeAction(
+        id: 1, action: .onBoarding(.router(.routeAction(
+          id: 0, action: .InviteCode(.delegate(.presentSignUpName))
+        )))
+      ))).finish()
+      await store.send(.router(.routeAction(
+        id: 1, action: .onBoarding(.router(.routeAction(
+          id: 1, action: .onBoardingName(.delegate(.presentBack))
+        )))
+      ))).finish()
+      #expect(store.editGeneration)
+
+      if systemBack {
+        await store.send(.router(.updateRoutes(Array(store.routes.prefix(1))))).finish()
+      } else {
+        await store.send(.router(.routeAction(
+          id: 1, action: .onBoarding(.router(.routeAction(
+            id: 0, action: .InviteCode(.delegate(.presentBack))
+          )))
+        ))).finish()
+      }
+      #expect(!store.editGeneration)
+      #expect(store.routes.count == 1)
+
+      await store.send(.router(.routeAction(id: 0, action: .profile(.async(.fetchUser))))).finish()
+      #expect(store.routes.first?.screen.profile?.viewState == .loaded)
+      #expect(store.routes.first?.screen.profile?.profile != nil)
+
+      // 내 정보를 닫았다 다시 열어도 같은 공유 플래그가 조회를 막지 않는다.
+      let reopened = Store(initialState: ProfileCoordinator.State()) { ProfileCoordinator() }
+      await reopened.send(.router(.routeAction(id: 0, action: .profile(.async(.fetchUser))))).finish()
+      #expect(reopened.routes.first?.screen.profile?.profile != nil)
+    }
+  }
+
+  @Test("기수 변경 후 홈이나 로그인으로 전환하면 편집 모드를 종료한다", arguments: [
+    ProfileCoordinator.NavigationAction.presentMember,
+    .presentStaff,
+    .presentLogin
+  ])
+  func generationEditNavigationClearsEditing(action: ProfileCoordinator.NavigationAction) async {
+    await withDependencies {
+      $0.defaultInMemoryStorage = InMemoryStorage()
+    } operation: {
+      let store = Store(initialState: ProfileCoordinator.State()) { ProfileCoordinator() }
+      await store.send(.router(.routeAction(
+        id: 0, action: .profile(.delegate(.presentEditGeneration))
+      ))).finish()
+      #expect(store.editGeneration)
+      await store.send(.navigation(action)).finish()
+      #expect(!store.editGeneration)
+    }
+  }
+
   @Test("Profile 로그아웃은 화면을 pop하지 않고 로그인 전환을 요청한다")
   func profileLogoutRequestsLoginWithoutPoppingRoutes() async {
     let store = TestStore(initialState: ProfileCoordinator.State()) {

Projects/Domain/AuthDomain/Interface/Sources/Entity/UserSessionSharedKey.swift

@@ -13,6 +13,14 @@ import ProfileDomainInterface
 import Dependencies
 import Sharing
 
+public extension SharedReaderKey where Self == InMemoryKey<Bool>.Default {
+  /// 기수 변경 화면이 열려 있는 동안만 유지한다.
+  /// 이전 버전의 UserDefaults 값은 읽지 않아 중단된 편집이 다음 실행의 조회를 막지 않는다.
+  static var isEditingGeneration: Self {
+    Self[.inMemory("isEditingGeneration"), default: false]
+  }
+}
+
 public extension SharedReaderKey where Self == PersistentSharedKey<UserSession>.Default {
   /// 앱 실행 사이에 유지되는 사용자 세션 메타데이터입니다.
   ///

Projects/Feature/Auth/Sources/Login/Reducer/LoginFeature.swift

@@ -26,7 +26,7 @@ public struct LoginFeature {
 
     @Shared var userSession: UserSession
     @Shared(.staffRole) var staffRole
-    @Shared(.appStorage("editGeneration")) var editGeneration: Bool = false
+    @Shared(.isEditingGeneration) var editGeneration
     var currentSocialType: SocialType?
     @Presents public var customAlert: CustomAlertState<CustomAlertAction>?
 
@@ -199,10 +199,9 @@ extension LoginFeature {
             let role = login.role ?? .member
             state.$staffRole.withLock { $0 = login.isNewUser ? nil : role }
             state.$userSession.withLock { $0.userRole = role }
+            state.$editGeneration.withLock { $0 = false }
 
             if login.isNewUser  {
-              // 신규 가입: editGeneration 잔재(true)로 인한 editProfile 오분기 방지 → 회원가입(signUp) 강제
-              state.$editGeneration.withLock { $0 = false }
               return .send(.view(.showPolicyPopUp))
             } else if role == .manager {
               return .send(.delegate(.presentStaffMain))

Projects/Feature/Auth/Tests/Sources/LoginActionCoverageTests.swift

@@ -135,35 +135,38 @@ struct LoginActionCoverageTests {
   /// 기존 멤버 로그인은 역할을 저장하고 멤버 메인으로 이동한다.
   @Test("기존 멤버 로그인 성공은 역할 저장 후 멤버 메인으로 이동한다")
   func inner_loginResponse_existingMember_navigatesToMemberMain() async {
-    let store = Self.makeStore()
+    let store = Self.makeStore(configureState: { $0.editGeneration = true })
 
     await store.send(.inner(.loginResponse(.success(LoginTestFixture.member)))) {
       $0.staffRole = .member
       $0.userSession.userRole = .member
+      $0.editGeneration = false
     }
     await store.receive(\.delegate.presentMemberMain)
   }
 
   /// 운영진 로그인은 staffMain 으로 분기한다.
   @Test("운영진 로그인 성공은 운영진 메인으로 이동한다")
   func inner_loginResponse_manager_navigatesToStaffMain() async {
-    let store = Self.makeStore()
+    let store = Self.makeStore(configureState: { $0.editGeneration = true })
 
     await store.send(.inner(.loginResponse(.success(LoginTestFixture.manager)))) {
       $0.staffRole = .manager
       $0.userSession.userRole = .manager
+      $0.editGeneration = false
     }
     await store.receive(\.delegate.presentStaffMain)
   }
 
   /// role 이 nil 인 경계값은 member 로 기본 처리된다.
   @Test("role 이 nil 인 로그인 성공은 member 로 기본 처리된다")
   func inner_loginResponse_nilRole_defaultsToMember() async {
-    let store = Self.makeStore()
+    let store = Self.makeStore(configureState: { $0.editGeneration = true })
 
     await store.send(.inner(.loginResponse(.success(LoginTestFixture.roleless)))) {
       $0.staffRole = .member
       $0.userSession.userRole = .member
+      $0.editGeneration = false
     }
     await store.receive(\.delegate.presentMemberMain)
   }

Projects/Feature/OnBoarding/Sources/SelectManaging/Reducer/SelectManagingFeature.swift

@@ -35,7 +35,7 @@ public struct SelectManagingFeature {
     var managers: IdentifiedArrayOf<SelectManaging> = []
 
     @Shared(.userSession) var userSession
-    @Shared(.appStorage("editGeneration")) var editGeneration: Bool = false
+    @Shared(.isEditingGeneration) var editGeneration
     @Shared(.staffRole) var staffRole
     @Presents var alert: AlertState<AlertAction>?
   }
@@ -300,7 +300,6 @@ extension SelectManagingFeature {
         }
 
       case let .failure(error):
-        state.$editGeneration.withLock { $0 = false }
         state.alert = AlertState {
           TextState("프로필 수정 실패")
         } actions: {

Projects/Feature/OnBoarding/Sources/SelectTeam/Reducer/SelectTeamFeature.swift

@@ -39,7 +39,7 @@ public struct SelectTeamFeature {
 
 
     @Shared(.userSession) var userSession
-    @Shared(.appStorage("editGeneration")) var editGeneration: Bool = false
+    @Shared(.isEditingGeneration) var editGeneration
     @Shared(.staffRole) var staffRole
     @Presents var alert: AlertState<AlertAction>?
   }
@@ -322,7 +322,6 @@ extension SelectTeamFeature {
             }
 
           case .failure(let error):
-            state.$editGeneration.withLock { $0 = false }
             state.alert = AlertState {
               TextState("프로필 수정 실패")
             } actions: {

Projects/Feature/OnBoarding/Tests/Sources/SelectManagingReducerTests.swift

@@ -141,14 +141,18 @@ struct SelectManagingReducerTests {
   func editGenerationNavigatesToMember() async {
     let profile = OnBoardingCoverageFixture.memberProfile
     let authUseCase = MockAuthRepository.refreshSuccess()
-    let appStorage = UserDefaults.inMemory
     let inMemoryStorage = InMemoryStorage()
-    appStorage.set(true, forKey: "editGeneration")
+    let initialState = withDependencies {
+      $0.defaultInMemoryStorage = inMemoryStorage
+    } operation: {
+      var state = SelectManagingFeature.State()
+      state.$editGeneration.withLock { $0 = true }
+      return state
+    }
 
-    let store = TestStore(initialState: SelectManagingFeature.State()) {
+    let store = TestStore(initialState: initialState) {
       SelectManagingFeature()
     } withDependencies: {
-      $0.defaultAppStorage = appStorage
       $0.defaultInMemoryStorage = inMemoryStorage
       $0.profileUseCase = StubProfileUseCase(profile: profile)
       $0.authUseCase = authUseCase
@@ -176,14 +180,18 @@ struct SelectManagingReducerTests {
   @Test("기수 변경 후 토큰 갱신 실패 시 운영진 API를 호출하지 않고 로그인으로 이동한다")
   func credentialRefreshFailureNavigatesToLogin() async {
     let profile = OnBoardingCoverageFixture.managerProfile
-    let appStorage = UserDefaults.inMemory
     let inMemoryStorage = InMemoryStorage()
-    appStorage.set(true, forKey: "editGeneration")
+    let initialState = withDependencies {
+      $0.defaultInMemoryStorage = inMemoryStorage
+    } operation: {
+      var state = SelectManagingFeature.State()
+      state.$editGeneration.withLock { $0 = true }
+      return state
+    }
 
-    let store = TestStore(initialState: SelectManagingFeature.State()) {
+    let store = TestStore(initialState: initialState) {
       SelectManagingFeature()
     } withDependencies: {
-      $0.defaultAppStorage = appStorage
       $0.defaultInMemoryStorage = inMemoryStorage
       $0.profileUseCase = StubProfileUseCase(profile: profile)
       $0.authUseCase = MockAuthRepository.tokenExpired()
@@ -223,7 +231,7 @@ struct SelectManagingReducerTests {
     await store.receive(\.delegate.presentManager)
   }
 
-  @Test("프로필 수정 실패는 기수 변경 플래그를 내리고 실패 알럿을 표시한다")
+  @Test("프로필 수정 실패는 기수 변경 상태를 유지하고 실패 알럿을 표시한다")
   func editProfileFailurePresentsAlert() async {
     var state = SelectManagingFeature.State()
     state.editGeneration = true
@@ -235,7 +243,6 @@ struct SelectManagingReducerTests {
     let error = ProfileError.profileNotFound
 
     await store.send(.inner(.editProfileResponse(.failure(error)))) {
-      $0.editGeneration = false
       $0.alert = AlertState {
         TextState("프로필 수정 실패")
       } actions: {
@@ -248,6 +255,44 @@ struct SelectManagingReducerTests {
     }
   }
 
+  @Test("기수 변경 수정 실패 후 재시도는 회원가입 대신 프로필 수정을 다시 호출한다")
+  func retryAfterEditProfileFailureKeepsEditFlow() async {
+    var state = SelectManagingFeature.State()
+    state.$editGeneration.withLock { $0 = true }
+
+    let store = TestStore(initialState: state) {
+      SelectManagingFeature()
+    } withDependencies: {
+      $0.profileUseCase = StubProfileUseCase(editFailure: .profileUpdateFailed)
+      $0.signUpUseCase = StubSignUpUseCase(failure: .accountAlreadyExists)
+    }
+    let error = ProfileError.from(EditProfileError.profileUpdateFailed)
+
+    for _ in 0..<2 {
+      await store.send(.view(.signUp))
+      await store.receive {
+        guard case .async(.editProfile) = $0 else { return false }
+        return true
+      }
+      await store.receive(\.inner) {
+        $0.alert = AlertState {
+          TextState("프로필 수정 실패")
+        } actions: {
+          ButtonState(action: .confirmTapped) {
+            TextState("확인")
+          }
+        } message: {
+          TextState(error.errorDescription ?? "알 수 없는 오류가 발생했습니다.")
+        }
+      }
+      #expect(store.state.editGeneration)
+      await store.send(.scope(.alert(.dismiss))) {
+        $0.alert = nil
+      }
+    }
+    await store.finish()
+  }
+
   // MARK: - Delegate / Scope / Binding
 
   @Test("delegate 액션은 모두 부수효과 없이 소비된다")