/**
  * {@inheritdoc}
  */
 public function retrieve(EntityMetadata $metadata, array $entityData)
 {
     $contextFields = $metadata->getEntityContext();
     $context = [];
     if (isset($contextFields[\Magento\Store\Model\Store::STORE_ID])) {
         $context[\Magento\Store\Model\Store::STORE_ID] = $this->addStoreIdContext($entityData, \Magento\Store\Model\Store::STORE_ID);
     }
     return $context;
 }
 public function testGetEntityContext()
 {
     $entityContext = ["store_id", "user_id"];
     $metadata = new EntityMetadata($this->appResourceMock, '', '', null, null, null, $entityContext);
     $this->assertEquals($entityContext, $metadata->getEntityContext());
 }