better-testing

PhabricatorObjectGraph.php
View Options

			<?php

			/*
			* Copyright 2011 Facebook, Inc.
			*
			* Licensed under the Apache License, Version 2.0 (the "License");
			* you may not use this file except in compliance with the License.
			* You may obtain a copy of the License at
			*
			* http://www.apache.org/licenses/LICENSE-2.0
			*
			* Unless required by applicable law or agreed to in writing, software
			* distributed under the License is distributed on an "AS IS" BASIS,
			* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
			* See the License for the specific language governing permissions and
			* limitations under the License.
			*/

			final class PhabricatorObjectGraph extends AbstractDirectedGraph {

			private $edgeType;

			public function setEdgeType($edge_type) {
			$this->edgeType = $edge_type;
			return $this;
			}

			protected function loadEdges(array $nodes) {
			if (!$this->edgeType) {
			throw new Exception("Set edge type before loading graph!");
			}

			$handle_data = new PhabricatorObjectHandleData($nodes);
			$objects = $handle_data->loadObjects();

			$result = array();
			foreach ($nodes as $phid) {
			$object = idx($objects, $phid);
			if ($object) {
			$result[$phid] = $object->getAttachedPHIDs($this->edgeType);
			} else {
			$result[$phid] = array();
			}
			}

			return $result;
			}

			}