Merge pull request #9 from tsabend/f/omit-tests

Omit tests from the search

lib/fus/swift_class.rb

@@ -6,7 +6,7 @@ def initialize(name)
     end
 
     def spec?
-      name.match(/Spec/)
+      name.match(/Spec|Tests|Test/)
     end
 
     def matches_classname?(path)

spec/fixtures/FooSpec.swift

@@ -1,3 +1,3 @@
-class FooSpec {
-
-}
+class FooSpec {}
+class FooTest {}
+class FooTests {}

spec/fus_spec.rb

@@ -12,11 +12,11 @@
       expect(swift_classnames).to include(
         "Foo", "ClassVar", "SuperFoo", 
         "NoSpaceSuperFoo", "UnusedClass", 
-        "ObjCH", "ObjCM", "FooSpec", 
+        "ObjCH", "ObjCM", "FooSpec", "FooTest", "FooTests",
         "UsedInStoryboardView", "UsedInStoryboardViewController", 
         "UsedInXib", "ObjCHForwardDeclarationOnly", "TypeAliasClass", "AssociatedTypeClass"
       )
-      expect(swift_classnames.count).to eq(14)
+      expect(swift_classnames.count).to eq(16)
     end
   end
 

spec/swift_class_spec.rb

@@ -6,8 +6,18 @@
       spec = Fus::SwiftClass.new("FooSpec")
       expect(spec.spec?).to be_truthy
     end
+
+    it "returns true if the classname ends in Test" do
+      spec = Fus::SwiftClass.new("FooTest")
+      expect(spec.spec?).to be_truthy
+    end
+
+    it "returns true if the classname ends in Tests" do
+      spec = Fus::SwiftClass.new("FooTests")
+      expect(spec.spec?).to be_truthy
+    end
 
-    it "returns false if the classname doesn't end in Spec" do
+    it "returns false if the classname doesn't end in Spec, Test or Tests" do
       not_spec = Fus::SwiftClass.new("FooBar")
       expect(not_spec.spec?).to be_falsy
     end