From 2730c31c3e5d29b08779bd805ee5dd4a73acc692 Mon Sep 17 00:00:00 2001 From: Ekaterina Tuzova Date: Tue, 28 Jun 2011 19:26:29 +0400 Subject: [PATCH] fixed PY-3942 Test runner doesn't seem to notice py.test tests have completed when using classes --- python/helpers/pytest_teamcity.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/python/helpers/pytest_teamcity.py b/python/helpers/pytest_teamcity.py index 76081d6d99a1..c1b69f5ad619 100644 --- a/python/helpers/pytest_teamcity.py +++ b/python/helpers/pytest_teamcity.py @@ -63,6 +63,8 @@ if PYVERSION > [1,3,4]: messages.testFinished(name) def pytest_sessionfinish(session, exitstatus): + if current_suite: + messages.testSuiteFinished(current_suite) if current_file_suite: messages.testSuiteFinished(current_file_suite)