rspec-kickstarter 0.0.6 → 0.0.7

data/bin/rspec-kickstarter

@@ -1,14 +1,32 @@
 #!/usr/bin/env ruby
+#
+# rspec-kickstarter [dir/*.rb]
+#
 
+require 'optparse'
 require 'rspec_kickstarter/generator'
 
-generator = RSpecKickstarter::Generator.new
+force_write = false
+dry_run = false
+spec_dir = './spec'
 
-dir_or_file = ARGV.first
+opt = OptionParser.new
+opt.on('-f', 'Creates new specs or appends on existing specs') { |_| force_write = true }
+opt.on('-n', 'Dry run') { |_| dry_run = true }
+opt.on('-o VAL', 'Output directory') { |dir| spec_dir = dir }
+
+args = opt.parse(ARGV)
+dir_or_file = args.first
+if dir_or_file.nil?
+  puts "Usage: rspec-kickstarter [dir/*.rb]"
+  exit 1
+end
 unless dir_or_file.match(/.rb$/)
   dir_or_file = dir_or_file.gsub(/\/$/, '') + "/**/*.rb"
 end
+
+generator = RSpecKickstarter::Generator.new(spec_dir)
 Dir.glob(dir_or_file).each do |file_path|
-  generator.wite_spec_if_absent(file_path)
+  generator.write_spec(file_path, force_write, dry_run)
 end
 

data/lib/rspec_kickstarter/generator.rb

@@ -37,8 +37,11 @@ module RSpecKickstarter
     end
 
     def instance_name(c)
-      # TODO prefer snake_case
-      c.name.downcase
+      c.name.gsub(/::/, '/').
+        gsub(/([A-Z]+)([A-Z][a-z])/,'\1_\2').
+        gsub(/([a-z\d])([A-Z])/,'\1_\2').
+        tr("-", "_").
+        downcase
     end
 
     def to_param_names_array(params)
@@ -80,7 +83,7 @@ module RSpecKickstarter
       end
     end
 
-    def wite_spec_if_absent(file_path)
+    def write_spec(file_path, force_write = false, dry_run = false)
 
       body = File.read(file_path)
       RDoc::TopLevel.reset()
@@ -99,8 +102,48 @@ module RSpecKickstarter
         puts "#{file_path} skipped (Class/Module not found)."
       else
 
-        self_path = file_path.gsub(/^(lib\/)|(app\/)/, '').gsub(/\.rb$/, '')
-        code = <<SPEC
+        spec_path = spec_dir + '/' + file_path.gsub(/^(lib\/)|(app\/)/, '').gsub(/\.rb$/, '_spec.rb')
+
+        if force_write && File.exist?(spec_path)
+          existing_spec = File.read(spec_path)
+          racking_methods = c.method_list.select { |m| m.visibility == :public }.reject { |m| existing_spec.match(m.name) }
+          if racking_methods.empty? 
+            puts "#{spec_path} skipped."
+          else
+            additional_spec = <<SPEC
+#{racking_methods.map { |method|
+          <<EACH_SPEC
+  # TODO auto-generated
+  describe '#{method.name}' do
+    it 'should work' do#{get_instantiation_code(c, method)}#{get_params_initialization_code(method)}
+      result = #{get_method_invocation_code(c, method)}
+      result.should_not be_nil
+    end
+  end
+EACH_SPEC
+}.join("\n")}
+SPEC
+            last_end_not_found = true
+            code = existing_spec.split("\n").reverse.reject { |line| 
+              if last_end_not_found 
+                last_end_not_found = line.strip.gsub(/#.+$/, '') != "end"
+                true
+              else
+                false
+              end
+            }.reverse.join("\n") + "\n" + additional_spec + "\nend\n"
+            if dry_run
+              puts "----- #{spec_path} -----"
+              puts code
+            else
+              File.open(spec_path, 'w') { |f| f.write(code) }
+            end
+            puts "#{spec_path} modified."
+          end
+
+        else
+          self_path = file_path.gsub(/^(lib\/)|(app\/)/, '').gsub(/\.rb$/, '')
+          code = <<SPEC
 # -*- encoding: utf-8 -*-
 require 'spec_helper'
 require '#{self_path}'
@@ -109,10 +152,11 @@ describe #{get_complete_class_name(c)} do
 
 #{c.method_list.select { |m| m.visibility == :public }.map { |method|
           <<EACH_SPEC
+  # TODO auto-generated
   describe '#{method.name}' do
     it 'should work' do#{get_instantiation_code(c, method)}#{get_params_initialization_code(method)}
       result = #{get_method_invocation_code(c, method)}
-      # result.should_not be_nil
+      result.should_not be_nil
     end
   end
 EACH_SPEC
@@ -120,17 +164,23 @@ EACH_SPEC
 end
 SPEC
 
-        # TODO improve the logic
-        spec_path = spec_dir + '/' + file_path.gsub(/^(lib\/)|(app\/)/, '').gsub(/\.rb$/, '_spec.rb')
-        if File.exist?(spec_path)
-          puts "#{spec_path} already exists."
-        else
-          FileUtils.mkdir_p(File.dirname(spec_path))
-          File.open(spec_path, 'w') { |f| f.write(code) }
-          puts "#{spec_path} created."
+          if File.exist?(spec_path)
+            puts "#{spec_path} already exists."
+          else
+            if dry_run
+              puts "----- #{spec_path} -----"
+              puts code
+            else
+              FileUtils.mkdir_p(File.dirname(spec_path))
+              File.open(spec_path, 'w') { |f| f.write(code) }
+              puts "#{spec_path} created."
+            end
+          end
         end
       end
+
     end
+
   end
 end
 

data/lib/rspec_kickstarter/version.rb

@@ -1,3 +1,3 @@
 module RSpecKickstarter
-  VERSION = "0.0.6"
+  VERSION = "0.0.7"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: rspec-kickstarter
 version: !ruby/object:Gem::Version
-  version: 0.0.6
+  version: 0.0.7
   prerelease: 
 platform: ruby
 authors:
@@ -46,7 +46,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 1.8.23
+rubygems_version: 1.8.24
 signing_key: 
 specification_version: 3
 summary: RSpec kickstarter generates spec files for existing code.