1 files changed, 8 insertions, 2 deletions

diff --git a/build.conf b/build.conf
index 898d6d4..769fa31 100644
--- a/build.conf
+++ b/build.conf
@@ -49,7 +49,7 @@ filesIn("src/tests") filter regexp("^src/tests/(.*)\\.cpp$", "tests/{re:1}"):
  "tests/itoheap"]:
         set "LDFLAGS" += "-lpthread"
 
-filesIn("src/unit") filter regexp("^src/unit/(.*)\\.cpp$", "unit/{re:1}"):
+filesIn("src/unit") filter regexp("^src/unit/(.*)\\.unit$", "unit/{re:1}"):
         rule "exe",
         target file,
         group "tests",
@@ -57,7 +57,7 @@ filesIn("src/unit") filter regexp("^src/unit/(.*)\\.cpp$", "unit/{re:1}"):
         requires "libbu++.a",
         set "CXXFLAGS" += "-I.",
         set "LDFLAGS" += "-L. -lbu++",
-        input "src/{target}.cpp"
+        input "src/{target}.unit"
 
 "tests/plugin": set "LDFLAGS" += "-ldl"
 
@@ -79,6 +79,12 @@ rule "cpp":
         requires commandToList("g++ -M {CXXFLAGS} {match}", "make"),
         perform command("g++ {CXXFLAGS} -c -o {target} {match}")
 
+rule "unit":
+        matches regexp("(.*)\\.unit$"),
+        produces "{re:1}.cpp",
+        requires "mkunit.sh",
+        perform command("./mkunit.sh \"{match}\" \"{target}\"")
+
 rule "hln":
         matches regexp("src/(.*)\\.h"),
         produces "bu/{re:1}.h",