1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44 package org.eclipse.jgit.http.test;
45
46 import static org.junit.Assert.assertEquals;
47 import static org.junit.Assert.assertNotNull;
48 import static org.junit.Assert.assertTrue;
49
50 import java.io.ByteArrayOutputStream;
51 import java.io.OutputStream;
52 import java.util.Collection;
53 import java.util.Collections;
54 import java.util.List;
55
56 import javax.servlet.http.HttpServletRequest;
57
58 import org.eclipse.jetty.servlet.ServletContextHandler;
59 import org.eclipse.jetty.servlet.ServletHolder;
60 import org.eclipse.jgit.errors.RepositoryNotFoundException;
61 import org.eclipse.jgit.http.server.GitServlet;
62 import org.eclipse.jgit.http.server.resolver.DefaultReceivePackFactory;
63 import org.eclipse.jgit.junit.TestRepository;
64 import org.eclipse.jgit.junit.http.AccessEvent;
65 import org.eclipse.jgit.junit.http.HttpTestCase;
66 import org.eclipse.jgit.lib.Constants;
67 import org.eclipse.jgit.lib.NullProgressMonitor;
68 import org.eclipse.jgit.lib.ObjectId;
69 import org.eclipse.jgit.lib.Repository;
70 import org.eclipse.jgit.lib.StoredConfig;
71 import org.eclipse.jgit.revwalk.RevBlob;
72 import org.eclipse.jgit.revwalk.RevCommit;
73 import org.eclipse.jgit.transport.PreReceiveHook;
74 import org.eclipse.jgit.transport.PushResult;
75 import org.eclipse.jgit.transport.ReceiveCommand;
76 import org.eclipse.jgit.transport.ReceivePack;
77 import org.eclipse.jgit.transport.RemoteRefUpdate;
78 import org.eclipse.jgit.transport.Transport;
79 import org.eclipse.jgit.transport.URIish;
80 import org.eclipse.jgit.transport.resolver.RepositoryResolver;
81 import org.eclipse.jgit.transport.resolver.ServiceNotAuthorizedException;
82 import org.eclipse.jgit.transport.resolver.ServiceNotEnabledException;
83 import org.junit.Before;
84 import org.junit.Test;
85
86 public class HookMessageTest extends HttpTestCase {
87 private Repository remoteRepository;
88
89 private URIish remoteURI;
90
91 @Override
92 @Before
93 public void setUp() throws Exception {
94 super.setUp();
95
96 final TestRepository<Repository> src = createTestRepository();
97 final String srcName = src.getRepository().getDirectory().getName();
98
99 ServletContextHandler app = server.addContext("/git");
100 GitServlet gs = new GitServlet();
101 gs.setRepositoryResolver(new RepositoryResolver<HttpServletRequest>() {
102 @Override
103 public Repository open(HttpServletRequest req, String name)
104 throws RepositoryNotFoundException,
105 ServiceNotEnabledException {
106 if (!name.equals(srcName))
107 throw new RepositoryNotFoundException(name);
108
109 final Repository db = src.getRepository();
110 db.incrementOpen();
111 return db;
112 }
113 });
114 gs.setReceivePackFactory(new DefaultReceivePackFactory() {
115 @Override
116 public ReceivePack create(HttpServletRequest req, Repository db)
117 throws ServiceNotEnabledException,
118 ServiceNotAuthorizedException {
119 ReceivePack recv = super.create(req, db);
120 recv.setPreReceiveHook(new PreReceiveHook() {
121 @Override
122 public void onPreReceive(ReceivePack rp,
123 Collection<ReceiveCommand> commands) {
124 rp.sendMessage("message line 1");
125 rp.sendError("no soup for you!");
126 rp.sendMessage("come back next year!");
127 }
128 });
129 return recv;
130 }
131
132 });
133 app.addServlet(new ServletHolder(gs), "/*");
134
135 server.setUp();
136
137 remoteRepository = src.getRepository();
138 remoteURI = toURIish(app, srcName);
139
140 StoredConfig cfg = remoteRepository.getConfig();
141 cfg.setBoolean("http", null, "receivepack", true);
142 cfg.save();
143 }
144
145 @Test
146 public void testPush_CreateBranch() throws Exception {
147 final TestRepository src = createTestRepository();
148 final RevBlob Q_txt = src.blob("new text");
149 final RevCommit Q = src.commit().add("Q", Q_txt).create();
150 final Repository db = src.getRepository();
151 final String dstName = Constants.R_HEADS + "new.branch";
152 PushResult result;
153
154 try (Transport t = Transport.open(db, remoteURI)) {
155 final String srcExpr = Q.name();
156 final boolean forceUpdate = false;
157 final String localName = null;
158 final ObjectId oldId = null;
159
160 RemoteRefUpdate update = new RemoteRefUpdate(src.getRepository(),
161 srcExpr, dstName, forceUpdate, localName, oldId);
162 result = t.push(NullProgressMonitor.INSTANCE, Collections
163 .singleton(update));
164 }
165
166 assertTrue(remoteRepository.getObjectDatabase().has(Q_txt));
167 assertNotNull("has " + dstName, remoteRepository.exactRef(dstName));
168 assertEquals(Q, remoteRepository.exactRef(dstName).getObjectId());
169 fsck(remoteRepository, Q);
170
171 List<AccessEvent> requests = getRequests();
172 assertEquals(2, requests.size());
173
174 AccessEvent service = requests.get(1);
175 assertEquals("POST", service.getMethod());
176 assertEquals(join(remoteURI, "git-receive-pack"), service.getPath());
177 assertEquals(200, service.getStatus());
178
179 assertEquals("message line 1\n"
180 + "error: no soup for you!\n"
181 + "come back next year!\n",
182 result.getMessages());
183 }
184
185 @Test
186 public void testPush_HookMessagesToOutputStream() throws Exception {
187 final TestRepository src = createTestRepository();
188 final RevBlob Q_txt = src.blob("new text");
189 final RevCommit Q = src.commit().add("Q", Q_txt).create();
190 final Repository db = src.getRepository();
191 final String dstName = Constants.R_HEADS + "new.branch";
192 PushResult result;
193
194 OutputStream out = new ByteArrayOutputStream();
195 try (Transport t = Transport.open(db, remoteURI)) {
196 final String srcExpr = Q.name();
197 final boolean forceUpdate = false;
198 final String localName = null;
199 final ObjectId oldId = null;
200
201 RemoteRefUpdate update = new RemoteRefUpdate(src.getRepository(),
202 srcExpr, dstName, forceUpdate, localName, oldId);
203 result = t.push(NullProgressMonitor.INSTANCE,
204 Collections.singleton(update), out);
205 }
206
207 String expectedMessage = "message line 1\n"
208 + "error: no soup for you!\n"
209 + "come back next year!\n";
210 assertEquals(expectedMessage,
211 result.getMessages());
212
213 assertEquals(expectedMessage, out.toString());
214 }
215
216 }