File size: 2,422 Bytes
d2897cd
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
<?php

declare(strict_types=1);

namespace Mautic\LeadBundle\Tests\Model;

use Mautic\CategoryBundle\Entity\Category;
use Mautic\CoreBundle\Test\MauticMysqlTestCase;
use Mautic\CoreBundle\Tests\Functional\CreateTestEntitiesTrait;
use Mautic\LeadBundle\Model\LeadModel;

final class SetFrequencyRulesFunctionalTest extends MauticMysqlTestCase
{
    use CreateTestEntitiesTrait;

    public function testSetFrequencyRulesForCategorySubscriptionUnsubscription(): void
    {
        $categoriesFlags = [
            'one'   => true,
            'two'   => false,
            'three' => true,
            'four'  => true,
            'five'  => false,
        ];

        $categories = $this->createCategories($categoriesFlags);

        $lead = $this->createLead('John', 'Doe', '[email protected]');

        $this->em->flush();

        // Subscribe categories.
        $categoriesToSubscribe   = [];
        $categoriesToUnsubscribe = [];
        foreach ($categories as $category) {
            $categoriesToSubscribe[$category->getId()] = $category->getId();
            if (!$categoriesFlags[$category->getTitle()]) {
                $categoriesToUnsubscribe[$category->getId()] = $category->getId();
            }
        }

        $data = [
            'global_categories' => array_keys($categoriesToSubscribe),
            'lead_lists'        => [],
        ];

        /** @var LeadModel $model */
        $model = static::getContainer()->get('mautic.lead.model.lead');
        $model->setFrequencyRules($lead, $data, [], []);

        $subscribedCategories   = $model->getLeadCategories($lead);
        $this->assertEmpty(array_diff($subscribedCategories, array_keys($categoriesToSubscribe)));

        // Unsubscribe categories.
        $data['global_categories'] = array_keys($categoriesToUnsubscribe);
        $model->setFrequencyRules($lead, $data, [], []);

        $unsubscribedCategories = $model->getUnsubscribedLeadCategoriesIds($lead);
        $this->assertEmpty(array_diff($unsubscribedCategories, array_keys($categoriesToSubscribe)));
    }

    /**
     * @param mixed[] $cats
     *
     * @return Category[]
     */
    private function createCategories(array $cats): array
    {
        $categories = [];
        foreach ($cats as $suffix => $flag) {
            $categories[$suffix] = $this->createCategory($suffix, $suffix);
        }

        $this->em->flush();

        return $categories;
    }
}