+ test for PinbaClient

Author: dovg

doc/ChangeLog

@@ -1,7 +1,12 @@
+2012-04-12	Evgeny V. Kokovikhin
+
+	* test/main/Utils/PinbaTest.class.php, main/Monitoring/PinbaClient.class.php:
+	test for PinbaClient added. Thanks to Artem A. Naumenko
+
 2012-04-11	Evgeny V. Kokovikhin
 
 	* main/Monitoring/PinbaClient.class.php: base ability of
-	tree (integral) logging added. thanks to Artem A. Naumenko
+	tree (integral) logging added. Thanks to Artem A. Naumenko
 
 2012-04-04	Georgiy T. Kutsurua
 

main/Monitoring/PinbaClient.class.php

@@ -49,6 +49,11 @@ public function isTreeLogEnabled()
 			return $this->treeLogEnabled;
 		}
 
+		public function getTreeQueue()
+		{
+			return $this->queue;
+		}
+		
 		public function timerStart($name, array $tags, array $data = array())
 		{
 			if (array_key_exists($name, $this->timers))
@@ -60,7 +65,7 @@ public function timerStart($name, array $tags, array $data = array())
 				$tags['treeId'] = $id;
 
 				if (!empty($this->queue))
-					list($tags['treeParentId']) = end($this->queue);
+					$tags['treeParentId'] = end($this->queue);
 				else
 					$tags['treeParentId'] = 'root';
 

test/main/Utils/PinbaTest.class.php

@@ -0,0 +1,125 @@
+<?php
+	
+	final class PinbaTest extends TestCase
+	{
+		protected function setUp()
+		{
+			if (!extension_loaded('pinba'))
+				$this->markTestSkipped(
+					'The pinba extension is not available.'
+				);
+			
+			if (!PinbaClient::isEnabled())
+				$this->markTestSkipped(
+					'The pinba is not enabled at php.ini (pinba.enabled=1).'
+				);
+			
+			if (!extension_loaded('runkit')) {
+				$this->markTestSkipped(
+					'The runkit extension is not available.'
+				);
+			}
+			
+			if (!ini_get('runkit.internal_override'))
+				$this->markTestSkipped(
+					'The runkit.internal_override is not enabled (enabled it at php.ini).'
+				);
+			
+			runkit_function_rename('pinba_timer_start', 'pinba_timer_start_bak');
+			runkit_function_rename('pinba_timer_stop', 'pinba_timer_stop_bak');
+			
+			runkit_function_rename('pinba_timer_start_callback', 'pinba_timer_start');
+			runkit_function_rename('pinba_timer_stop_callback', 'pinba_timer_stop');
+		}
+		
+		public static function tearDownAfterClass(){
+			
+			if (
+				!extension_loaded('runkit') 
+				|| !ini_get('runkit.internal_override')
+			)
+				return;
+			
+			runkit_function_rename('pinba_timer_start', 'pinba_timer_start_callback');
+			runkit_function_rename('pinba_timer_stop', 'pinba_timer_stop_callback');
+			
+			runkit_function_rename('pinba_timer_start_bak', 'pinba_timer_start');
+			runkit_function_rename('pinba_timer_stop_bak', 'pinba_timer_stop');
+		}
+		
+		public function testTreeLog()
+		{
+			PinbaClient::me()->setTreeLogEnabled();
+			
+			$this->assertEquals(count(PinbaClient::me()->getTreeQueue()), 0);
+			
+			PinbaClient::me()->timerStart(
+				'test',
+				array("test" => "main")
+			);
+			
+			$this->assertEquals(count(PinbaClient::me()->getTreeQueue()), 1);
+			
+			PinbaClient::me()->timerStart(
+				'subtest',
+				array("test" => "submain")
+			);
+			
+			$this->assertEquals(count(PinbaClient::me()->getTreeQueue()), 2);
+			
+			PinbaClient::me()->timerStop('subtest');
+			
+			$this->assertEquals(count(PinbaClient::me()->getTreeQueue()), 1);
+			
+			PinbaClient::me()->timerStop('test');
+			
+			$this->assertEquals(count(PinbaClient::me()->getTreeQueue()), 0);
+			
+		}
+	}
+	
+	final class RunkitCallback
+	{
+		public static $queue = array();
+		public static $log = array();
+		
+		public static function start($tags, $data = array())
+		{
+			self::$log[] = $tags;
+			end(self::$log);
+			
+			if (!empty($tags['treeParentId']) && $tags['treeParentId'] != "root") {
+				if ($tags['treeParentId'] != end(self::$queue)) {
+					throw new Exception("Error generatin tree");
+				}
+			}
+			
+			if (!empty($tags['treeId'])) {
+				self::$queue[] = $tags['treeId'];
+			}
+			
+			return key(self::$log);
+		}
+		
+		public static function stop($id)
+		{
+			$current = self::$log[$id];
+			$tree_id = $current['treeId'];
+			
+			if (end(self::$queue) != $tree_id) {
+				throw new Exception("Error generatin tree");
+			}
+			
+			array_pop(self::$queue);
+			unset(self::$log[$id]);
+		}
+	}
+	
+	function pinba_timer_start_callback ($tags, $data = array()) {
+		return RunkitCallback::start($tags, $data);
+	}
+
+	function pinba_timer_stop_callback($id){
+		return RunkitCallback::stop($id);
+	}
+?>