From 482e22960391b02e3630ae3af4dce64d16dd63e7 Mon Sep 17 00:00:00 2001
From: Robert Widmann <devteam.codafi@gmail.com>
Date: Wed, 20 Sep 2017 16:25:13 -0400
Subject: [PATCH] Add more multi-prefix tests

---
 Sources/FileCheck/Pattern.swift            |  4 +++
 Tests/FileCheckTests/DAGSpec.swift         |  8 +++++
 Tests/FileCheckTests/EmptySpec.swift       |  7 +++++
 Tests/FileCheckTests/FileCheckSpec.swift   | 26 ++++++++++++++++
 Tests/FileCheckTests/LabelSpec.swift       |  8 +++++
 Tests/FileCheckTests/MultiPrefixSpec.swift | 36 ++++++++++++++++++++++
 Tests/LinuxMain.swift                      |  4 +++
 7 files changed, 93 insertions(+)

diff --git a/Sources/FileCheck/Pattern.swift b/Sources/FileCheck/Pattern.swift
index 963ad20..b1f76c5 100644
--- a/Sources/FileCheck/Pattern.swift
+++ b/Sources/FileCheck/Pattern.swift
@@ -162,7 +162,11 @@ final class Pattern {
         patternStr = String(patternStr[patternStr.index(end, offsetBy: 4)...])
 
         // Get the regex name (e.g. "foo").
+        #if os(macOS)
         let nameEnd = matchStr.range(of: ":")
+        #else
+        let nameEnd = String(matchStr).range(of: ":")
+        #endif
         let name : String
         if let end = nameEnd?.lowerBound {
           name = String(matchStr[..<end])
diff --git a/Tests/FileCheckTests/DAGSpec.swift b/Tests/FileCheckTests/DAGSpec.swift
index 0281b48..4493bee 100644
--- a/Tests/FileCheckTests/DAGSpec.swift
+++ b/Tests/FileCheckTests/DAGSpec.swift
@@ -143,4 +143,12 @@ class DAGSpec : XCTestCase {
       ].joined(separator: "\n"))
     })
   }
+
+  #if !os(macOS)
+  static var allTests = testCase([
+    ("testPrefixOrderInvariant", testPrefixOrderInvariant),
+    ("testDAGWithInst", testDAGWithInst),
+    ("testDAGXFailWithInst", testDAGXFailWithInst),
+  ])
+  #endif
 }
diff --git a/Tests/FileCheckTests/EmptySpec.swift b/Tests/FileCheckTests/EmptySpec.swift
index 632df46..bc4741c 100644
--- a/Tests/FileCheckTests/EmptySpec.swift
+++ b/Tests/FileCheckTests/EmptySpec.swift
@@ -16,4 +16,11 @@ class EmptySpec : XCTestCase {
       XCTAssert(fileCheckOutput(options: [.allowEmptyInput]) {})
     })
   }
+
+  #if !os(macOS)
+  static var allTests = testCase([
+    ("testAllowEmpty", testAllowEmpty),
+    ("testEmptyError", testEmptyError),
+  ])
+  #endif
 }
diff --git a/Tests/FileCheckTests/FileCheckSpec.swift b/Tests/FileCheckTests/FileCheckSpec.swift
index 5e92956..9a434fd 100644
--- a/Tests/FileCheckTests/FileCheckSpec.swift
+++ b/Tests/FileCheckTests/FileCheckSpec.swift
@@ -140,13 +140,39 @@ class FileCheckSpec : XCTestCase {
     })
   }
 
