Add test stubs for init mode

Author: aciidgh

Sources/swift-build/initPackage.swift

@@ -20,6 +20,8 @@ func initPackage() throws {
     let gitignore = Path.join(rootd, ".gitignore")
     let sources = Path.join(rootd, "Sources")
     let tests = Path.join(rootd, "Tests")
+    let testModule = Path.join(tests, pkgname)
+    let testsFile = Path.join(testModule, "\(pkgname).swift")
     let main = Path.join(sources, "main.swift")
 
     guard !manifest.exists else {
@@ -67,5 +69,54 @@ func initPackage() throws {
     if !tests.exists {
         print("Creating Tests/")
         try mkdir(tests)
+        
+        print("Creating Tests/\(pkgname)/")
+        try mkdir(testModule)
+        try writeTestFileStubs(testsFile: testsFile, pkgname: pkgname)
+        try writeLinuxMain(path: Path.join(tests, "LinuxMain.swift"), pkgname: pkgname)
+    }
+}
+
+private func writeLinuxMain(path filePath: String, pkgname: String) throws {
+    let linuxMainFP = try fopen(filePath, mode: .Write)
+    defer {
+        fclose(linuxMainFP)
     }
+
+    print("Creating Tests/LinuxMain.swift")
+    try fputs("import XCTest\n", linuxMainFP)
+    try fputs("@testable import \(pkgname)test\n\n", linuxMainFP)
+    try fputs("XCTMain([\n", linuxMainFP)
+    try fputs("\t\(pkgname)(),\n", linuxMainFP)
+    try fputs("])\n", linuxMainFP)
+}
+
+private func writeTestFileStubs(testsFile testsFile: String, pkgname: String) throws {
+    print("Creating Tests/\(pkgname)/\(pkgname).swift")
+    let testsFileFP = try fopen(testsFile, mode: .Write)
+    defer {
+        fclose(testsFileFP)
+    }
+    
+    try fputs("import XCTest\n", testsFileFP)
+    try fputs("@testable import \(pkgname)\n\n", testsFileFP)
+
+    try fputs("class \(pkgname): XCTestCase {\n\n", testsFileFP)
+
+    try fputs("\tfunc testExample() {\n", testsFileFP)
+    try fputs("\t\t// This is an example of a functional test case.\n", testsFileFP)
+    try fputs("\t\t// Use XCTAssert and related functions to verify your tests produce the correct results.\n", testsFileFP)
+    try fputs("\t}\n\n", testsFileFP)
+
+    try fputs("}\n", testsFileFP)
+
+    try fputs("\n#if os(Linux)\n", testsFileFP)
+    try fputs("extension \(pkgname): XCTestCaseProvider {\n", testsFileFP)
+    try fputs("\tvar allTests : [(String, () throws -> Void)] {\n", testsFileFP)
+    try fputs("\t\treturn [\n", testsFileFP)
+    try fputs("\t\t\t(\"testExample\", testExample),\n", testsFileFP)
+    try fputs("\t\t]\n", testsFileFP)
+    try fputs("\t}\n", testsFileFP)
+    try fputs("}\n", testsFileFP)
+    try fputs("#endif\n", testsFileFP)
 }