From c2188a840bc4153ae92112b04b2e06a90d3944aa Mon Sep 17 00:00:00 2001
From: Paul Martin <paul@paulsputer.com>
Date: Wed, 27 Apr 2016 18:58:06 -0400
Subject: [PATCH] Ticket Reference handling #1048

---
 src/test/java/com/gitblit/tests/GroovyScriptTest.java |   81 ++++++++++++++++++++--------------------
 1 files changed, 41 insertions(+), 40 deletions(-)

diff --git a/src/test/java/com/gitblit/tests/GroovyScriptTest.java b/src/test/java/com/gitblit/tests/GroovyScriptTest.java
index 2954fa1..ff40972 100644
--- a/src/test/java/com/gitblit/tests/GroovyScriptTest.java
+++ b/src/test/java/com/gitblit/tests/GroovyScriptTest.java
@@ -15,8 +15,6 @@
  */
 package com.gitblit.tests;
 
-import static org.junit.Assert.assertEquals;
-import static org.junit.Assert.assertTrue;
 import groovy.lang.Binding;
 import groovy.util.GroovyScriptEngine;
 
@@ -41,7 +39,6 @@
 import org.junit.BeforeClass;
 import org.junit.Test;
 
-import com.gitblit.GitBlit;
 import com.gitblit.GitBlitException;
 import com.gitblit.models.RepositoryModel;
 import com.gitblit.models.TeamModel;
@@ -50,11 +47,11 @@
 
 /**
  * Test class for Groovy scripts. Mostly this is to facilitate development.
- * 
+ *
  * @author James Moger
- * 
+ *
  */
-public class GroovyScriptTest {
+public class GroovyScriptTest extends GitblitUnitTest {
 
 	private static final AtomicBoolean started = new AtomicBoolean(false);
 
@@ -83,7 +80,7 @@
 				.fromString("c18877690322dfc6ae3e37bb7f7085a24e94e887"), ObjectId
 				.fromString("3fa7c46d11b11d61f1cbadc6888be5d0eae21969"), "refs/heads/master2"));
 
-		RepositoryModel repository = GitBlit.self().getRepositoryModel("helloworld.git");
+		RepositoryModel repository = repositories().getRepositoryModel("helloworld.git");
 		repository.customFields = new HashMap<String,String>();
 		repository.customFields.put( "fogbugzUrl", "http://bugs.test.com" );
 		repository.customFields.put( "fogbugzRepositoryId", "1" );
@@ -105,7 +102,7 @@
 				.fromString("c18877690322dfc6ae3e37bb7f7085a24e94e887"), ObjectId
 				.fromString("3fa7c46d11b11d61f1cbadc6888be5d0eae21969"), "refs/heads/master2"));
 
-		RepositoryModel repository = GitBlit.self().getRepositoryModel("helloworld.git");
+		RepositoryModel repository = repositories().getRepositoryModel("helloworld.git");
 		repository.mailingLists.add("list@helloworld.git");
 
 		test("sendmail-html.groovy", gitblit, logger, clientLogger, commands, repository);
@@ -130,7 +127,7 @@
 				.fromString("c18877690322dfc6ae3e37bb7f7085a24e94e887"), ObjectId
 				.fromString("3fa7c46d11b11d61f1cbadc6888be5d0eae21969"), "refs/heads/master2"));
 
-		RepositoryModel repository = GitBlit.self().getRepositoryModel("helloworld.git");
+		RepositoryModel repository = repositories().getRepositoryModel("helloworld.git");
 		repository.mailingLists.add("list@helloworld.git");
 
 		test("sendmail.groovy", gitblit, logger, clientLogger, commands, repository);
@@ -140,7 +137,7 @@
 		assertEquals(5, m.toAddresses.size());
 		assertTrue(m.message.contains("BIT"));
 	}
