re issue #6 Accept duration parameter in TeamCity-TestFinished

psake · Apr 6, 2012 · c67c2b2 · c67c2b2
1 parent 393a478
commit c67c2b2
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/teamcity.psm1 b/teamcity.psm1
@@ -19,8 +19,14 @@ function TeamCity-TestStarted([string]$name) {
 	TeamCity-WriteServiceMessage 'testStarted' @{ name=$name }
 }
 
-function TeamCity-TestFinished([string]$name) {
-	TeamCity-WriteServiceMessage 'testFinished' @{ name=$name }
+function TeamCity-TestFinished([string]$name, [int]$duration) {
+	$messageAttributes = @{name=$name; duration=$duration}
+
+	if ($duration -gt 0) {
+		$messageAttributes.duration=$duration
+	}
+
+	TeamCity-WriteServiceMessage 'testFinished' $messageAttributes
 }
 
 function TeamCity-TestIgnored([string]$name, [string]$message='') {