@@ -70,16 +70,19 @@ class AuthTokenDataStoreImplTest {
7070 fun `토큰 전체 업데이트가 성공하면 저장된 토큰을 반환해야 한다` () =
7171 runTest {
7272 // given
73+ val accessToken = " access"
74+ val refreshToken = " refresh"
7375 val token =
7476 AuthToken (
75- accessToken = " access " ,
76- refreshToken = " refresh " ,
77+ accessToken = accessToken ,
78+ refreshToken = refreshToken ,
7779 )
7880
7981 // when
80- val result = authTokenDataStore.updateAuthToken(token )
82+ authTokenDataStore.updateAuthToken(accessToken, refreshToken )
8183
8284 // then
85+ val result = authTokenDataStore.tokenFlow.first()
8386 assertEquals(token, result)
8487 }
8588
@@ -88,54 +91,51 @@ class AuthTokenDataStoreImplTest {
8891 runTest {
8992 // given
9093 authTokenDataStore.updateAuthToken(
91- AuthToken (
92- accessToken = " oldAccess" ,
93- refreshToken = " oldRefresh" ,
94- ),
94+ accessToken = " oldAccess" ,
95+ refreshToken = " oldRefresh" ,
9596 )
9697
9798 // when
98- val updated = authTokenDataStore.updateAccessToken(accessToken = " newAccess" )
99+ authTokenDataStore.updateAccessToken(accessToken = " newAccess" )
99100
100101 // then
101- assertEquals(" newAccess" , updated.accessToken)
102- assertEquals(" oldRefresh" , updated.refreshToken)
102+ val result = authTokenDataStore.tokenFlow.first()
103+ assertEquals(" newAccess" , result.accessToken)
104+ assertEquals(" oldRefresh" , result.refreshToken)
103105 }
104106
105107 @Test
106108 fun `refreshToken만 업데이트하면 기존 accessToken은 유지되어야 한다` () =
107109 runTest {
108110 // given
109111 authTokenDataStore.updateAuthToken(
110- AuthToken (
111- accessToken = " oldAccess" ,
112- refreshToken = " oldRefresh" ,
113- ),
112+ accessToken = " oldAccess" ,
113+ refreshToken = " oldRefresh" ,
114114 )
115115
116116 // when
117- val updated = authTokenDataStore.updateRefreshToken(refreshToken = " newRefresh" )
117+ authTokenDataStore.updateRefreshToken(refreshToken = " newRefresh" )
118118
119119 // then
120- assertEquals(" oldAccess" , updated.accessToken)
121- assertEquals(" newRefresh" , updated.refreshToken)
120+ val result = authTokenDataStore.tokenFlow.first()
121+ assertEquals(" oldAccess" , result.accessToken)
122+ assertEquals(" newRefresh" , result.refreshToken)
122123 }
123124
124125 @Test
125126 fun `토큰을 클리어하면 기본값이 저장되어야 한다` () =
126127 runTest {
127128 // given
128129 authTokenDataStore.updateAuthToken(
129- AuthToken (
130- accessToken = " someAccess" ,
131- refreshToken = " someRefresh" ,
132- ),
130+ accessToken = " someAccess" ,
131+ refreshToken = " someRefresh" ,
133132 )
134133
135134 // when
136- val cleared = authTokenDataStore.clearAuthToken()
135+ authTokenDataStore.clearAuthToken()
137136
138137 // then
138+ val cleared = authTokenDataStore.tokenFlow.first()
139139 assertEquals(AuthToken (), cleared)
140140 }
141141
@@ -150,7 +150,7 @@ class AuthTokenDataStoreImplTest {
150150 )
151151
152152 // when
153- authTokenDataStore.updateAuthToken(token )
153+ authTokenDataStore.updateAuthToken(" flowAccess " , " flowRefresh " )
154154
155155 // then
156156 val flowValue = authTokenDataStore.tokenFlow.first()
@@ -172,7 +172,7 @@ class AuthTokenDataStoreImplTest {
172172 val failingDataStore = AuthTokenDataStoreImpl (brokenStore)
173173
174174 // when & then
175- failingDataStore.updateAuthToken(AuthToken ( " access" , " refresh" ) )
175+ failingDataStore.updateAuthToken(" access" , " refresh" )
176176 }
177177
178178 @Test(expected = RuntimeException ::class )
0 commit comments