1 files changed, 32 insertions, 15 deletions
diff --git a/src/screens/badge/BadgeSelection.tsx b/src/screens/badge/BadgeSelection.tsx
index 38a2b01c..d0dcfa4c 100644
--- a/src/screens/badge/BadgeSelection.tsx
+++ b/src/screens/badge/BadgeSelection.tsx
@@ -8,14 +8,20 @@ import LinearGradient from 'react-native-linear-gradient';
 import {useDispatch, useSelector} from 'react-redux';
 import {BACKGROUND_GRADIENT_MAP} from '../../constants';
 import {BADGE_DATA} from '../../constants/badges';
-import {ERROR_BADGES_EXCEED_LIMIT} from '../../constants/strings';
+import {
+  ERROR_BADGES_EXCEED_LIMIT,
+  SUCCESS_BADGES_UPDATE,
+} from '../../constants/strings';
 import {MainStackParams} from '../../routes';
 import {
   addBadgesService,
-  getSuggestedPeopleProfile,
+  getBadgesService,
   updateBadgesService,
 } from '../../services';
-import {suggestedPeopleBadgesFinished} from '../../store/actions';
+import {
+  suggestedPeopleBadgesFinished,
+  updateUserBadges,
+} from '../../store/actions';
 import {RootState} from '../../store/rootReducer';
 import {BackgroundGradientType} from '../../types';
 import {SCREEN_HEIGHT, StatusBarHeight} from '../../utils';
@@ -36,37 +42,47 @@ const BadgeSelection: React.FC<BadgeSelectionProps> = ({route}) => {
   const {editing} = route.params;
   const {
     user: {userId: loggedInUserId},
-    profile: {university},
+    profile: {university, badges},
   } = useSelector((state: RootState) => state.user);
   const [selectedBadges, setSelectedBadges] = useState<string[]>([]);
   const dispatch = useDispatch();
   const navigation = useNavigation();
 
-  // Loading badges data and extracting into a string []
+  // Extracting badges data into a string []
   useEffect(() => {
     const loadData = async () => {
-      const response = await getSuggestedPeopleProfile(loggedInUserId);
-      if (response) {
-        const data = response.badges;
-        let extractedBadgeNames: string[] = [];
-        data.forEach((badge) => {
-          extractedBadgeNames.push(badge.name);
-        });
-        setSelectedBadges(extractedBadgeNames);
-      }
+      let extractedBadgeNames: string[] = [];
+      badges.forEach((badge) => {
+        extractedBadgeNames.push(badge.name);
+      });
+      setSelectedBadges(extractedBadgeNames);
     };
     if (editing) {
       loadData();
     }
   }, []);
 
+  // Retrieve updated badges using get badges service and udpate the store
+  const loadUserBadges = async () => {
+    const newBadges = await getBadgesService(loggedInUserId);
+    dispatch(updateUserBadges(newBadges));
+  };
+
   navigation.setOptions({
     headerRight: () => (
       <TouchableOpacity
         style={styles.rightButtonContainer}
         onPress={async () => {
           if (editing) {
-            await updateBadgesService(selectedBadges, university);
+            const success = await updateBadgesService(
+              selectedBadges,
+              university,
+            );
+            if (success === true) {
+              // Load updated badges to store
+              loadUserBadges();
+              Alert.alert(SUCCESS_BADGES_UPDATE);
+            }
             if (navigation.canGoBack()) {
               navigation.goBack();
             } else {
@@ -82,6 +98,7 @@ const BadgeSelection: React.FC<BadgeSelectionProps> = ({route}) => {
               );
               if (success) {
                 dispatch(suggestedPeopleBadgesFinished());
+                loadUserBadges();
                 navigation.navigate('SuggestedPeople');
               }
             } else {