-	
+
 	@Test
 	public void testProtectRefsCreateBranch() throws Exception {
 		MockGitblit gitblit = new MockGitblit();
@@ -149,12 +146,12 @@
 		List<ReceiveCommand> commands = new ArrayList<ReceiveCommand>();
 		commands.add(new ReceiveCommand(ObjectId.zeroId(), ObjectId
 				.fromString("3fa7c46d11b11d61f1cbadc6888be5d0eae21969"), "refs/heads/master"));
-		
-		RepositoryModel repository = new RepositoryModel("ex@mple.git", "", "admin", new Date());		
+
+		RepositoryModel repository = new RepositoryModel("ex@mple.git", "", "admin", new Date());
 
 		test("protect-refs.groovy", gitblit, logger, clientLogger, commands, repository);
 	}
-	
+
 	@Test
 	public void testProtectRefsCreateTag() throws Exception {
 		MockGitblit gitblit = new MockGitblit();
@@ -163,13 +160,13 @@
 		List<ReceiveCommand> commands = new ArrayList<ReceiveCommand>();
 		commands.add(new ReceiveCommand(ObjectId.zeroId(), ObjectId
 				.fromString("3fa7c46d11b11d61f1cbadc6888be5d0eae21969"), "refs/tags/v1.0"));
-		
-		RepositoryModel repository = new RepositoryModel("ex@mple.git", "", "admin", new Date());		
+
+		RepositoryModel repository = new RepositoryModel("ex@mple.git", "", "admin", new Date());
 
 		test("protect-refs.groovy", gitblit, logger, clientLogger, commands, repository);
 		assertEquals(0, logger.messages.size());
 	}
-	
+
 	@Test
 	public void testProtectRefsFastForward() throws Exception {
 		MockGitblit gitblit = new MockGitblit();
@@ -179,13 +176,13 @@
 		commands.add(new ReceiveCommand(ObjectId
 				.fromString("c18877690322dfc6ae3e37bb7f7085a24e94e887"), ObjectId
 				.fromString("3fa7c46d11b11d61f1cbadc6888be5d0eae21969"), "refs/heads/master"));
-		
-		RepositoryModel repository = new RepositoryModel("ex@mple.git", "", "admin", new Date());		
+
+		RepositoryModel repository = new RepositoryModel("ex@mple.git", "", "admin", new Date());
 
 		test("protect-refs.groovy", gitblit, logger, clientLogger, commands, repository);
 		assertEquals(0, logger.messages.size());
 	}
-	
+
 	@Test
 	public void testProtectRefsDeleteMasterBranch() throws Exception {
 		MockGitblit gitblit = new MockGitblit();
@@ -196,14 +193,14 @@
 				.fromString("3fa7c46d11b11d61f1cbadc6888be5d0eae21969"), ObjectId.zeroId(),
 				"refs/heads/master");
 		commands.add(command);
-		
-		RepositoryModel repository = new RepositoryModel("ex@mple.git", "", "admin", new Date());		
+
+		RepositoryModel repository = new RepositoryModel("ex@mple.git", "", "admin", new Date());
 
 		test("protect-refs.groovy", gitblit, logger, clientLogger, commands, repository);
 		assertEquals(ReceiveCommand.Result.REJECTED_NODELETE, command.getResult());
 		assertEquals(0, logger.messages.size());
 	}
-	
+
 	@Test
 	public void testProtectRefsDeleteOtherBranch() throws Exception {
 		MockGitblit gitblit = new MockGitblit();
@@ -213,13 +210,13 @@
 		commands.add(new ReceiveCommand(ObjectId
 				.fromString("3fa7c46d11b11d61f1cbadc6888be5d0eae21969"), ObjectId.zeroId(),
 				"refs/heads/other"));
-		
-		RepositoryModel repository = new RepositoryModel("ex@mple.git", "", "admin", new Date());		
+
+		RepositoryModel repository = new RepositoryModel("ex@mple.git", "", "admin", new Date());
 
 		test("protect-refs.groovy", gitblit, logger, clientLogger, commands, repository);
 		assertEquals(0, logger.messages.size());
 	}