+  func testNotDiagInfo() {
+    XCTAssert(fileCheckOutput(of: .stdout, withPrefixes: ["NOTDIAGINFO-TEXT"]) {
+      // NOTDIAGINFO-TEXT: error: NOTDIAGINFO-NOT: string occurred!
+      // NOTDIAGINFO-TEXT-NEXT: test
+      // NOTDIAGINFO-TEXT-NEXT: note: NOTDIAGINFO-NOT: pattern specified here
+      XCTAssertFalse(fileCheckOutput(of: .stdout, withPrefixes: ["NOTDIAGINFO"], options: .disableColors) {
+        // NOTDIAGINFO-NOT: test
+        print("test")
+      })
+    })
+  }
+
+  func testNonExistentPrefix() {
+    XCTAssert(fileCheckOutput(of: .stdout, withPrefixes: ["CHECK-NONEXISTENT-PREFIX-ERR"]) {
+      // CHECK-NONEXISTENT-PREFIX-ERR: error: no check strings found with prefixes
+      // CHECK-NONEXISTENT-PREFIX-ERR-NEXT: CHECK-NONEXISTENT-PREFIX{{:}}
+      XCTAssertFalse(fileCheckOutput(of: .stdout, withPrefixes: ["CHECK-NONEXISTENT-PREFIX"], options: [.disableColors]) {
+        // A-DIFFERENT-PREFIX: foobar
+        print("foobar")
+      })
+    })
+  }
+
   #if !os(macOS)
   static var allTests = testCase([
     ("testWhitespace", testWhitespace),
     ("testSame", testSame),
+    ("testCheckDAG", testCheckDAG),
     ("testImplicitCheckNot", testImplicitCheckNot),
     ("testUndefinedVariablePattern", testUndefinedVariablePattern),
     ("testNearestPattern", testNearestPattern),
+    ("testNotDiagInfo", testNotDiagInfo),
+    ("testNonExistentPrefix", testNonExistentPrefix),
   ])
   #endif
 }
diff --git a/Tests/FileCheckTests/LabelSpec.swift b/Tests/FileCheckTests/LabelSpec.swift
index 119c415..c60dc50 100644
--- a/Tests/FileCheckTests/LabelSpec.swift
+++ b/Tests/FileCheckTests/LabelSpec.swift
@@ -82,4 +82,12 @@ class LabelSpec : XCTestCase {
       print(["bar", "foo", "foo", "zed"].joined(separator: "\n"))
     })
   }
+
+  #if !os(macOS)
+  static var allTests = testCase([
+    ("testLabels", testLabels),
+    ("testLabelFail", testLabelFail),
+    ("testLabelDAG", testLabelDAG),
+  ])
+  #endif
 }
diff --git a/Tests/FileCheckTests/MultiPrefixSpec.swift b/Tests/FileCheckTests/MultiPrefixSpec.swift
index f7054f3..f89ec69 100644
--- a/Tests/FileCheckTests/MultiPrefixSpec.swift
+++ b/Tests/FileCheckTests/MultiPrefixSpec.swift
@@ -52,5 +52,41 @@ class MultiPrefixSpec : XCTestCase {
       })
     })
   }
+
+  func testMultiplePrefix2() {
+    XCTAssert(fileCheckOutput(of: .stdout, withPrefixes: ["BAZ", "BAZQUUX", "QUUXBAZ"], options: [.disableColors]) {
+      // BAZ
+      // BAZQUUX
+      // BAZQUUX: this is the {{match}}
+      // QUUXBAZ: this is {{another}}
+      print("""
+            this is the match
+            this is another
+            """)
+    })
+  }
+
+  func testMultiplePrefixOverlap() {
+    XCTAssert(fileCheckOutput(of: .stdout, withPrefixes: ["AAAOVERLAP", "OVERLAP"]) {
+      // OVERLAP: fo{{o}}
+      // AAAOVERLAP: ba{{r}}
+      // OVERLAP: buz{{z}}
+      print("""
+            foo
+            bar
+            buzz
+            """)
+    })
+  }
+
+  #if !os(macOS)
+  static var allTests = testCase([
+    ("testMultiplePrefixSubstr", testMultiplePrefixSubstr),
+    ("testMultiPrefixMixed", testMultiPrefixMixed),
+    ("testMultiplePrefixNoMatch", testMultiplePrefixNoMatch),
+    ("testMultiplePrefix2", testMultiplePrefix2),
+    ("testMultiplePrefixOverlap", testMultiplePrefixOverlap),
+  ])
+  #endif
 }
 
diff --git a/Tests/LinuxMain.swift b/Tests/LinuxMain.swift
index 2258cdc..feb2604 100644
--- a/Tests/LinuxMain.swift
+++ b/Tests/LinuxMain.swift
@@ -4,7 +4,11 @@ import XCTest
 
 #if !os(macOS)
 XCTMain([
+  DAGSpec.allTests,
+  EmptySpec.allTests,
   FileCheckSpec.allTests,
+  LabelSpec.allTests,
   LineCountSpec.allTests,
+  MultiPrefixSpec.allTests,
 ])
 #endif