PhabricatorTestDataGenerator.php
View Options

	<?php

	abstract class PhabricatorTestDataGenerator extends Phobject {

	private $viewer;

	abstract public function getGeneratorName();
	abstract public function generateObject();

	final public function setViewer(PhabricatorUser $viewer) {
	$this->viewer = $viewer;
	return $this;
	}

	final public function getViewer() {
	return $this->viewer;
	}

	final public function getGeneratorKey() {
	return $this->getPhobjectClassConstant('GENERATORKEY', 64);
	}

	protected function loadRandomPHID($table) {
	$conn_r = $table->establishConnection('r');

	$row = queryfx_one(
	$conn_r,
	'SELECT phid FROM %T ORDER BY RAND() LIMIT 1',
	$table->getTableName());

	if (!$row) {
	return null;
	}

	return $row['phid'];
	}

	protected function loadRandomUser() {
	$viewer = $this->getViewer();

	$user_phid = $this->loadRandomPHID(new PhabricatorUser());

	$user = null;
	if ($user_phid) {
	$user = id(new PhabricatorPeopleQuery())
	->setViewer($viewer)
	->withPHIDs(array($user_phid))
	->needUserSettings(true)
	->executeOne();
	}

	if (!$user) {
	throw new Exception(
	pht(
	'Failed to load a random user. You may need to generate more '.
	'test users first.'));
	}

	return $user;
	}

	protected function getLipsumContentSource() {
	return PhabricatorContentSource::newForSource(
	PhabricatorLipsumContentSource::SOURCECONST);
	}

	/**
	* Roll `n` dice with `d` sides each, then add `bonus` and return the sum.
	*/
	protected function roll($n, $d, $bonus = 0) {
	$sum = 0;
	for ($ii = 0; $ii < $n; $ii++) {
	$sum += mt_rand(1, $d);
	}

	$sum += $bonus;

	return $sum;
	}

	protected function newEmptyTransaction() {
	throw new PhutilMethodNotImplementedException();
	}

	protected function newTransaction($type, $value, $metadata = array()) {
	$xaction = $this->newEmptyTransaction()
	->setTransactionType($type)
	->setNewValue($value);

	foreach ($metadata as $key => $value) {
	$xaction->setMetadataValue($key, $value);
	}

	return $xaction;
	}

	public function loadOneRandom($classname) {
	try {
	return newv($classname, array())
	->loadOneWhere('1 = 1 ORDER BY RAND() LIMIT 1');
	} catch (PhutilMissingSymbolException $ex) {
	throw new PhutilMissingSymbolException(
	$classname,
	pht('class'),
	pht(
	'Unable to load symbol %s: this class does not exist.',
	$classname));
	}
	}

	public function loadPhabricatorUserPHID() {
	return $this->loadOneRandom('PhabricatorUser')->getPHID();
	}

	public function loadPhabricatorUser() {
	return $this->loadOneRandom('PhabricatorUser');
	}

	}

File Metadata

Mime Type: text/x-php
Expires: Sun, Mar 23, 20:02 (1 w, 5 d ago)
Storage Engine: blob
Storage Format: Raw Data
Storage Handle: 1141380
Default Alt Text: PhabricatorTestDataGenerator.php (2 KB)

PhabricatorTestDataGenerator.php
No OneTemporary
Actions

PhabricatorTestDataGenerator.php
View Options

File Metadata

Event Timeline

PhabricatorTestDataGenerator.phpNo OneTemporaryActions

PhabricatorTestDataGenerator.phpView Options

File Metadata

Event Timeline

PhabricatorTestDataGenerator.php
No OneTemporary
Actions

PhabricatorTestDataGenerator.php
View Options