private void SetUpRepositoryCalls(Movement[] movements, ShipmentInfo shipment) { A.CallTo(() => movementRepository.GetMovementsByStatus(NotificationId, MovementStatus.Received)).Returns(movements.Where(m => m.Status == MovementStatus.Received)); A.CallTo(() => movementRepository.GetMovementsByStatus(NotificationId, MovementStatus.Completed)).Returns(movements.Where(m => m.Status == MovementStatus.Completed)); A.CallTo(() => shipmentRepository.GetByNotificationId(NotificationId)).Returns(shipment); }