001 /* 002 * The Apache Software License, Version 1.1 003 * 004 * Copyright (C) 2000-2002 The Apache Software Foundation. All rights 005 * reserved. 006 * Copyright (C) 2003 jcoverage ltd. 007 * Copyright (C) 2005 Mark Doliner <thekingant@users.sourceforge.net> 008 * 009 * Redistribution and use in source and binary forms, with or without 010 * modification, are permitted provided that the following conditions 011 * are met: 012 * 013 * 1. Redistributions of source code must retain the above copyright 014 * notice, this list of conditions and the following disclaimer. 015 * 016 * 2. Redistributions in binary form must reproduce the above copyright 017 * notice, this list of conditions and the following disclaimer in 018 * the documentation and/or other materials provided with the 019 * distribution. 020 * 021 * 3. The end-user documentation included with the redistribution, if 022 * any, must include the following acknowlegement: 023 * "This product includes software developed by the 024 * Apache Software Foundation (http://www.apache.org/)." 025 * Alternately, this acknowlegement may appear in the software itself, 026 * if and wherever such third-party acknowlegements normally appear. 027 * 028 * 4. The names "Ant" and "Apache Software 029 * Foundation" must not be used to endorse or promote products derived 030 * from this software without prior written permission. For written 031 * permission, please contact apache@apache.org. 032 * 033 * 5. Products derived from this software may not be called "Apache" 034 * nor may "Apache" appear in their names without prior written 035 * permission of the Apache Group. 036 * 037 * THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESSED OR IMPLIED 038 * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 039 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE 040 * DISCLAIMED. IN NO EVENT SHALL THE APACHE SOFTWARE FOUNDATION OR 041 * ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 042 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 043 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF 044 * USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND 045 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, 046 * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT 047 * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 048 * SUCH DAMAGE. 049 * ==================================================================== 050 * 051 * This software consists of voluntary contributions made by many 052 * individuals on behalf of the Apache Software Foundation. For more 053 * information on the Apache Software Foundation, please see 054 * <http://www.apache.org/>. 055 */ 056 057 package net.sourceforge.cobertura.ant; 058 059 import java.net.URL; 060 import java.net.URLClassLoader; 061 import java.util.HashSet; 062 import java.util.Iterator; 063 import java.util.Set; 064 065 import org.apache.tools.ant.AntClassLoader; 066 import org.apache.tools.ant.BuildException; 067 import org.apache.tools.ant.taskdefs.Java; 068 import org.apache.tools.ant.taskdefs.MatchingTask; 069 import org.apache.tools.ant.types.Path; 070 import org.apache.tools.ant.types.Reference; 071 072 /** 073 * An ant task that can be used to optionally fail an ant build if 074 * the coverage percentage for lines or branches is below a certain, 075 * user specifiable threshold. 076 */ 077 public class CheckTask extends MatchingTask 078 { 079 080 final Set regexes = new HashSet(); 081 082 protected String branchCoverageRate = null; 083 protected String lineCoverageRate = null; 084 085 private Java java = null; 086 087 public void setBranch(String branchCoverageRate) 088 { 089 this.branchCoverageRate = branchCoverageRate; 090 } 091 092 public void setLine(String lineCoverageRate) 093 { 094 this.lineCoverageRate = lineCoverageRate; 095 } 096 097 public void execute() throws BuildException 098 { 099 if (lineCoverageRate != null) 100 { 101 getJava().createArg().setValue("-l"); 102 getJava().createArg().setValue(lineCoverageRate); 103 } 104 105 if (branchCoverageRate != null) 106 { 107 getJava().createArg().setValue("-b"); 108 getJava().createArg().setValue(branchCoverageRate); 109 } 110 111 Iterator i = regexes.iterator(); 112 while (i.hasNext()) 113 { 114 getJava().createArg().setValue("-r"); 115 getJava().createArg().setValue(i.next().toString()); 116 } 117 118 if (getJava().executeJava() != 0) 119 { 120 throw new BuildException(); 121 } 122 } 123 124 public Regex createRegex() 125 { 126 Regex regex = new Regex(); 127 regexes.add(regex); 128 return regex; 129 } 130 131 protected Java getJava() 132 { 133 if (java == null) 134 { 135 java = (Java)getProject().createTask("java"); 136 java.setTaskName(getTaskName()); 137 java.setClassname("net.sourceforge.cobertura.check.Main"); 138 java.setFork(true); 139 java.setDir(getProject().getBaseDir()); 140 141 if (getClass().getClassLoader() instanceof AntClassLoader) 142 { 143 createClasspath().setPath( 144 ((AntClassLoader)getClass().getClassLoader()) 145 .getClasspath()); 146 } 147 else if (getClass().getClassLoader() instanceof URLClassLoader) 148 { 149 URL[] earls = ((URLClassLoader)getClass().getClassLoader()) 150 .getURLs(); 151 for (int i = 0; i < earls.length; i++) 152 { 153 createClasspath().setPath(earls[i].getFile()); 154 } 155 } 156 } 157 158 return java; 159 } 160 161 public Path createClasspath() 162 { 163 return getJava().createClasspath().createPath(); 164 } 165 166 public void setClasspath(Path classpath) 167 { 168 createClasspath().append(classpath); 169 } 170 171 public void setClasspathRef(Reference r) 172 { 173 createClasspath().setRefid(r); 174 } 175 176 }