diff --git a/includes/migrate/plugins/destinations/og_membership.inc b/includes/migrate/plugins/destinations/og_membership.inc index ffd3335..5244a12 100644 --- a/includes/migrate/plugins/destinations/og_membership.inc +++ b/includes/migrate/plugins/destinations/og_membership.inc @@ -8,7 +8,11 @@ /** * Destination class implementing migration into an OG Audience field. */ -class MigrateDestinationOGMembership extends MigrateDestination { +class MigrateDestinationOGMembership extends MigrateDestinationEntity { + + public function __construct($bundle = 'og_membership_type_default', array $options = array()) { + parent::__construct('og_membership', $bundle, $options); + } public function __toString() { return t('og-membership'); @@ -86,6 +90,7 @@ class MigrateDestinationOGMembership extends MigrateDestination { 'created' => isset($entity->created) ? $entity->created : REQUEST_TIME, ); + $this->prepare($entity, $row); $og_membership = og_group($entity->group_type, $entity->gid, $values); if ($entity->entity_type == 'user') { @@ -108,6 +113,7 @@ class MigrateDestinationOGMembership extends MigrateDestination { } } } + $this->complete($entity, $row); return array($og_membership->id); }