Browse Source

allow custom filter in UnitBuilder

Simon Krajewski 11 years ago
parent
commit
866821eef5
1 changed files with 2 additions and 2 deletions
  1. 2 2
      tests/unit/UnitBuilder.hx

+ 2 - 2
tests/unit/UnitBuilder.hx

@@ -29,7 +29,7 @@ using StringTools;
 
 class UnitBuilder {
 	
-	static public macro function build(basePath:String):Array<Field> {
+	static public macro function build(basePath:String, filter:String = ".unit.hx"):Array<Field> {
 		var ret = Context.getBuildFields();
 		var numFiles = 0;
 			
@@ -38,7 +38,7 @@ class UnitBuilder {
 			path = path.endsWith("\\") || path.endsWith("/") ? path : path + "/";
 			for (file in dir) {
 				var filePath = path + file;
-				if (file.endsWith(".unit.hx")) {
+				if (file.endsWith(filter)) {
 					numFiles++;
 					var func = {
 						args: [],