8
8
namespace Magento \SalesGraphQl \Test \Unit \Model \Resolver ;
9
9
10
10
use Magento \Framework \GraphQl \Query \Resolver \ContextInterface ;
11
- use Magento \Sales \Api \Data \ OrderExtensionInterface ;
11
+ use Magento \Framework \Api \ExtensionAttributesInterface ;
12
12
use PHPUnit \Framework \MockObject \MockObject ;
13
13
use PHPUnit \Framework \TestCase ;
14
14
use Magento \SalesGraphQl \Model \Resolver \OrderTotal ;
15
15
use Magento \Sales \Api \Data \OrderInterface ;
16
- use Magento \Quote \Model \QuoteFactory ;
17
16
use Magento \Framework \GraphQl \Config \Element \Field ;
18
17
use Magento \Framework \GraphQl \Schema \Type \ResolveInfo ;
19
18
use Magento \Framework \Exception \LocalizedException ;
@@ -46,7 +45,7 @@ class OrderTotalTest extends TestCase
46
45
private $ resolveInfoMock ;
47
46
48
47
/**
49
- * @var OrderExtensionInterface |MockObject
48
+ * @var ExtensionAttributesInterface |MockObject
50
49
*/
51
50
private $ extensionAttributesMock ;
52
51
@@ -55,7 +54,6 @@ protected function setUp(): void
55
54
$ this ->contextMock = $ this ->createMock (ContextInterface::class);
56
55
$ this ->fieldMock = $ this ->createMock (Field::class);
57
56
$ this ->resolveInfoMock = $ this ->createMock (ResolveInfo::class);
58
- $ this ->extensionAttributesMock = $ this ->createMock (OrderExtensionInterface::class);
59
57
$ this ->orderMock = $ this ->createMock (OrderInterface::class);
60
58
$ this ->orderMock ->method ('getOrderCurrencyCode ' )->willReturn ('USD ' );
61
59
$ this ->orderMock ->method ('getBaseCurrencyCode ' )->willReturn ('USD ' );
@@ -76,12 +74,12 @@ public function testResolve(): void
76
74
$ resolveInfoMock = $ this ->createMock (ResolveInfo::class);
77
75
$ value = ['model ' => $ this ->orderMock ];
78
76
$ args = [];
79
- $ this ->extensionAttributesMock -> expects ( $ this ->once () )
80
- ->method ( 'getAppliedTaxes ' )
81
- ->willReturn ([]);
82
- $ this -> extensionAttributesMock -> expects ( $ this -> once ())
83
- -> method ('getItemAppliedTaxes ' )
84
- ->willReturn ([]);
77
+ $ this ->extensionAttributesMock = $ this ->getMockBuilder (ExtensionAttributesInterface::class )
78
+ ->addMethods ([ 'getAppliedTaxes ' , ' getItemAppliedTaxes ' ] )
79
+ ->disableOriginalConstructor ()
80
+ -> getMock ();
81
+ $ this -> extensionAttributesMock -> expects ( $ this -> once ())-> method ('getAppliedTaxes ' )-> willReturn ([]);
82
+ $ this -> extensionAttributesMock -> expects ( $ this -> once ())-> method ( ' getItemAppliedTaxes ' ) ->willReturn ([]);
85
83
$ this ->orderMock ->method ('getExtensionAttributes ' )->willReturn ($ this ->extensionAttributesMock );
86
84
$ result = $ this ->orderTotal ->resolve ($ fieldMock , $ this ->contextMock , $ resolveInfoMock , $ value , $ args );
87
85
$ this ->assertArrayHasKey ('base_grand_total ' , $ result );
0 commit comments