-	
+
 	@Test
 	public void testProtectRefsDeleteTag() throws Exception {
 		MockGitblit gitblit = new MockGitblit();
@@ -230,14 +227,14 @@
 				.fromString("3fa7c46d11b11d61f1cbadc6888be5d0eae21969"), ObjectId.zeroId(),
 				"refs/tags/v1.0");
 		commands.add(command);
-		
-		RepositoryModel repository = new RepositoryModel("ex@mple.git", "", "admin", new Date());		
+
+		RepositoryModel repository = new RepositoryModel("ex@mple.git", "", "admin", new Date());
 
 		test("protect-refs.groovy", gitblit, logger, clientLogger, commands, repository);
 		assertEquals(ReceiveCommand.Result.REJECTED_NODELETE, command.getResult());
 		assertEquals(0, logger.messages.size());
 	}
-	
+
 	@Test
 	public void testBlockPush() throws Exception {
 		MockGitblit gitblit = new MockGitblit();
@@ -247,8 +244,8 @@
 		commands.add(new ReceiveCommand(ObjectId
 				.fromString("c18877690322dfc6ae3e37bb7f7085a24e94e887"), ObjectId
 				.fromString("3fa7c46d11b11d61f1cbadc6888be5d0eae21969"), "refs/heads/master"));
-		
-		RepositoryModel repository = new RepositoryModel("ex@mple.git", "", "admin", new Date());		
+
+		RepositoryModel repository = new RepositoryModel("ex@mple.git", "", "admin", new Date());
 
 		try {
 			test("blockpush.groovy", gitblit, logger, clientLogger, commands, repository);
@@ -257,7 +254,7 @@
 			assertTrue(e.getMessage().contains("failed"));
 		}
 	}
-	
+
 	@Test
 	public void testClientLogging() throws Exception {
 		MockGitblit gitblit = new MockGitblit();
@@ -267,15 +264,15 @@
 		commands.add(new ReceiveCommand(ObjectId
 				.fromString("c18877690322dfc6ae3e37bb7f7085a24e94e887"), ObjectId
 				.fromString("3fa7c46d11b11d61f1cbadc6888be5d0eae21969"), "refs/heads/master"));
-		
+
 		RepositoryModel repository = new RepositoryModel("ex@mple.git", "", "admin", new Date());
-		
-		File groovyDir = GitBlit.getGroovyScriptsFolder();
+
+		File groovyDir = repositories().getHooksFolder();
 		File tempScript = File.createTempFile("testClientLogging", "groovy", groovyDir);
 		tempScript.deleteOnExit();
-		
+
 		BufferedWriter writer = new BufferedWriter(new FileWriter(tempScript));
-		
+
 		writer.write("clientLogger.info('this is a test message')\n");
 		writer.flush();
 		writer.close();
@@ -291,7 +288,7 @@
 
 		String gitblitUrl = GitBlitSuite.url;
 
-		File groovyDir = GitBlit.getGroovyScriptsFolder();
+		File groovyDir = repositories().getHooksFolder();
 		GroovyScriptEngine gse = new GroovyScriptEngine(groovyDir.getAbsolutePath());
 
 		Binding binding = new Binding();
@@ -341,6 +338,10 @@
 			return dv;
 		}
 
+		public int getInteger(String key, int defaultValue) {
+			return defaultValue;
+		}
+
 		public void sendMail(String subject, String message, Collection<String> toAddresses) {
 			messages.add(new MockMail(subject, message, toAddresses));
 		}
@@ -356,18 +357,18 @@
 			messages.add(message);
 		}
 	}
-	
+
 	class MockClientLogger {
 		List<String> messages = new ArrayList<String>();
 
 		public void info(String message) {
 			messages.add(message);
 		}
-		
+
 		public void error(String message) {
 			messages.add(message);
 		}
-		
+
 		public void error(String message, Throwable t) {
 			PrintWriter writer = new PrintWriter(new StringWriter());
 			if (!StringUtils.isEmpty(message)) {

--
Gitblit v1.9.1