TW-57802 Fix tests
[teamcity/git-plugin.git] / git-tests / src / jetbrains / buildServer / buildTriggers / vcs / git / tests / LoggingGitMetaFactory.java
1 /*
2  * Copyright 2000-2018 JetBrains s.r.o.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  * http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16
17 package jetbrains.buildServer.buildTriggers.vcs.git.tests;
18
19 import jetbrains.buildServer.buildTriggers.vcs.git.agent.*;
20 import org.jetbrains.annotations.NotNull;
21
22 import java.io.File;
23 import java.util.Collections;
24 import java.util.HashMap;
25 import java.util.List;
26 import java.util.Map;
27
28 public class LoggingGitMetaFactory implements GitMetaFactory {
29   private final Map<String, List<String>> myInvokedMethods = new HashMap<String, List<String>>();
30   private final Map<String, GitCommandProxyCallback> myCallbacks = new HashMap<String, GitCommandProxyCallback>();
31
32   @NotNull
33   public GitFactory createFactory(@NotNull GitAgentSSHService sshService,
34                                   @NotNull AgentPluginConfig config,
35                                   @NotNull GitProgressLogger logger,
36                                   @NotNull File tempDir,
37                                   @NotNull Map<String, String> env,
38                                   @NotNull Context ctx) {
39     return new GitFactoryProxy(sshService, config, tempDir, env, myInvokedMethods, myCallbacks, ctx);
40   }
41
42
43   public void clear() {
44     myInvokedMethods.clear();
45   }
46
47   @NotNull
48   public List<String> getInvokedMethods(@NotNull Class gitCommandClass) {
49     List<String> methods = myInvokedMethods.get(gitCommandClass.getName());
50     if (methods != null)
51       return methods;
52     return Collections.emptyList();
53   }
54
55   public int getNumberOfCalls(@NotNull Class gitCommandClass) {
56     List<String> methods = myInvokedMethods.get(gitCommandClass.getName());
57     if (methods == null)
58       return 0;
59     int result = 0;
60     for (String method : methods) {
61       if ("call".equals(method) || "callWithIgnoreExitCode".equals(method)) {
62         result++;
63       }
64     }
65     return result;
66   }
67
68   public void addCallback(@NotNull String gitFacadeMethodName, @NotNull GitCommandProxyCallback callback) {
69     myCallbacks.put(gitFacadeMethodName, callback);
70   }
71 }