9
9
use Magento \Customer \ViewModel \Address \RegionProvider ;
10
10
use Magento \Framework \View \Element \Template ;
11
11
use Magento \TestFramework \Helper \Bootstrap ;
12
+ use Magento \Customer \ViewModel \CreateAccountButton ;
12
13
13
14
/**
14
15
* Test class for \Magento\Customer\Block\Form\Register
@@ -27,7 +28,8 @@ public function testCompanyDefault(): void
27
28
/** @var \Magento\Customer\Block\Widget\Company $block */
28
29
$ block = Bootstrap::getObjectManager ()->create (Register::class)
29
30
->setTemplate ('Magento_Customer::form/register.phtml ' )
30
- ->setShowAddressFields (true );
31
+ ->setShowAddressFields (true )
32
+ ->setCreateAccountButtonViewModel (Bootstrap::getObjectManager ()->create (CreateAccountButton::class));
31
33
$ this ->setAttributeDataProvider ($ block );
32
34
$ this ->setRegionProvider ($ block );
33
35
@@ -42,10 +44,10 @@ public function testCompanyDefault(): void
42
44
public function testTelephoneDefault (): void
43
45
{
44
46
/** @var \Magento\Customer\Block\Widget\Company $block */
45
- $ block = Bootstrap::getObjectManager ()->create (
46
- Register::class
47
- )-> setTemplate ( ' Magento_Customer::form/register.phtml ' )
48
- -> setShowAddressFields ( true );
47
+ $ block = Bootstrap::getObjectManager ()->create (Register::class)
48
+ -> setTemplate ( ' Magento_Customer::form/register.phtml ' )
49
+ -> setShowAddressFields ( true )
50
+ -> setCreateAccountButtonViewModel (Bootstrap:: getObjectManager ()-> create (CreateAccountButton::class) );
49
51
$ this ->setAttributeDataProvider ($ block );
50
52
$ this ->setRegionProvider ($ block );
51
53
@@ -60,10 +62,10 @@ public function testTelephoneDefault(): void
60
62
public function testFaxDefault (): void
61
63
{
62
64
/** @var \Magento\Customer\Block\Widget\Company $block */
63
- $ block = Bootstrap::getObjectManager ()->create (
64
- Register::class
65
- )-> setTemplate ( ' Magento_Customer::form/register.phtml ' )
66
- -> setShowAddressFields ( true );
65
+ $ block = Bootstrap::getObjectManager ()->create (Register::class)
66
+ -> setTemplate ( ' Magento_Customer::form/register.phtml ' )
67
+ -> setShowAddressFields ( true )
68
+ -> setCreateAccountButtonViewModel (Bootstrap:: getObjectManager ()-> create (CreateAccountButton::class) );
67
69
$ this ->setAttributeDataProvider ($ block );
68
70
$ this ->setRegionProvider ($ block );
69
71
@@ -85,10 +87,10 @@ public function testCompanyDisabled(): void
85
87
$ model ->save ();
86
88
87
89
/** @var \Magento\Customer\Block\Widget\Company $block */
88
- $ block = Bootstrap::getObjectManager ()->create (
89
- Register::class
90
- )-> setTemplate ( ' Magento_Customer::form/register.phtml ' )
91
- -> setShowAddressFields ( true );
90
+ $ block = Bootstrap::getObjectManager ()->create (Register::class)
91
+ -> setTemplate ( ' Magento_Customer::form/register.phtml ' )
92
+ -> setShowAddressFields ( true )
93
+ -> setCreateAccountButtonViewModel (Bootstrap:: getObjectManager ()-> create (CreateAccountButton::class) );
92
94
$ this ->setAttributeDataProvider ($ block );
93
95
$ this ->setRegionProvider ($ block );
94
96
@@ -110,10 +112,10 @@ public function testTelephoneDisabled(): void
110
112
$ model ->save ();
111
113
112
114
/** @var \Magento\Customer\Block\Widget\Company $block */
113
- $ block = Bootstrap::getObjectManager ()->create (
114
- Register::class
115
- )-> setTemplate ( ' Magento_Customer::form/register.phtml ' )
116
- -> setShowAddressFields ( true );
115
+ $ block = Bootstrap::getObjectManager ()->create (Register::class)
116
+ -> setTemplate ( ' Magento_Customer::form/register.phtml ' )
117
+ -> setShowAddressFields ( true )
118
+ -> setCreateAccountButtonViewModel (Bootstrap:: getObjectManager ()-> create (CreateAccountButton::class) );
117
119
$ this ->setAttributeDataProvider ($ block );
118
120
$ this ->setRegionProvider ($ block );
119
121
@@ -135,10 +137,10 @@ public function testFaxEnabled(): void
135
137
$ model ->save ();
136
138
137
139
/** @var \Magento\Customer\Block\Widget\Company $block */
138
- $ block = Bootstrap::getObjectManager ()->create (
139
- Register::class
140
- )-> setTemplate ( ' Magento_Customer::form/register.phtml ' )
141
- -> setShowAddressFields ( true );
140
+ $ block = Bootstrap::getObjectManager ()->create (Register::class)
141
+ -> setTemplate ( ' Magento_Customer::form/register.phtml ' )
142
+ -> setShowAddressFields ( true )
143
+ -> setCreateAccountButtonViewModel (Bootstrap:: getObjectManager ()-> create (CreateAccountButton::class) );
142
144
$ this ->setAttributeDataProvider ($ block );
143
145
$ this ->setRegionProvider ($ block );
144
146
@@ -151,10 +153,10 @@ public function testFaxEnabled(): void
151
153
public function testCityWithStoreLabel (): void
152
154
{
153
155
/** @var \Magento\Customer\Block\Form\Register $block */
154
- $ block = Bootstrap::getObjectManager ()->create (
155
- Register::class
156
- )-> setTemplate ( ' Magento_Customer::form/register.phtml ' )
157
- ->setShowAddressFields ( true );
156
+ $ block = Bootstrap::getObjectManager ()->create (Register::class)
157
+ -> setTemplate ( ' Magento_Customer::form/register.phtml ' )
158
+ -> setShowAddressFields ( true )
159
+ ->setCreateAccountButtonViewModel (Bootstrap:: getObjectManager ()-> create (CreateAccountButton::class) );
158
160
$ this ->setAttributeDataProvider ($ block );
159
161
$ this ->setRegionProvider ($ block );
160
162
0 commit comments