better-testing

PhrequentTrackingEditor.php
View Options

			<?php

			final class PhrequentTrackingEditor extends PhabricatorEditor {

			public function startTracking(PhabricatorUser $user, $phid, $timestamp) {
			$usertime = new PhrequentUserTime();
			$usertime->setDateStarted($timestamp);
			$usertime->setUserPHID($user->getPHID());
			$usertime->setObjectPHID($phid);
			$usertime->save();

			return $phid;
			}

			public function stopTracking(
			PhabricatorUser $user,
			$phid,
			$timestamp,
			$note) {

			if (!PhrequentUserTimeQuery::isUserTrackingObject($user, $phid)) {
			// Don't do anything, it's not being tracked.
			return null;
			}

			$usertime_dao = new PhrequentUserTime();
			$conn = $usertime_dao->establishConnection('r');

			queryfx(
			$conn,
			'UPDATE %T usertime '.
			'SET usertime.dateEnded = %d, '.
			'usertime.note = %s '.
			'WHERE usertime.userPHID = %s '.
			'AND usertime.objectPHID = %s '.
			'AND usertime.dateEnded IS NULL '.
			'ORDER BY usertime.dateStarted, usertime.id DESC '.
			'LIMIT 1',
			$usertime_dao->getTableName(),
			$timestamp,
			$note,
			$user->getPHID(),
			$phid);

			return $phid;
			}

			public function stopTrackingTop(PhabricatorUser $user, $timestamp, $note) {
			$times = id(new PhrequentUserTimeQuery())
			->setViewer($user)
			->withUserPHIDs(array($user->getPHID()))
			->withEnded(PhrequentUserTimeQuery::ENDED_NO)
			->setOrder(PhrequentUserTimeQuery::ORDER_STARTED_DESC)
			->execute();

			if (count($times) === 0) {
			// Nothing to stop tracking.
			return null;
			}

			$current = head($times);

			return $this->stopTracking(
			$user,
			$current->getObjectPHID(),
			$timestamp,
			$note);
			}

			}