Spec/Core/Media/Delegates/PropagatePropertiesSpec.php

@@ -29,8 +29,8 @@ class PropagatePropertiesSpec extends ObjectBehavior
  public function it_should_propagate_changes_to_activity()
  {
  $this->entity->get('title')->shouldBeCalled()->willReturn('new title');
- $this->activity->getMessage()->shouldBeCalled()->willReturn('old title');
- $this->activity->setMessage('new title')->shouldBeCalled();
+ $this->activity->get('title')->shouldBeCalled()->willReturn('old title');
+ $this->activity->set('title', 'new title')->shouldBeCalled();
 
  $activityParameters = [
  'batch',
@@ -49,6 +49,10 @@ class PropagatePropertiesSpec extends ObjectBehavior
 
  public function it_should_propogate_properties_from_activity()
  {
+ $this->activity->get('title')->shouldBeCalled()->willReturn('new title');
+ $this->entity->get('title')->shouldbeCalled()->willReturn('old title');
+ $this->entity->set('title', 'new title')->shouldBeCalled();
+
  $this->fromActivity($this->activity, $this->entity);
  }
 }