@@ -66,17 +66,17 @@ protected function tearDown()
6666 */
6767 public function testCoreAttributes ($ storeId , $ productData , $ expectedHasSpecialPrice )
6868 {
69- $ extensionAttributesStub = $ this ->getMockBuilder (ProductExtensionInterface ::class)
70- ->setMethods (['getSolrBoost ' ])
69+ $ customAttributeStub = $ this ->getMockBuilder (\ Magento \ Framework \ Api \AttributeInterface ::class)
70+ ->setMethods (['getValue ' ])
7171 ->getMockForAbstractClass ();
7272
7373 $ this ->magentoProductStub ->method ('getId ' )->willReturn ($ productData ['id ' ]);
7474 $ this ->magentoProductStub ->method ('getPrice ' )->willReturn ($ productData ['price ' ]);
7575 $ finalPrice = isset ($ productData ['special_price ' ]) ? $ productData ['special_price ' ] : $ productData ['price ' ];
7676 $ this ->magentoProductStub ->method ('getFinalPrice ' )->willReturn ($ finalPrice );
7777 $ this ->magentoProductStub ->method ('getStoreId ' )->willReturn ($ storeId );
78- $ this ->magentoProductStub ->method ('getExtensionAttributes ' )->willReturn ($ extensionAttributesStub );
79- $ extensionAttributesStub ->method ('getSolrBoost ' )->willReturn ($ productData ['solr_boost ' ]);
78+ $ this ->magentoProductStub ->method ('getCustomAttribute ' )->willReturn ($ customAttributeStub );
79+ $ customAttributeStub ->method ('getValue ' )->willReturn ($ productData ['solr_boost ' ]);
8080 $ this ->magentoProductStub ->method ('getCategoryIds ' )->willReturn ($ productData ['category_ids ' ]);
8181 $ productBridge = $ this ->makeProductBridge ($ storeId );
8282
@@ -176,16 +176,16 @@ public function testIndexable($storeAndWebsiteId, $status, $visibility, $website
176176 ->getMockForAbstractClass ();
177177 $ storeStub ->method ('getWebsiteId ' )->willReturn ($ storeAndWebsiteId );
178178
179- $ extensionAttributesStub = $ this ->getMockBuilder (ProductExtensionInterface ::class)
180- ->setMethods (['getStockItem ' , ' getSolrExclude ' ])
179+ $ customAttributeStub = $ this ->getMockBuilder (\ Magento \ Framework \ Api \AttributeInterface ::class)
180+ ->setMethods (['getValue ' ])
181181 ->getMockForAbstractClass ();
182- $ extensionAttributesStub ->method ('getSolrExclude ' )->willReturn ($ solrExclude );
182+ $ customAttributeStub ->method ('getValue ' )->willReturn ($ solrExclude );
183183
184184 $ this ->magentoProductStub ->method ('getStatus ' )->willReturn ($ status );
185185 $ this ->magentoProductStub ->method ('getVisibility ' )->willReturn ($ visibility );
186186 $ this ->magentoProductStub ->method ('getStore ' )->willReturn ($ storeStub );
187187 $ this ->magentoProductStub ->method ('getWebsiteIds ' )->willReturn ($ websiteIds );
188- $ this ->magentoProductStub ->method ('getExtensionAttributes ' )->willReturn ($ extensionAttributesStub );
188+ $ this ->magentoProductStub ->method ('getCustomAttribute ' )->willReturn ($ customAttributeStub );
189189 $ this ->magentoProductStub ->method ('getData ' )->with ('is_salable ' )->willReturn ($ inStock );
190190
191191 $ productBridge = $ this ->makeProductBridge ($ storeAndWebsiteId );
0 commit comments