Add Tests For Social Post

social_post.post_update.php

@@ -0,0 +1,20 @@
+<?php
+
+/**
+ * Update encryption tokens in Social Post.
+ */
+function social_post_post_update_add_encryption_to_access_tokens(&$sandbox = NULL) {
+	$entity = \Drupal::entityTypeManager()
+	  ->getStorage('social_post');
+
+	foreach($entity as $user) {
+		$token = $user->get('token');
+		$user->setToken($token);
+		$user->save();
+		$result = t('Token %nid saved', [
+    	'%nid' => $user
+    	  ->id(),
+  		]);
+	}
+
+}

src/Entity/SocialPost.php

@@ -7,6 +7,7 @@
 use Drupal\Core\Entity\EntityTypeInterface;
 use Drupal\Core\Entity\ContentEntityInterface;
 use Drupal\Core\Entity\EntityChangedTrait;
+use Drupal\social_api\Entity\SocialApi;
 
 /**
  * Defines the Social Post entity.
@@ -37,7 +38,7 @@
  *   }
  * )
  */
-class SocialPost extends ContentEntityBase implements ContentEntityInterface {
+class SocialPost extends SocialApi implements ContentEntityInterface {
   use EntityChangedTrait;
 
   /**

src/SocialPostManager.php

@@ -183,6 +183,7 @@ public function addRecord($name, $provider_user_id, $token, $additional_data = N
     ];
 
     $user_info = $this->entityTypeManager->getStorage('social_post')->create($values);
+    $user_info->setToken($token);
 
     // Save the entity.
     $user_info->save();
@@ -269,14 +270,7 @@ public function getToken($provider_user_id) {
         ])
     );
 
-    if (!$user_data) {
-      return FALSE;
-    }
-    else {
-      // Get token and decrypt it.
-      return $this->decryptToken($user_data->get('token')->getValue()[0]['value']);
-    }
-
+    return $this->decryptToken($user_data->get('token')->getValue()[0]['value']);
   }
 
   /**