--- a/mercurial-tests/src/jetbrains/buildServer/buildTriggers/vcs/mercurial/AgentMirrorCleanerTest.java	Fri Nov 07 14:55:14 2014 +0100
+++ b/mercurial-tests/src/jetbrains/buildServer/buildTriggers/vcs/mercurial/AgentMirrorCleanerTest.java	Fri Nov 07 16:04:52 2014 +0100
@@ -69,10 +69,13 @@
     AgentRepoFactory repoFactory = new AgentRepoFactory(pluginConfig, new CommandSettingsForRootImpl(new TestCommandSettingsFactory(), new ExtensionsWeaver(), new CommandlineViaFileWrapperWeaver()), new AgentHgPathProvider(agentConfig));
     myVcsSupport = new MercurialAgentSideVcsSupport(pluginConfig, myMirrorManager, repoFactory, new MercurialExtensionManager());
     myCleaner = new AgentMirrorCleaner(myMirrorManager);
-    myLogger = myContext.mock(BuildProgressLogger.class);
+    myLogger = myContext.mock(FlowLogger.class);
     myContext.checking(new Expectations() {{
       allowing(myLogger).message(with(any(String.class)));
       allowing(myLogger).warning(with(any(String.class)));
+      allowing(myLogger).activityStarted(with(any(String.class)), with(any(String.class)));
+      allowing(myLogger).activityFinished(with(any(String.class)), with(any(String.class)));
+      allowing(myLogger).getFlowLogger(with(any(String.class))); will(returnValue(myLogger));
     }});
   }