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 @Before
92 public void setUp() throws Exception {
93 super.setUp();
94
95 final TestRepository<Repository> src = createTestRepository();
96 final String srcName = src.getRepository().getDirectory().getName();
97
98 ServletContextHandler app = server.addContext("/git");
99 GitServlet gs = new GitServlet();
100 gs.setRepositoryResolver(new RepositoryResolver<HttpServletRequest>() {
101 public Repository open(HttpServletRequest req, String name)
102 throws RepositoryNotFoundException,
103 ServiceNotEnabledException {
104 if (!name.equals(srcName))
105 throw new RepositoryNotFoundException(name);
106
107 final Repository db = src.getRepository();
108 db.incrementOpen();
109 return db;
110 }
111 });
112 gs.setReceivePackFactory(new DefaultReceivePackFactory() {
113 public ReceivePack create(HttpServletRequest req, Repository db)
114 throws ServiceNotEnabledException,
115 ServiceNotAuthorizedException {
116 ReceivePack recv = super.create(req, db);
117 recv.setPreReceiveHook(new PreReceiveHook() {
118 public void onPreReceive(ReceivePack rp,
119 Collection<ReceiveCommand> commands) {
120 rp.sendMessage("message line 1");
121 rp.sendError("no soup for you!");
122 rp.sendMessage("come back next year!");
123 }
124 });
125 return recv;
126 }
127
128 });
129 app.addServlet(new ServletHolder(gs), "/*");
130
131 server.setUp();
132
133 remoteRepository = src.getRepository();
134 remoteURI = toURIish(app, srcName);
135
136 StoredConfig cfg = remoteRepository.getConfig();
137 cfg.setBoolean("http", null, "receivepack", true);
138 cfg.save();
139 }
140
141 @Test
142 public void testPush_CreateBranch() throws Exception {
143 final TestRepository src = createTestRepository();
144 final RevBlob Q_txt = src.blob("new text");
145 final RevCommit Q = src.commit().add("Q", Q_txt).create();
146 final Repository db = src.getRepository();
147 final String dstName = Constants.R_HEADS + "new.branch";
148 Transport t;
149 PushResult result;
150
151 t = Transport.open(db, remoteURI);
152 try {
153 final String srcExpr = Q.name();
154 final boolean forceUpdate = false;
155 final String localName = null;
156 final ObjectId oldId = null;
157
158 RemoteRefUpdate update = new RemoteRefUpdate(src.getRepository(),
159 srcExpr, dstName, forceUpdate, localName, oldId);
160 result = t.push(NullProgressMonitor.INSTANCE, Collections
161 .singleton(update));
162 } finally {
163 t.close();
164 }
165
166 assertTrue(remoteRepository.hasObject(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 Transport t;
193 PushResult result;
194
195 t = Transport.open(db, remoteURI);
196 OutputStream out = new ByteArrayOutputStream();
197 try {
198 final String srcExpr = Q.name();
199 final boolean forceUpdate = false;
200 final String localName = null;
201 final ObjectId oldId = null;
202
203 RemoteRefUpdate update = new RemoteRefUpdate(src.getRepository(),
204 srcExpr, dstName, forceUpdate, localName, oldId);
205 result = t.push(NullProgressMonitor.INSTANCE,
206 Collections.singleton(update), out);
207 } finally {
208 t.close();
209 }
210
211 String expectedMessage = "message line 1\n"
212 + "error: no soup for you!\n"
213 + "come back next year!\n";
214 assertEquals(expectedMessage,
215 result.getMessages());
216
217 assertEquals(expectedMessage, out.toString());
218 }
219
220 }