File size: 2,531 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
<?php

namespace Mautic\LeadBundle\Tests\Helper;

use Mautic\LeadBundle\Entity\CompanyLeadRepository;
use Mautic\LeadBundle\Entity\Lead;
use Mautic\LeadBundle\Helper\PrimaryCompanyHelper;
use PHPUnit\Framework\Exception;

class PrimaryCompanyHelperTest extends \PHPUnit\Framework\TestCase
{
    /**
     * @var CompanyLeadRepository|Exception
     */
    private \PHPUnit\Framework\MockObject\MockObject $leadRepository;

    protected function setUp(): void
    {
        $this->leadRepository = $this->createMock(CompanyLeadRepository::class);

        $this->leadRepository->expects($this->once())
            ->method('getCompaniesByLeadId')
            ->willReturn(
                [
                    [
                        'score'           => 0,
                        'date_added'      => '2018-06-02 00:00:00',
                        'date_associated' => '2018-06-02 00:00:00',
                        'is_primary'      => 1,
                        'companywebsite'  => 'https://foo.com',
                    ],
                    [
                        'score'           => 0,
                        'date_added'      => '2018-06-02 00:00:00',
                        'date_associated' => '2018-06-02 00:00:00',
                        'is_primary'      => 0,
                        'companywebsite'  => 'https://bar.com',
                    ],
                ]
            );
    }

    public function testProfileFieldsReturnedWithPrimaryCompany(): void
    {
        $lead = $this->createMock(Lead::class);
        $lead->expects($this->once())
            ->method('getProfileFields')
            ->willReturn(
                [
                    'email' => '[email protected]',
                ]
            );

        $profileFields = $this->getPrimaryCompanyHelper()->getProfileFieldsWithPrimaryCompany($lead);

        $this->assertEquals(['email' => '[email protected]', 'companywebsite' => 'https://foo.com'], $profileFields);
    }

    public function testPrimaryCompanyMergedIntoProfileFields(): void
    {
        $leadFields = [
            'email' => '[email protected]',
        ];

        $profileFields = $this->getPrimaryCompanyHelper()->mergePrimaryCompanyWithProfileFields(1, $leadFields);

        $this->assertEquals(['email' => '[email protected]', 'companywebsite' => 'https://foo.com'], $profileFields);
    }

    /**
     * @return PrimaryCompanyHelper
     */
    private function getPrimaryCompanyHelper()
    {
        return new PrimaryCompanyHelper($this->leadRepository);
    }
}