cutest-cj 1.3.0 → 1.3.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 14a96c44d653cd8a8372783b6f4cb0907715402f
-  data.tar.gz: 584bf8ef2bed9fc3a40c09f5a2f3c47497ab6aae
+  metadata.gz: 5e0da3c613aef3fc93c3ac27ed1b87d7c6692e48
+  data.tar.gz: b88f125813019b0f2fad5f17d8ffa01d906bff7a
 SHA512:
-  metadata.gz: 733d91f8afd0fc4843bb324c098d5434c20298bc670e6834b2e8f9336825108d1b5b24cf80f76ee5454da56c8f912b73071a5eb6fdc7586238c126a3533db18d
-  data.tar.gz: de42a0fdce510e763bf37428f5147fc7b3f6998efcab15f5117f63507aba2ab6cd19ac2c343ee23e3237c8ea75b03be1b57094c8846d3829692c3ba57ce8c2b5
+  metadata.gz: 869dd7f32331cef95d6d466ed1352e6f1cfdb533d9eab5fde59a347d9c899b7df7348977575117098ad7a299e5f621cd09b518d68d93216a980fe7881c7bc0e5
+  data.tar.gz: 93e979f6511b572a51533f3271e259c380e8cf25672a2fb9f552b5bcda53e26dd171066586cda7c4d28491dd89efd90a0394e4bffba6fe9991d60d39c8862ad8

data/bin/cutest

@@ -7,9 +7,9 @@ if ARGV.empty?
   exit
 end
 
-require_relative "../lib/cutest"
 require "clap"
 require "mocoso"
+require_relative "../lib/cutest"
 
 files = Clap.run ARGV,
   "-r"      => lambda { |file| require file },
@@ -25,7 +25,7 @@ files = Clap.run ARGV,
 if files.any?
   begin
     env      = '.env'
-    rack_env = ".#{env}.#{ENV['RACK_ENV']}"
+    rack_env = "#{env}.#{ENV['RACK_ENV']}"
 
     if File.file? rack_env
       Cutest.load_envs rack_env

data/lib/cutest.rb

@@ -1,112 +1,131 @@
 require 'benchmark'
+require 'ostruct'
 
 class Cutest
+  autoload :Database, 'database'
+
   unless defined?(VERSION)
-    VERSION = "1.3.0"
+    VERSION = "1.3.1"
     FILTER = %r[/(ruby|jruby|rbx)[-/]([0-9\.])+]
     CACHE = Hash.new { |h, k| h[k] = File.readlines(k) }
   end
 
-  def self.load_envs env
-    File.foreach env do |line|
-      key, value = line.split "="
-      ENV[key] = value.gsub('\n', '').strip
+  class AssertionFailed < StandardError; end
+
+  class << self
+
+    attr_accessor :config, :reset_config
+
+    def setup
+      yield config
     end
-  end
 
-  def self.silence_warnings
-    old_verbose, $VERBOSE = $VERBOSE, nil
-    yield
-  ensure
-    $VERBOSE = old_verbose
-  end
+    def config
+      @config || reset_config!
+    end
 
-  def self.run(files)
-    if !cutest[:warnings]
-      Cutest.silence_warnings do
-        Cutest.now_run files
+    def reset_config!
+      @config = OpenStruct.new database: {}
+    end
+
+    def load_envs env
+      File.foreach env do |line|
+        key, value = line.split "="
+        ENV[key] = value.gsub('\n', '').strip
       end
-    else
-      Cutest.now_run files
     end
-  end
 
-  def self.now_run files
-    status = files.all? do |file|
-      run_file(file)
+    def silence_warnings
+      old_verbose, $VERBOSE = $VERBOSE, nil
+      yield
+    ensure
+      $VERBOSE = old_verbose
+    end
 
-      Process.wait2.last.success?
+    def run(files)
+      if !cutest[:warnings]
+        Cutest.silence_warnings do
+          Cutest.now_run files
+        end
+      else
+        Cutest.now_run files
+      end
     end
 
-    puts
+    def now_run files
+      status = files.all? do |file|
+        run_file(file)
 
-    status
-  end
+        Process.wait2.last.success?
+      end
 
-  def self.run_file(file)
-    fork do
-      begin
-        load(file)
-      rescue LoadError, SyntaxError
-        display_error
-        exit 1
-
-      rescue StandardError
-        trace = $!.backtrace
-        pivot = trace.index { |line| line.match(file) }
-
-        puts "  \e[93mTest: \e[0m%s\e[31m✘\e[0m\n" % (cutest[:test] != '' ? "#{cutest[:test]} " : '')
-
-        if pivot
-          other = trace[0..pivot].select { |line| line !~ FILTER }
-          other.reverse.each { |line| display_trace(line) }
-        else
-          display_trace(trace.first)
-        end
+      puts
 
-        display_error
+      status
+    end
 
-        if not cutest[:pry_rescue]
+    def run_file(file)
+      fork do
+        begin
+          load(file)
+        rescue LoadError, SyntaxError
+          display_error
           exit 1
-        else
-          begin
-            Process.waitall
-          rescue ThreadError, Interrupt
-            # Ignore this as it's caused by Process.waitall when using -p
+
+        rescue StandardError
+          trace = $!.backtrace
+          pivot = trace.index { |line| line.match(file) }
+
+          puts "  \e[93mTest: \e[0m%s\e[31m✘\e[0m\n" % (cutest[:test] != '' ? "#{cutest[:test]} " : '')
+
+          if pivot
+            other = trace[0..pivot].select { |line| line !~ FILTER }
+            other.reverse.each { |line| display_trace(line) }
+          else
+            display_trace(trace.first)
+          end
+
+          display_error
+
+          if not cutest[:pry_rescue]
+            exit 1
+          else
+            begin
+              Process.waitall
+            rescue ThreadError, Interrupt
+              # Ignore this as it's caused by Process.waitall when using -p
+            end
           end
         end
       end
     end
-  end
-
-  def self.code(fn, ln)
-    begin
-      CACHE[fn][ln.to_i - 1].strip
-    rescue
-      "(Can't display line)"
-    end
-  end
 
-  def self.display_error
-    if cutest[:backtrace]
-      bt = $!.backtrace
-      bt.each do |line|
-        display_trace line
+    def code(fn, ln)
+      begin
+        CACHE[fn][ln.to_i - 1].strip
+      rescue
+        "(Can't display line)"
       end
     end
 
-    puts "  \033[93m#{$!.class}: \033[31m#{$!.message}"
-    puts ""
-  end
+    def display_error
+      if cutest[:backtrace]
+        bt = $!.backtrace
+        bt.each do |line|
+          display_trace line
+        end
+      end
 
-  def self.display_trace(line)
-    fn, ln = line.split(":")
+      puts "  \033[93m#{$!.class}: \033[31m#{$!.message}"
+      puts ""
+    end
 
-    puts "  → \033[0mfile: #{fn} ↪#{ln}\e[0m"
-    puts "  → \033[90mline: #{code(fn, ln)}\e[0m"
-  end
+    def display_trace(line)
+      fn, ln = line.split(":")
 
-  class AssertionFailed < StandardError
+      puts "  → \033[0mfile: #{fn} ↪#{ln}\e[0m"
+      puts "  → \033[90mline: #{code(fn, ln)}\e[0m"
+    end
   end
 
   class Scope
@@ -184,10 +203,12 @@ module Kernel
     cutest[:test] = name
 
     if !cutest[:only] || cutest[:only] == name
+      print '  '
       time_taken = Benchmark.measure do
         prepare.each { |blk| blk.call }
         block.call(setup && setup.call)
-        end
+      end
+      puts ''
       puts "  \033[93mTest: \033[0m#{cutest[:test]} \033[32m✔\033[0m"
       puts "\e[94m#{time_taken}\033[0m"
     end
@@ -229,6 +250,6 @@ module Kernel
 
   # Executed when an assertion succeeds.
   def success
-    puts "  •"
+    print "•"
   end
 end

data/lib/database.rb

@@ -0,0 +1,79 @@
+class Cutest
+  module Database
+    extend self
+
+    module Helper
+      require 'active_record'
+
+      extend ActiveSupport::Concern
+
+      private
+
+      def factory *attrs
+        attrs.each do |key, value|
+          if key.is_a? Hash
+            key.each do |k, v|
+              send("#{k}=", v) unless send("#{k}").present?
+            end
+          else
+            send("#{key}=", value) unless send("#{key}").present?
+          end
+        end
+
+        if respond_to?(:creator_id) and not creator_id
+          self.creator_id = Cutest.config.creator_id
+        end
+
+        if respond_to?(:updater_id) and not updater_id
+          self.updater_id = Cutest.config.updater_id
+        end
+      end
+    end
+
+    def reset
+      connect
+
+      ignore_tables = %w(schema_migrations)
+
+      if config.key? :ignore_tables
+        ignore_tables.concat config[:ignore_tables]
+      end
+
+      conn   = ActiveRecord::Base.connection
+      tables = conn.execute("show tables").map { |r| r[0]  }
+
+      tables.each do |t|
+        unless ignore_tables.include? t
+          conn.execute("SET FOREIGN_KEY_CHECKS = 0")
+          conn.execute("TRUNCATE #{t}")
+          conn.execute("SET FOREIGN_KEY_CHECKS = 1")
+        end
+      end
+    end
+
+    def connect
+      return if ActiveRecord::Base.connected?
+
+      ActiveRecord::Base.default_timezone = Time.zone
+
+      db = URI.parse config[:url]
+
+      ActiveRecord::Base.establish_connection(
+          adapter:      db.scheme == 'postgres' ? 'postgresql' : db.scheme,
+          encoding:     'utf8',
+          reconnect:    true,
+          database:     db.path[1..-1],
+          host:         db.host,
+          port:         db.port,
+          username:     db.user,
+          password:     db.password,
+      )
+    end
+
+    def config
+      Cutest.config.database
+    end
+  end
+end
+
+ActiveRecord::Base.send :include, Cutest::Database::Helper

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: cutest-cj
 version: !ruby/object:Gem::Version
-  version: 1.3.0
+  version: 1.3.1
 platform: ruby
 authors:
 - Damian Janowski
@@ -10,7 +10,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-06-19 00:00:00.000000000 Z
+date: 2014-06-20 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: pry
@@ -86,6 +86,7 @@ files:
 - bin/cutest
 - cutest.gemspec
 - lib/cutest.rb
+- lib/database.rb
 - test/assert.rb
 - test/assert_equal.rb
 - test/assert_raise.rb