Fix Swift compatibility issue around XCTMain (#465)

* Fix Swift compatibility issue around XCTMain

Author: kazk

lib/runners/swift.js

@@ -52,7 +52,7 @@ module.exports.run = function run(opts, cb) {
         args.push('/workspace/setup.swift');
       }
       fs.writeFileSync('/workspace/solution.swift', opts.solution);
-      fs.writeFileSync('/workspace/main.swift', opts.fixture);
+      fs.writeFileSync('/workspace/main.swift', opts.fixture.replace(/^\s*XCTMain\s*\(/m, '_XCTMain('));
       args.push('/workspace/solution.swift', '/workspace/main.swift');
 
       return exec(args.join(' '), (err, stdout, stderr) => {

test/runners/swift_spec.js

@@ -73,7 +73,7 @@ describe('swift runner', function() {
                               }
                             }
 
-                            _XCTMain([
+                            XCTMain([
                                 testCase(CalculatorTest.allTests)
                             ])
                         `,
@@ -87,6 +87,41 @@ describe('swift runner', function() {
       });
     });
 
+    it('should allow whitespace in "XCTMain ("', function(done) {
+      var code = `
+                    func add(_ a: Int, _ b: Int) -> Int {
+                      return a + b
+                    }
+                `;
+      runner.run({
+        language: 'swift',
+        languageVersion: '3',
+        code: code,
+        fixture: `
+                            import XCTest
+                            class AddTest: XCTestCase {
+                              static var allTests = [
+                                    ("testAddCheck", testAddCheck),
+                              ]
+                              func testAddCheck() {
+                                XCTAssertEqual(add(1, 1), 2, "add(1, 1) should be 2")
+                              }
+                            }
+
+                            XCTMain  ([
+                                testCase(AddTest.allTests)
+                            ])
+                        `,
+        testFramework: 'xctest'
+      }, function(buffer) {
+        expect(buffer.stdout).to.contain('<DESCRIBE::>AddTest');
+        expect(buffer.stdout).to.contain('<IT::>testAddCheck');
+        expect(buffer.stdout).to.contain('<PASSED::>Test Passed');
+        expect(buffer.stdout).to.contain('<COMPLETEDIN::>');
+        done();
+      });
+    });
+
     it('should replace newlines in error messages with <:LF:>', function(done) {
       var code = `
                     class Calculator {
@@ -113,7 +148,7 @@ describe('swift runner', function() {
                               }
                             }
 
-                            _XCTMain([
+                            XCTMain([
                                 testCase(CalculatorTest.allTests)
                             ])
                         `,
@@ -174,7 +209,7 @@ describe('swift runner', function() {
                               }
                             }
 
-                            _XCTMain([
+                            XCTMain([
                                 testCase(CalculatorTest.allTests)
                             ])
                         `,
@@ -227,7 +262,7 @@ describe('swift runner', function() {
                               }
                             }
 
-                            _XCTMain([
+                            XCTMain([
                                 testCase(CalculatorTest.allTests)
                             ])
                         `,
@@ -274,7 +309,7 @@ describe('swift runner', function() {
                               }
                             }
 
-                            _XCTMain([
+                            XCTMain([
                                 testCase(GreetingsTest.allTests)
                             ])
                         `,
@@ -332,7 +367,7 @@ describe('swift runner', function() {
                                 }
                             }
 
-                            _XCTMain([
+                            XCTMain([
                                 testCase(CalculatorTest.allTests),
                                 testCase(PersonTest.allTests),
                             ])