after all / before all

Readme.md

@@ -11,6 +11,7 @@ Features
  - **red-green** output
  - `mtest` executable to **run by line number** and by folder
  - full backtrace for errors and assertions with verbose (`-v`)
+ - `before :all` + `after :all` with `require 'maxitest/before_and_after_all'`
  - `let!`
  - `order_dependent!` to make your tests run in given order
  - `context` for more expression

lib/maxitest/before_and_after_all.rb

@@ -0,0 +1,52 @@
+module Maxitest
+  def self.before_all_ran
+    @before_all_ran ||= {}
+  end
+end
+
+Minitest::Spec::DSL.class_eval do
+  alias_method :before_without_all, :before
+
+  def before(*args, &block)
+    if args.first == :all
+      id = rand(99999999999999)
+      before_without_all(*args[1..-1]) do
+        ran = Maxitest.before_all_ran
+        if variables = ran[id]
+          variables.each { |k,v,| instance_variable_set(k,v) }
+        else
+          old_variables = instance_variables
+          instance_exec(&block)
+          new_variables = instance_variables - old_variables - [:@_memoized]
+          ran[id] = new_variables.map { |k| [k, instance_variable_get(k)] }
+        end
+      end
+    else
+      before_without_all(*args, &block)
+    end
+  end
+
+  alias_method :after_without_all, :after
+
+  def after(*args, &block)
+    if args.first == :all
+      after_without_all(*args[1..-1]) do
+        c = self.class
+        runnable_methods_count = if c.class_variable_defined?(:@@runnable_methods)
+          c.class_variable_get(:@@runnable_methods)
+        else
+          c.class_variable_set(:@@runnable_methods, c.runnable_methods.size)
+        end
+
+        current = c.instance_variable_get(:@after_ran_count) || 0
+        if current == runnable_methods_count
+          instance_exec(&block)
+        else
+          c.instance_variable_set(:@after_ran_count, current + 1)
+        end
+      end
+    else
+      after_without_all(*args, &block)
+    end
+  end
+end

spec/cases/before_and_after_all.rb

@@ -0,0 +1,61 @@
+require "./spec/cases/helper"
+require "maxitest/before_and_after_all"
+
+describe 2 do
+  order_dependent!
+
+  let(:from_let) { "from let" }
+
+  before :all do
+    puts "BEFORE ALL A -- #{from_let}"
+    @before_all = true
+  end
+
+  after :all do
+    puts "AFTER ALL A -- #{from_let}"
+  end
+
+  before do
+    puts "BEFORE A -- #{@before_all}"
+  end
+
+  after do
+    puts "AFTER A"
+  end
+
+  it "a 1" do
+    true.must_equal true
+  end
+
+  describe "nested" do
+    order_dependent!
+
+    before :all do
+      puts "BEFORE ALL B -- #{from_let}"
+    end
+
+    after :all do
+      puts "AFTER ALL B -- #{from_let}"
+    end
+
+    before do
+      puts "BEFORE B -- #{@before_all}"
+    end
+
+    after do
+      puts "AFTER B"
+    end
+
+    it "b 1" do
+      true.must_equal true
+    end
+
+    it "b 2" do
+      true.must_equal true
+    end
+  end
+
+  it "a 2" do
+    true.must_equal true
+  end
+end