orb 0.1.3 → 0.1.4

data/README.md

@@ -7,9 +7,6 @@ When the runner gets to that line, it will open a REPL where you can
 interact with the environment and build up a test to be written back to
 the file.
 
-Orb is still a work in progress. It does not currently actually write
-back to the test file, but it will format a test to be pasted in.
-
 ## Usage
 
 The first step, after adding `gem "orb"` to your Gemfile, is to
@@ -28,8 +25,8 @@ commands, in addition to the functionality of a typical IRB
 
 `,n`: Give a name to the current test. Called like `,n music is loud`.
 
-`,w`: Print a version of the buffer suitable for pasting back into the
-test file. Eventually this will modify the test file with the new test.
+`,w`: Write the buffer back to the test file in place of the call to
+`orb!`.
 
 Within the context of the REPL, assertions are evaluated immediately,
 rather than being deferred until later in the run. This lets you run

data/lib/orb/drivers/test-unit.rb

@@ -2,11 +2,12 @@ require 'test/unit/testcase'
 
 class Test::Unit::TestCase
   def self.orb!
-    define_method("test_orb_#{rand 100000}") { orb!(self) }
+    source_location = caller[0]
+    define_method("test_orb_#{rand 100000}") { orb!(self, source_location) }
   end
 
-  def orb!(ctx)
-    with_immediate_asserts(ctx) { Orb::Runner.new(ctx).run }
+  def orb!(ctx, source_location = caller[0])
+    with_immediate_asserts(ctx) { Orb::Runner.new(ctx, source_location).run }
   end
 
   def with_immediate_asserts(ctx)

data/lib/orb/pry_extensions.rb

@@ -40,6 +40,8 @@ module Orb
 
     command ",w", "Write the buffer to file" do
       orb_test_buffer.write
+      puts "Test written!"
+      throw(:breakout)
     end
   end
 end

data/lib/orb/runner.rb

@@ -2,18 +2,24 @@ require 'pry'
 
 module Orb
   class Runner
-    attr_reader :test_buffer
+    attr_reader :test_buffer, :source_file, :source_line
+
     class << self
       attr_accessor :current
     end
 
-    def initialize(context)
+    def initialize(context, source_location)
       @context = context
-      @test_buffer = TestBuffer.new
+      @test_buffer = TestBuffer.new(*parse_source_location(source_location))
       setup_pry
       Orb::Runner.current = self
     end
 
+    def parse_source_location(source_location)
+      source_location.scan(/(.*):(\d)+:.*/)
+      [$1, $2.to_i]
+    end
+
     def setup_pry
       Pry.instance_eval do
         if initial_session?

data/lib/orb/test_buffer.rb

@@ -3,6 +3,11 @@ require 'tempfile'
 class TestBuffer < Array
   attr_accessor :name
 
+  def initialize(source_file, source_line)
+    @source_file = source_file
+    @source_line = source_line
+  end
+
   def edit_contents!
     path = "#{Dir::tmpdir}/orb-#{Kernel.rand 1000000}.rb"
     begin
@@ -24,15 +29,30 @@ class TestBuffer < Array
   end
 
   def write
-    str = "".tap do |test|
-      test << "test \"#{name}\" do\n"
+    splice_into_file(generate_test(indentation_level))
+  end
+
+  def generate_test(indentation_level)
+    indent = "  " * indentation_level
+    "".tap do |test|
+      test << indent << "test \"#{name}\" do\n"
       each do |line|
-        test << "  " << line << "\n"
+        test << indent << "  " << line << "\n"
       end
-      test << "end\n"
+      test << indent << "end\n"
     end
+  end
+
+  def indentation_level
+    orb_line = File.readlines(@source_file)[@source_line - 1]
+    orb_line.scan(/(\s*).*/).flatten.first.size / 2
+  end
 
-    puts str
+  def splice_into_file(content)
+    lines = File.readlines(@source_file)
+    index = @source_line - 1
+    lines[index] = content
+    File.open(@source_file, "w") { |f| f.write lines.join("") }
   end
 
 end

data/lib/orb/version.rb

@@ -1,3 +1,3 @@
 module Orb
-  VERSION = "0.1.3"
+  VERSION = "0.1.4"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: orb
 version: !ruby/object:Gem::Version
-  version: 0.1.3
+  version: 0.1.4
   prerelease: 
 platform: ruby
 authors:
@@ -13,7 +13,7 @@ date: 2012-02-15 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: pry
-  requirement: &70161996083800 !ruby/object:Gem::Requirement
+  requirement: &70232544483980 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - =
@@ -21,10 +21,10 @@ dependencies:
         version: 0.9.8.2
   type: :runtime
   prerelease: false
-  version_requirements: *70161996083800
+  version_requirements: *70232544483980
 - !ruby/object:Gem::Dependency
   name: test-unit
-  requirement: &70161995983400 !ruby/object:Gem::Requirement
+  requirement: &70232544483480 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -32,7 +32,7 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *70161995983400
+  version_requirements: *70232544483480
 description: REPL-driven testing for ruby -- integrates with pry and lets you build
   tests incrementally in the REPL
 email: