test_group.rb in petitest-0.2.1

- old
+ new

@@ -1,33 +1,82 @@
 module Petitest
   class TestGroup
     TEST_METHOD_NAME_PREFIX = "test_"
 
     class << self
+      attr_writer :description
+
+      attr_writer :metadata
+
+      attr_writer :nest_level
+
       # @return [Array<Class>]
+      def children
+        @children ||= []
+      end
+
+      # @return [Array<Class>]
       def descendants
-        @@descendants ||= []
+        children.flat_map(&:children)
       end
 
+      # @return [String]
+      def description
+        @description ||= name
+      end
+
+      # @return [String, nil]
+      def full_description
+        descriptions = concrete_test_group_ancestors.reverse.map(&:description)
+        unless descriptions.empty?
+          descriptions.join(" ")
+        end
+      end
+
       # @note Override
-      def inherited(sub_class)
+      def inherited(child)
         super
-        descendants << sub_class
+        children << child
       end
 
+      # @return [Hash{Symbol => Object}]
+      def metadata
+        @metadata ||= {}
+      end
+
+      # @return [Integer]
+      def nest_level
+        @nest_level ||= 0
+      end
+
+      # @param description [String]
+      # @param metadata [Hash{Symbol => Object}]
+      def sub_test_group(description, metadata = {}, &block)
+        child = ::Class.new(self)
+        child.nest_level = nest_level + 1
+        child.description = description
+        child.metadata = self.metadata.merge(metadata)
+        child.undefine_test_methods
+        child.class_eval(&block)
+        child
+      end
+
       # @return [Array<Petit::TestCase>]
       def test_cases
-        descendants.flat_map do |test_group_class|
-          test_group_class.test_methods.map do |test_method|
-            ::Petitest::TestCase.new(
-              test_group_class: test_group_class,
-              test_method: test_method,
-            )
-          end
+        @test_cases ||= test_methods.map do |test_method|
+          ::Petitest::TestCase.new(
+            test_group_class: self,
+            test_method: test_method,
+          )
         end
       end
 
+      # @return [Array<Petit::TestCase>]
+      def test_cases_and_children_test_cases
+        test_cases + children.flat_map(&:test_cases_and_children_test_cases)
+      end
+
       # @return [Array<String>]
       def test_method_names
         public_instance_methods.map(&:to_s).select do |method_name|
           method_name.start_with?(TEST_METHOD_NAME_PREFIX)
         end
@@ -42,20 +91,40 @@
             method_name: method_name.to_s,
             path: unbound_method.source_location[0],
           )
         end
       end
+
+      def undefine_test_methods
+        test_method_names.each do |method_name|
+          undef_method(method_name)
+        end
+      end
+
+      private
+
+      # @return [Array<Class>]
+      def concrete_test_group_ancestors
+        ancestors.each_with_object([]) do |klass, classes|
+          if klass == ::Petitest::TestGroup
+            break classes
+          end
+          if klass.is_a?(::Class)
+            classes << klass
+          end
+        end
+      end
     end
 
     # @param actual_or_message [Object]
     # @param message [String, nil]
     def assert(actual_or_message = nil, message = nil, &block)
       if block
         message = actual_or_message
-        check(message || "Given block returned falsy", &block)
+        check(message || "Expected given block to return truthy", &block)
       else
         actual = actual_or_message
-        check(message || "#{actual.inspect} is not truthy") do
+        check(message || "Expected #{actual.inspect} to be truthy") do
           actual
         end
       end
     end
 