GEEKING

boostlibs/Base.UnitTest.Lib/teamcity_messages.cpp

@@ -0,0 +1,115 @@
+/* Copyright 2009 JetBrains
+ * 
+ * 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.
+ *
+ * $Revision: 51263 $
+*/
+
+#include <stdlib.h>
+
+#include "teamcity_messages.h"
+
+using namespace std;
+
+namespace JetBrains {
+
+bool underTeamcity() {
+	return getenv("TEAMCITY_PROJECT_NAME") != NULL;
+}
+
+TeamcityMessages::TeamcityMessages()
+: m_out(&cout)
+{}
+
+void TeamcityMessages::setOutput(ostream &out) {
+    m_out = &out;
+}
+
+string TeamcityMessages::escape(string s) {
+    string result;
+    
+    for (size_t i = 0; i < s.length(); i++) {
+        char c = s[i];
+        
+        switch (c) {
+        case '\n': result.append("|n"); break;
+        case '\r': result.append("|r"); break;
+        case '\'': result.append("|'"); break;
+        case '|':  result.append("||"); break;
+        case ']':  result.append("|]"); break;
+        default:   result.append(&c, 1);
+        }
+    }
+    
+    return result;
+}
+
+void TeamcityMessages::openMsg(const string &name) {
+    // endl for http://jetbrains.net/tracker/issue/TW-4412
+    *m_out << endl << "##teamcity[" << name;
+}
+
+void TeamcityMessages::closeMsg() {
+    *m_out << "]";
+    // endl for http://jetbrains.net/tracker/issue/TW-4412
+    *m_out << endl;
+    m_out->flush();
+}
+
+void TeamcityMessages::writeProperty(string name, string value) {
+    *m_out << " " << name << "='" << escape(value) << "'";
+}
+
+void TeamcityMessages::suiteStarted(string name) {
+    openMsg("testSuiteStarted");
+    writeProperty("name", name);
+    closeMsg();
+}
+
+void TeamcityMessages::suiteFinished(string name) {
+    openMsg("testSuiteFinished");
+    writeProperty("name", name);
+    closeMsg();
+}
+
+void TeamcityMessages::testStarted(string name) {
+    openMsg("testStarted");
+    writeProperty("name", name);
+    closeMsg();
+}
+
+void TeamcityMessages::testFinished(string name, unsigned long elapsed) {
+    openMsg("testFinished");
+	writeProperty("name", name);
+	char buffer[32];
+	sprintf(buffer, "%lu", elapsed/1000);
+	writeProperty("duration", buffer);
+    closeMsg();
+}
+
+void TeamcityMessages::testFailed(string name, string message, string details) {
+    openMsg("testFailed");
+    writeProperty("name", name);
+    writeProperty("message", message);
+    writeProperty("details", details);
+    closeMsg();
+}
+
+void TeamcityMessages::testIgnored(std::string name, std::string message) {
+    openMsg("testIgnored");
+    writeProperty("name", name);
+    writeProperty("message", message);
+    closeMsg();
+}
+
+}