public function testCustomerLoggedIn()
{
$this->moduleManagerMock->expects($this->once())->method('isEnabled')->with('Magento_PageCache')->willReturn(true);
$this->cacheConfigMock->expects($this->once())->method('isEnabled')->willReturn(true);
$this->taxHelperMock->expects($this->any())->method('isCatalogPriceDisplayAffectedByTax')->willReturn(true);
$customerMock = $this->getMockBuilder('Magento\\Customer\\Model\\Data\\Customer')->disableOriginalConstructor()->getMock();
$this->observerMock->expects($this->once())->method('getData')->with('customer')->willReturn($customerMock);
$customerMock->expects($this->once())->method('getGroupId')->willReturn(1);
$customerGroupMock = $this->getMockBuilder('Magento\\Customer\\Model\\Data\\Group')->disableOriginalConstructor()->getMock();
$this->groupRepositoryMock->expects($this->once())->method('getById')->with(1)->willReturn($customerGroupMock);
$customerGroupMock->expects($this->once())->method('getTaxClassId')->willReturn(1);
$this->customerSessionMock->expects($this->once())->method('setCustomerTaxClassId')->with(1);
$address = $this->objectManager->getObject('Magento\\Customer\\Model\\Data\\Address');
$address->setIsDefaultShipping(true);
$address->setIsDefaultBilling(true);
$address->setCountryId(1);
$address->setPostCode(11111);
$addresses = [$address];
$customerMock->expects($this->once())->method('getAddresses')->willReturn($addresses);
$this->customerSessionMock->expects($this->once())->method('setDefaultTaxBillingAddress')->with(['country_id' => 1, 'region_id' => null, 'postcode' => 11111]);
$this->customerSessionMock->expects($this->once())->method('setDefaultTaxShippingAddress')->with(['country_id' => 1, 'region_id' => null, 'postcode' => 11111]);
$this->session->customerLoggedIn($this->observerMock);
}
