entityManager = ContainerWrapper::getInstance()->get(EntityManager::class); } public function createWordPressUser(string $email, string $role) { $userId = wp_insert_user([ 'user_login' => explode('@', $email)[0], 'user_email' => $email, 'role' => $role, 'user_pass' => '12123154', ]); if ($userId instanceof WP_Error) { throw new Exception(sprintf("Unable to create WordPress user with email $email: %s", $userId->get_error_message())); } $this->createdUsers[] = $email; return $userId; } public function createCustomer(string $email, string $role = 'customer'): int { return $this->createWordPressUser($email, $role); } public function deleteCreatedUsers() { foreach ($this->createdUsers as $createdUserEmail) { $this->deleteWordPressUser($createdUserEmail); } $this->createdUsers = []; } public function deleteWordPressUser(string $email) { $user = get_user_by('email', $email); if (!$user) { return; } if (is_multisite()) { wpmu_delete_user($user->ID); } else { wp_delete_user($user->ID); } } public function createWooCommerceOrder(array $data = []): \WC_Order { $helper = ContainerWrapper::getInstance()->get(Helper::class); $order = $helper->wcCreateOrder([]); if (isset($data['date_created'])) { $order->set_date_created($data['date_created']); } if (isset($data['billing_email'])) { $order->set_billing_email($data['billing_email']); } $order->save(); $this->wooOrderIds[] = $order->get_id(); return $order; } public function updateWooOrderStats(int $orderId): void { if (!class_exists('Automattic\WooCommerce\Admin\API\Reports\Orders\Stats\DataStore')) { return; } DataStore::sync_order($orderId); } public function deleteTestWooOrder(int $wooOrderId) { $helper = ContainerWrapper::getInstance()->get(Helper::class); $order = $helper->wcGetOrder($wooOrderId); if ($order instanceof \WC_Order) { $order->delete(true); } } public function deleteTestWooOrders() { $helper = ContainerWrapper::getInstance()->get(Helper::class); foreach ($this->wooOrderIds as $wooOrderId) { $order = $helper->wcGetOrder($wooOrderId); if ($order instanceof \WC_Order) { $order->delete(true); } } $this->wooOrderIds = []; } public function uniqueId($length = 10): string { return Security::generateRandomString($length); } /** * Compares two DateTimeInterface objects by comparing timestamp values. * $delta parameter specifies tolerated difference */ public function assertEqualDateTimes(?DateTimeInterface $date1, ?DateTimeInterface $date2, int $delta = 0) { if (!$date1 instanceof DateTimeInterface) { throw new \Exception('$date1 is not DateTimeInterface'); } if (!$date2 instanceof DateTimeInterface) { throw new \Exception('$date2 is not DateTimeInterface'); } expect($date1->getTimestamp())->equals($date2->getTimestamp(), $delta); } public function createAutomation(string $name, Step ...$steps): ?Automation { $automationStorage = ContainerWrapper::getInstance()->get(AutomationStorage::class); if (!$steps) { $steps[] = new Step('trigger', Step::TYPE_TRIGGER, \MailPoet\Automation\Integrations\MailPoet\Triggers\SomeoneSubscribesTrigger::KEY, [], []); } //If we only have a trigger, add a delay step to make the automation valid. if (count($steps) === 1) { $delay = ContainerWrapper::getInstance()->get(DelayAction::class); $delayStep = new Step('delay', Step::TYPE_ACTION, $delay->getKey(), [], []); $steps[0]->setNextSteps([new NextStep($delayStep->getId())]); $steps[] = $delayStep; } $steps = array_merge( [ 'root' => new Step('root', Step::TYPE_ROOT, 'root', [], [new NextStep($steps[0]->getId())]), ], $steps ); $stepsWithIds = []; foreach ($steps as $step) { $stepsWithIds[$step->getId()] = $step; } $automation = new Automation($name, $stepsWithIds, wp_get_current_user()); $automation->setStatus(Automation::STATUS_ACTIVE); return $automationStorage->getAutomation($automationStorage->createAutomation($automation)); } public function createAutomationRun(Automation $automation, $subjects = []): ?AutomationRun { $trigger = array_filter($automation->getSteps(), function(Step $step): bool { return $step->getType() === Step::TYPE_TRIGGER; }); $triggerKeys = array_map(function(Step $step): string { return $step->getKey(); }, $trigger); $triggerKey = count($triggerKeys) > 0 ? current($triggerKeys) : ''; $automationRun = new AutomationRun( $automation->getId(), $automation->getVersionId(), $triggerKey, $subjects ); $automationRunStorage = ContainerWrapper::getInstance()->get(AutomationRunStorage::class); return $automationRunStorage->getAutomationRun($automationRunStorage->createAutomationRun($automationRun)); } public function getSubscriberEmailsMatchingDynamicFilter(DynamicSegmentFilterData $data, Filter $filter): array { $segment = new SegmentEntity('temporary segment', SegmentEntity::TYPE_DYNAMIC, 'description'); $this->entityManager->persist($segment); $filterEntity = new DynamicSegmentFilterEntity($segment, $data); $this->entityManager->persist($filterEntity); $segment->addDynamicFilter($filterEntity); $queryBuilder = $filter->apply($this->getSubscribersQueryBuilder(), $filterEntity); return $this->getSubscriberEmailsFromQueryBuilder($queryBuilder); } /** * @param QueryBuilder $queryBuilder * @return string[] - array of subscriber emails */ public function getSubscriberEmailsFromQueryBuilder(QueryBuilder $queryBuilder): array { $statement = $queryBuilder->execute(); $results = $statement instanceof Statement ? $statement->fetchAllAssociative() : []; return array_map(function($row) { $subscriber = $this->entityManager->find(SubscriberEntity::class, $row['inner_subscriber_id']); if (!$subscriber instanceof SubscriberEntity) { throw new \Exception('this is for PhpStan'); } return $subscriber->getEmail(); }, $results); } public function getSubscribersQueryBuilder(): QueryBuilder { $subscribersTable = $this->entityManager->getClassMetadata(SubscriberEntity::class)->getTableName(); return $this->entityManager ->getConnection() ->createQueryBuilder() ->select("DISTINCT $subscribersTable.id as inner_subscriber_id") ->from($subscribersTable); } public function cleanup() { $this->deleteCreatedUsers(); $this->deleteTestWooOrders(); } }