Fix/profile/solve editprofile bug

app/src/main/java/com/android/periodpals/ui/components/ProfileComponents.kt

@@ -220,7 +220,7 @@ fun ProfileSaveButton(
     descriptionState: TextFieldState,
     profileImageState: TextFieldState,
     byteArray: ByteArray?,
-    preferredDistance: Int,
+    preferredDistance: Int?,
     context: Context,
     userViewModel: UserViewModel,
     navigationActions: NavigationActions,
@@ -241,6 +241,11 @@ fun ProfileSaveButton(
           Toast.makeText(context, errorMessage, Toast.LENGTH_SHORT).show()
           return@Button
         }
+        if (preferredDistance == null) {
+          Log.d(LOG_TAG, LOG_FAILURE)
+          Toast.makeText(context, TOAST_FAILURE, Toast.LENGTH_SHORT).show()
+          return@Button
+        }
 
         Log.d(LOG_TAG, LOG_SAVING_PROFILE)
         val newUser =

app/src/main/java/com/android/periodpals/ui/profile/EditProfile.kt

@@ -72,23 +72,27 @@ fun EditProfileScreen(userViewModel: UserViewModel, navigationActions: Navigatio
   formState.reset()
 
   val nameState = formState.getState<TextFieldState>(UserViewModel.NAME_STATE_NAME)
-  nameState.change(userState.value?.name ?: "")
   val dobState = formState.getState<TextFieldState>(UserViewModel.DOB_STATE_NAME)
-  dobState.change(userState.value?.dob ?: "")
   val descriptionState = formState.getState<TextFieldState>(UserViewModel.DESCRIPTION_STATE_NAME)
-  descriptionState.change(userState.value?.description ?: "")
   val profileImageState = formState.getState<TextFieldState>(UserViewModel.PROFILE_IMAGE_STATE_NAME)
-  profileImageState.change(userState.value?.imageUrl ?: DEFAULT_PROFILE_PICTURE)
+
+  userState.value?.let {
+    nameState.change(it.name)
+    dobState.change(it.dob)
+    descriptionState.change(it.description)
+    profileImageState.change(it.imageUrl)
+  }
+
   var userAvatarState by remember {
-    mutableStateOf(userAvatar?.value ?: Uri.parse(DEFAULT_PROFILE_PICTURE).uriToByteArray(context))
+    mutableStateOf(userAvatar.value ?: Uri.parse(DEFAULT_PROFILE_PICTURE).uriToByteArray(context))
   }
 
   val launcher =
       rememberLauncherForActivityResult(
           contract = ActivityResultContracts.StartActivityForResult()) { result ->
-            if (result.resultCode == Activity.RESULT_OK) {
-              profileImageState.change(result.data?.data.toString())
-              userAvatarState = result.data?.data?.uriToByteArray(context)
+            if ((result.resultCode == Activity.RESULT_OK) && (result.data != null)) {
+              profileImageState.change(result.data!!.data.toString())
+              userAvatarState = result.data!!.data?.uriToByteArray(context)
             }
           }
 
@@ -165,7 +169,7 @@ fun EditProfileScreen(userViewModel: UserViewModel, navigationActions: Navigatio
           descriptionState,
           profileImageState,
           userAvatarState,
-          userViewModel.user.value!!.preferredDistance,
+          userViewModel.user.value?.preferredDistance,
           context,
           userViewModel,
           navigationActions,

app/src/main/java/com/android/periodpals/ui/profile/ProfileScreen.kt

@@ -245,17 +245,17 @@ fun init(
       onSuccess = {
         Log.d(TAG, "Authentication data loaded successfully")
         chatViewModel.connectUser(userState, authenticationViewModel = authenticationViewModel)
+        userViewModel.loadUser(
+            authenticationViewModel.authUserData.value!!.uid,
+            onSuccess = {
+              userViewModel.user.value?.let {
+                userViewModel.downloadFile(
+                    it.imageUrl,
+                    onSuccess = { onSuccess() },
+                    onFailure = { e: Exception -> onFailure(Exception(e)) })
+              }
+            },
+            onFailure = { e: Exception -> onFailure(Exception(e)) })
       },
       onFailure = { Log.d(TAG, "Authentication data is null") })
-  userViewModel.loadUser(
-      authenticationViewModel.authUserData.value!!.uid,
-      onSuccess = {
-        userViewModel.user.value?.let {
-          userViewModel.downloadFile(
-              it.imageUrl,
-              onSuccess = { onSuccess() },
-              onFailure = { e: Exception -> onFailure(Exception(e)) })
-        }
-      },
-      onFailure = { e: Exception -> onFailure(Exception(e)) })
 }

app/src/test/java/com/android/periodpals/ui/profile/EditProfileTest.kt

@@ -118,6 +118,7 @@ class EditProfileTest {
 
     `when`(navigationActions.currentRoute()).thenReturn(Route.PROFILE)
     `when`(userViewModel.formState).thenReturn(formState)
+    `when`(userViewModel.avatar).thenReturn(mutableStateOf(ByteArray(0)))
   }
 
   @Test