46be8aba2b3cd3ce90d50991fd3fb92f2acd6548
[teamcity/git-plugin.git] / git-tests / src / jetbrains / buildServer / buildTriggers / vcs / git / tests / MirrorManagerTest.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.TempFiles;
20 import jetbrains.buildServer.buildTriggers.vcs.git.*;
21 import jetbrains.buildServer.serverSide.ServerPaths;
22 import org.eclipse.jgit.transport.URIish;
23 import org.jetbrains.annotations.NotNull;
24 import org.testng.annotations.AfterMethod;
25 import org.testng.annotations.BeforeMethod;
26 import org.testng.annotations.Test;
27
28 import java.io.File;
29 import java.util.Arrays;
30 import java.util.HashMap;
31 import java.util.List;
32 import java.util.Map;
33
34 import static jetbrains.buildServer.buildTriggers.vcs.git.GitServerUtil.getRepository;
35 import static org.testng.AssertJUnit.*;
36
37 /**
38  * @author dmitry.neverov
39  */
40 @Test
41 public class MirrorManagerTest {
42
43   private TempFiles myTempFiles = new TempFiles();
44   private ServerPluginConfig myConfig;
45
46
47   @BeforeMethod
48   public void setUp() throws Exception {
49     ServerPaths paths = new ServerPaths(myTempFiles.createTempDir().getAbsolutePath());
50     myConfig = new PluginConfigBuilder(paths).build();
51   }
52
53
54   @AfterMethod
55   public void tearDown() {
56     myTempFiles.cleanup();
57   }
58
59
60   public void should_handle_clashing_urls() {
61     String clashingUrl1 = "git://some.org/first-clashing-url.git";
62     String clashingUrl2 = "git://some.org/second-clashing-url.git";
63     HashCalculator clashingHash = new ClashingHashCalculator(Arrays.asList(clashingUrl1, clashingUrl2));
64     MirrorManager mirrorManager = new MirrorManagerImpl(myConfig, clashingHash);
65     File dir1 = mirrorManager.getMirrorDir(clashingUrl1);
66     File dir2 = mirrorManager.getMirrorDir(clashingUrl2);
67     assertFalse(dir1.equals(dir2));
68   }
69
70
71   public void should_restore_mapping_from_existing_repositories() throws Exception {
72     //There are 3 existing repositories and no map file (this is the first start after
73     //mirror manager is introduced). After the start map file should be created
74     //and mirrorManager should respect old repositories locations.
75
76     File baseMirrorsDir = myConfig.getCachesDir();
77     Map<String, String> existingRepositories = new HashMap<String, String>() {{
78       put("git://some.org/repository1.git", "git-11111111.git");
79       put("git://some.org/repository2.git", "git-22222222.git");
80       put("git://some.org/repository3.git", "git-33333333.git");
81     }};
82
83     File map = new File(baseMirrorsDir, "map");
84     createRepositories(baseMirrorsDir, existingRepositories);
85     assertFalse(map.exists());
86
87     MirrorManager mirrorManager = new MirrorManagerImpl(myConfig, new HashCalculatorImpl());
88     assertTrue(map.exists());
89     for (Map.Entry<String, String> entry : existingRepositories.entrySet()) {
90       String url = entry.getKey();
91       String dir = entry.getValue();
92       assertEquals(new File(baseMirrorsDir, dir), mirrorManager.getMirrorDir(url));
93     }
94   }
95
96
97   public void should_give_different_dirs_for_same_url_if_dir_was_invalidated() {
98     MirrorManager mirrorManager = new MirrorManagerImpl(myConfig, new HashCalculatorImpl());
99     String url = "git://some.org/repository.git";
100     File dir1 = mirrorManager.getMirrorDir(url);
101     mirrorManager.invalidate(dir1);
102     File dir2 = mirrorManager.getMirrorDir(url);
103     assertFalse(dir1.equals(dir2));
104   }
105
106
107   public void should_remember_invalidated_dirs_after_restart() {
108     String url1 = "git://some.org/repository1.git";
109     String url2 = "git://some.org/repository2.git";
110     HashCalculator clashingHash = new ClashingHashCalculator(Arrays.asList(url1, url2));
111     MirrorManager mirrorManager = new MirrorManagerImpl(myConfig, clashingHash);
112     File dir1 = mirrorManager.getMirrorDir(url1);
113     mirrorManager.invalidate(dir1);
114     mirrorManager = new MirrorManagerImpl(myConfig, clashingHash); //restart
115     File dir2 = mirrorManager.getMirrorDir(url2);
116     assertFalse(dir1.equals(dir2));
117   }
118
119
120   private void createRepositories(File baseDir, Map<String, String> url2dir) throws Exception {
121     for (Map.Entry<String, String> entry : url2dir.entrySet()) {
122       String url = entry.getKey();
123       String dir = entry.getValue();
124       getRepository(new File(baseDir, dir), new URIish(url));
125     }
126   }
127
128
129   private static class ClashingHashCalculator implements HashCalculator {
130     private final List<String> myValuesWhenShouldClash;
131
132     ClashingHashCalculator(List<String> valuesWhenShouldClash) {
133       myValuesWhenShouldClash = valuesWhenShouldClash;
134     }
135
136     public long getHash(@NotNull String value) {
137       if (myValuesWhenShouldClash.contains(value))
138         return 42;
139       else
140         return value.hashCode();
141     }
142   }
143 }