ttnt 0.0.2 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: d71f52f7c676f37872c83efc2a834699fa0b0651
-  data.tar.gz: e2fc54d245e45e14677701c66bc114a0ee881a90
+  metadata.gz: 9947cd4256b1b4745ec7f4d215aa9c690fd48239
+  data.tar.gz: 787357a06a7c10ec0bffecef39f835991ebf04d5
 SHA512:
-  metadata.gz: 7dd2fb1c983347930da189aa08f969ab9167c1b5bda8155def7d95b84bb96afdf0d130cbe2b5a1ca357cc515696eb37bfa513a92c983af654e3448be4ed9e769
-  data.tar.gz: a551fe32c109fc43f47a875416bbf81c7e8fac8607eae68efa3928b314aee602c4d0fe0f5d257f06e8e03d41f2dfca2be433df4584314187f2439f0ae56c9e7d
+  metadata.gz: 79cb9265651ac4148288f8c74cebb20e360323818fce52fb96089dc9ef0ec4478cb9f17b1fad60946adf62f6c99276e3753614c3d8daa31f686bc13c079190ae
+  data.tar.gz: 2f7182fe104b563b933af1c28c51130ff87f6338f8d002d1053d028528e3c5a6eadb4a2bd720686edf0b2663008ffc2c3a74cf6f285c74d1bcd77130992c8ccb

data/.travis.yml

@@ -1,4 +1,8 @@
 language: ruby
 rvm:
-  - 2.2.0
+  - 2.2.2
 before_install: gem install bundler -v 1.10.2
+install:
+  - ./bin/setup
+script:
+  - bundle exec rake test

data/.ttnt/commit_obj.txt

@@ -0,0 +1 @@
+8b4bac9f164b6389d657f4c1de100dccfc245651

data/.ttnt/test_to_code_mapping.json

@@ -0,0 +1 @@
+{"test/fixtures/repositories/fizzbuzz/test/buzz_test.rb":{"lib/ttnt/version.rb":[1,2],"test/fixtures/repositories/fizzbuzz/lib/fizzbuzz.rb":[1,2,4,6,7]},"test/fixtures/repositories/fizzbuzz/test/fizz_test.rb":{"lib/ttnt/version.rb":[1,2],"test/fixtures/repositories/fizzbuzz/lib/fizzbuzz.rb":[1,2,4,5]},"test/fixtures/repositories/fizzbuzz/test/fizzbuzz_test.rb":{"lib/ttnt/version.rb":[1,2],"test/fixtures/repositories/fizzbuzz/lib/fizzbuzz.rb":[1,2,3]},"test/fixtures/repositories/fizzbuzz/test/non_fizzbuzz_test.rb":{"lib/ttnt/version.rb":[1,2],"test/fixtures/repositories/fizzbuzz/lib/fizzbuzz.rb":[1,2,4,6,9]},"test/lib/tasks_test.rb":{"lib/ttnt/version.rb":[1,2],"test/test_helper.rb":[1,2,4,6,7,8,10,14,18,20],"lib/ttnt.rb":[1,3],"lib/ttnt/tasks.rb":[1,2,3,4,5,8,9,10,12,13,16,17,18,19,35,36,56],"lib/ttnt/testtask.rb":[1,2,3,5,9,21,22,24,26],"lib/ttnt/test_selector.rb":[1,2,3,5,6,7,18]},"test/lib/test_selector_test.rb":{"lib/ttnt/version.rb":[1,2],"test/test_helper.rb":[1,2,4,6,7,8,10,11,14,15,18,20,21,22,23,24,25,26,28],"lib/ttnt.rb":[1,3],"lib/ttnt/test_selector.rb":[1,2,3,5,6,7,8,9,13,14,15,18,19,20,23,24,25,26,28,29,33,35,42]},"test/lib/test_to_code_mapping_test.rb":{"lib/ttnt/version.rb":[1,2],"test/test_helper.rb":[1,2,4,6,7,8,10,11,14,15,18,20,21,22,23,24,25,26,28],"lib/ttnt.rb":[1,3]},"test/lib/testtask_test.rb":{"lib/ttnt/version.rb":[1,2],"test/test_helper.rb":[1,2,4,6,7,8,10,14,18,20],"lib/ttnt.rb":[1,3],"lib/ttnt/testtask.rb":[1,2,3,5,6,9,21,22,24,26,27,28,29,33,34]},"test/ttnt_test.rb":{"lib/ttnt/version.rb":[1,2],"test/test_helper.rb":[1,2,4,6,7,8,10,14,18,20],"lib/ttnt.rb":[1,3]}}

data/.yardopts

@@ -0,0 +1,4 @@
+--protected
+--private
+--list-undoc
+--readme README.md

data/Gemfile

@@ -2,3 +2,7 @@ source 'https://rubygems.org'
 
 # Specify your gem's dependencies in ttnt.gemspec
 gemspec
+
+group :development do
+  gem "coveralls", require: false
+end

data/README.md

@@ -1,5 +1,10 @@
 # TTNT: Test This, Not That!
 
+[![Build Status](https://travis-ci.org/Genki-S/ttnt.svg?branch=master)](https://travis-ci.org/Genki-S/ttnt)
+[![Coverage Status](https://coveralls.io/repos/Genki-S/ttnt/badge.svg?branch=master)](https://coveralls.io/r/Genki-S/ttnt?branch=master)
+[![Code Climate](https://codeclimate.com/github/Genki-S/ttnt/badges/gpa.svg)](https://codeclimate.com/github/Genki-S/ttnt)
+[![Dependency Status](https://gemnasium.com/Genki-S/ttnt.svg)](https://gemnasium.com/Genki-S/ttnt)
+
 Developing under [Google Summer of Code 2015](http://www.google-melange.com/gsoc/homepage/google/gsoc2015) with mentoring organization [Ruby on Rails](http://rubyonrails.org/).
 
 Full proposal of this project is [here](https://github.com/Genki-S/gsoc2015/blob/master/proposal.md) (might be obsolete in some ways, read this README about this project itself rather than the proposal).
@@ -33,14 +38,6 @@ Currently, this program does:
 - Generate test-to-code mapping for a given commit
 - Given base commit and target commit, output test files you should run
 
-Limitations:
-
-- Test selection algorithm is not perfect yet (it may produce false-positives and false-negatives)
-- Only supports git
-- Only supports MiniTest
-- Only select test files, not fine-grained test cases
-- And a lot more!
-
 ## Roadmap
 
 Roadmap is under construction.
@@ -61,38 +58,63 @@ Or install it yourself as:
 
     $ gem install ttnt
 
+### Define rake tasks
+
+You can define TTNT rake tasks by following steps:
+
+1. `require 'ttnt/testtask'`
+2. Define `TTNT::TestTask` when defining `Rake::TestTask`
+
+Your `Rakefile` will look like this:
+
+```
+require 'rake/testtask'
+require 'ttnt/testtask'
+
+Rake::TestTask.new { |t|
+  t.name = 'my_test_name'
+  t.libs << 'test'
+  t.pattern = 'test/**/*_test.rb'
+  TTNT::TestTask.new(t)
+}
+```
+
+This will define 2 rake tasks `ttnt:my_test_name:anchor` and `ttnt:my_test_name:run` (portion of `my_test_name` depends on the name you specify for your `Rake::TestTask`).
+Usage for those tasks are described later in this document.
+
 ## Requirements
 
-Developed and only tested under ruby version 2.2.1.
+Developed and only tested under ruby version 2.2.2.
 
 ## Usage
 
 ### Produce test-to-code mapping for a given commit
 
-This pre-computation is required to select tests later.
-
-Basically, you can substitute `ruby` command with `ttnt-anchor` command.
-For example, if you want to produce test-to-code mapping for `test/mail_layout_test.rb` in `ActionMailer` of rails/rails:
+If you defined TTNT rake task as described above, you can run following command to produce test-to-code mapping:
 
 ```
-$ cd /your/local/rails/rails
-$ git checkout BASE_COMMIT
-$ cd actionmailer
-$ ttnt-anchor -w -Itest test/mail_layout_test.rb -n test_explicit_class_layout
+$ rake ttnt:my_test_name:anchor
 ```
 
-(this example is taken from ["Contributing to Ruby on Rails — Ruby on Rails Guides"](http://edgeguides.rubyonrails.org/contributing_to_ruby_on_rails.html#running-tests)).
-
-This will produce test-to-code mapping for the test file `test/mail_layout_test.rb` under `.ttnt/BASE_COMMIT_SHA/test_to_code_mapping/TEST_FILE_NAME.json` which resides in your project base directory (the same repository in which `.git` directory resides).
-
 ### Select tests
 
+If you defined TTNT rake task as described above, you can run following command to run selected tests.
+
 ```
-$ git checkout TARGET_COMMIT
-$ ttnt BASE_COMMIT
+$ rake ttnt:my_test_name:run
 ```
 
-This will assume test-to-code mapping is properly produced for the BASE\_COMMIT.
+## Current Limitations
+
+- Test selection algorithm is not perfect yet (it may produce false-positives and false-negatives)
+- Only supports git
+- Only supports MiniTest
+- Only select test files, not fine-grained test cases
+- And a lot more!
+
+This gem can only produce test-to-code mapping "from a single test file to code lines executed"
+(not fine-grained mapping "from a single test **case** to code lines executed").
+This is due to the way Ruby's coverage library works. Details are covered in [my proposal](https://github.com/Genki-S/gsoc2015/blob/master/proposal.md#2-run-each-test-case-from-scratch-requiring-all-files-for-every-run).
 
 ## Development
 
@@ -104,7 +126,7 @@ To install this gem onto your local machine, run `bundle exec rake install`. To
 
 Bug reports and pull requests are welcome on GitHub at https://github.com/Genki-S/ttnt. This project is intended to be a safe, welcoming space for collaboration, and contributors are expected to adhere to the [Contributor Covenant](contributor-covenant.org) code of conduct.
 
-Please don't hesitate to [open a issue](https://github.com/Genki-S/ttnt/issues/new) to share your ideas for me! Any comment will be valuable especially in this early development stage.
+I really :heart: getting interesting ideas, so please don't hesitate to [open a issue](https://github.com/Genki-S/ttnt/issues/new) to share your ideas for me! Any comment will be valuable especially in this early development stage. I am collecting interesting ideas which I cannot start working on soon [here at Trello](https://trello.com/b/z232DXnq/ttnt).
 
 ## License
 

data/Rakefile

@@ -1,10 +1,13 @@
 require "bundler/gem_tasks"
 require "rake/testtask"
+require "ttnt/testtask"
 
 Rake::TestTask.new(:test) do |t|
   t.libs << "test"
   t.libs << "lib"
-  t.test_files = FileList['test/**/*_test.rb']
+  t.warning = true
+  t.test_files = FileList['test/**/*_test.rb'] - FileList['test/fixtures/**/*_test.rb']
+  TTNT::TestTask.new(t)
 end
 
 task :default => :test

data/bin/setup

@@ -5,3 +5,4 @@ IFS=$'\n\t'
 bundle install
 
 # Do any other automated setup that you need to do here
+git submodule update --init

data/lib/ttnt.rb

@@ -1,5 +1,8 @@
 require "ttnt/version"
 
+# Test This, Not That!
+#
+# See {file:README.md} for more details.
 module TTNT
   # Your code goes here...
 end

data/lib/ttnt/anchor.rb

@@ -1,5 +1,6 @@
 require 'coverage'
 require 'ttnt/test_to_code_mapping'
+require 'ttnt/metadata'
 require 'rugged'
 
 test_file = $0
@@ -8,7 +9,13 @@ Coverage.start
 
 at_exit do
   # Use current HEAD
-  sha = Rugged::Repository.discover('.').head.target_id
-  mapping = TTNT::TestToCodeMapping.new(sha)
+  repo = Rugged::Repository.discover('.')
+  sha = repo.head.target_id
+  mapping = TTNT::TestToCodeMapping.new(repo)
   mapping.append_from_coverage(test_file, Coverage.result)
+  mapping.write!
+
+  metadata = TTNT::MetaData.new(repo)
+  metadata['anchored_commit'] = sha
+  metadata.write!
 end

data/lib/ttnt/metadata.rb

@@ -0,0 +1,31 @@
+require 'ttnt/storage'
+
+module TTNT
+  class MetaData
+    STORAGE_SECTION = 'meta'
+
+    # @param repo [Rugged::Repository]
+    # @param sha [String] sha of commit from which metadata is read from.
+    #   nil means to read from current working tree. see {Storage} for more.
+    def initialize(repo, sha = nil)
+      @storage = Storage.new(repo, sha)
+      read!
+    end
+
+    def [](name)
+      @data[name]
+    end
+
+    def []=(name, value)
+      @data[name] = value
+    end
+
+    def read!
+      @data = @storage.read(STORAGE_SECTION)
+    end
+
+    def write!
+      @storage.write!(STORAGE_SECTION, @data)
+    end
+  end
+end

data/lib/ttnt/storage.rb

@@ -0,0 +1,63 @@
+module TTNT
+  # A utility class to store TTNT data such as test-to-code mapping and metadata.
+  class Storage
+    # Initialize the storage from given repo and sha. This reads contents from
+    # a file `.ttnt`. When sha is not nil, contents of the file on that commit
+    # is read. Data can be written only when sha is nil (written to current
+    # working tree).
+    #
+    # @param repo [Rugged::Repository]
+    # @param sha [String] sha of the commit from which data should be read.
+    #   nil means reading from/writing to current working tree.
+    def initialize(repo, sha = nil)
+      @repo = repo
+      @sha = sha
+    end
+
+    # Read data in section from the storage.
+    #
+    # @param section [String]
+    # @return [Hash]
+    def read(section)
+      str = read_storage_content
+      if str.length > 0
+        JSON.parse(str)[section] || {}
+      else
+        {}
+      end
+    end
+
+    # Write value to section in the storage.
+    # Locks the file so that concurrent write does not occur.
+    #
+    # @param section [String]
+    # @param value [Hash]
+    def write!(section, value)
+      raise 'Data cannot be written to the storage back in git history' unless @sha.nil?
+      File.open(filename, File::RDWR|File::CREAT, 0644) do |f|
+        f.flock(File::LOCK_EX)
+        str = f.read
+        data = str.length > 0 ? JSON.parse(str) : {}
+        data[section] = value
+        f.rewind
+        f.write(data.to_json)
+        f.flush
+        f.truncate(f.pos)
+      end
+    end
+
+    private
+
+    def filename
+      "#{@repo.workdir}/.ttnt"
+    end
+
+    def read_storage_content
+      if @sha
+        @repo.lookup(@repo.lookup(@sha).tree['.ttnt'][:oid]).content
+      else
+        File.exist?(filename) ? File.read(filename) : ''
+      end
+    end
+  end
+end

data/lib/ttnt/test_selector.rb

@@ -1,40 +1,96 @@
 require 'set'
 require 'rugged'
-require_relative './test_to_code_mapping'
+require 'ttnt/metadata'
+require 'ttnt/test_to_code_mapping'
 
 module TTNT
+  # Select tests using git information and {TestToCodeMapping}.
   class TestSelector
-    def initialize(target_sha, base_sha)
-      @repo = Rugged::Repository.discover('.')
-      @target_obj = @repo.lookup(target_sha)
-      @base_obj = @repo.lookup(base_sha)
-    end
 
-    def select_tests
-      tests = Set.new
-      mapping = TTNT::TestToCodeMapping.new(@base_obj.oid)
-      # TODO: if mapping is not found (ttnt-anchor has not been run)
+    attr_reader :tests
+
+    # @param repo [Rugged::Reposiotry] repository of the project
+    # @param target_sha [String] sha of the target object
+    #   (nil means to target current working tree)
+    # @param test_files [#include?] candidate test files
+    def initialize(repo, target_sha, test_files)
+      @repo = repo
+      @metadata = MetaData.new(repo, target_sha)
+      @target_obj = @repo.lookup(target_sha) if target_sha
+
+      # Base should be the commit `ttnt:anchor` has run on.
+      # NOT the one test-to-code mapping was commited to.
+      @base_obj = find_anchored_commit
+
+      @test_files = test_files
+    end
 
-      diff = @base_obj.diff(@target_obj)
+    # Select tests using differences in anchored commit and target commit
+    # (or current working tree) and {TestToCodeMapping}.
+    #
+    # @return [Set] a set of tests that might be affected by changes in base_sha...target_sha
+    def select_tests!
+      # TODO: if test-to-code-mapping is not found (ttnt-anchor has not been run)
+      @tests ||= Set.new
+      diff = @target_obj ? @base_obj.diff(@target_obj) : @base_obj.diff_workdir
       diff.each_patch do |patch|
         file = patch.delta.old_file[:path]
-        patch.each_hunk do |hunk|
-          # TODO: think if this selection covers all possibilities
-          hunk.each_line do |line|
-            case line.line_origin
-            when :addition
-              # FIXME: new_lineno is suspicious
-              #        (what if hunk1 adds 100 lines and hunk2 add 1 line?)
-              tests += mapping.get_tests(file: file, lineno: line.new_lineno)
-            when :deletion
-              tests += mapping.get_tests(file: file, lineno: line.old_lineno)
-            else
-              # do nothing
+        if test_file?(file)
+          @tests << file
+        else
+          select_tests_from_patch(patch)
+        end
+      end
+      @tests.delete(nil)
+    end
+
+    private
+
+    def mapping
+      sha = @target_obj ? @target_obj.oid : @repo.head.target_id
+      @mapping ||= TTNT::TestToCodeMapping.new(@repo, sha)
+    end
+
+    # Select tests which are affected by the change of given patch.
+    #
+    # @param patch [Rugged::Patch]
+    # @return [Set] set of selected tests
+    def select_tests_from_patch(patch)
+      target_lines = Set.new
+      file = patch.delta.old_file[:path]
+      prev_line = nil
+      patch.each_hunk do |hunk|
+        hunk.each_line do |line|
+          case line.line_origin
+          when :addition
+            if prev_line && !prev_line.addition?
+              target_lines << prev_line.old_lineno
+            elsif prev_line.nil?
+              target_lines << hunk.old_start
             end
+          when :deletion
+            target_lines << line.old_lineno
           end
+
+          prev_line = line
         end
       end
-      tests.delete(nil)
+
+      target_lines.each do |line|
+        @tests += mapping.get_tests(file: file, lineno: line)
+      end
+    end
+
+    # Find the commit `rake ttnt:test:anchor` has been run on.
+    def find_anchored_commit
+      @repo.lookup(@metadata['anchored_commit'])
+    end
+
+    # Check if given file is a test file.
+    #
+    # @param filename [String]
+    def test_file?(filename)
+      @test_files.include?(filename)
     end
   end
 end

data/lib/ttnt/test_to_code_mapping.rb

@@ -1,40 +1,63 @@
+require 'ttnt/storage'
 require 'rugged'
 require 'json'
-
-# Terminologies:
-#   spectra: { filename => [line, numbers, executed], ... }
+require 'set'
 
 module TTNT
+  # Mapping from test file to executed code (i.e. coverage without execution count).
+  #
+  # Terminologies:
+  #   spectra: { filename => [line, numbers, executed], ... }
+  #   mapping: { test_file => spectra }
   class TestToCodeMapping
-    DIRECTORY_SEPARATOR_PLACEHOLDER = '=+='.freeze
+    STORAGE_SECTION = 'mapping'
+
+    attr_reader :mapping
 
-    def initialize(sha)
-      @sha = sha
-      @repo = Rugged::Repository.discover('.')
+    # @param repo [Rugged::Reposiotry] repository to save test-to-code mapping
+    #   (only repo.workdir is used to determine where to save the mapping file)
+    # @param sha [String] sha of commit from which mapping is read.
+    #   nil means to read from current working tree. see {Storage} for more.
+    def initialize(repo, sha = nil)
+      @repo = repo
+      @storage = Storage.new(repo, sha)
+      read!
       raise 'Not in a git repository' unless @repo
     end
 
+    # Append the new mapping to test-to-code mapping file.
+    #
+    # @param test [String] test file for which the coverage data is produced
+    # @param coverage [Hash] coverage data generated using `Coverage.start` and `Coverage.result`
+    # @return [void]
     def append_from_coverage(test, coverage)
-      spectra =  normalize_path(select_project_files(spectra_from_coverage(coverage)))
-      save_spectra(test: test, spectra: spectra)
+      spectra = normalize_paths(select_project_files(spectra_from_coverage(coverage)))
+      @mapping[test] = spectra
     end
 
-    def read_spectra(test:)
-      if File.exists?(spectra_file(test: test))
-        JSON.parse(File.read(spectra_file(test: test)))
-      else
-        {}
-      end
+    # Read test-to-code mapping from storage.
+    def read!
+      @mapping = @storage.read(STORAGE_SECTION)
+    end
+
+    # Write test-to-code mapping to storage.
+    def write!
+      @storage.write!(STORAGE_SECTION, @mapping)
     end
 
+    # Get tests affected from change of file `file` at line number `lineno`
+    #
+    # @param file [String] file name which might have effects on some tests
+    # @param lineno [Integer] line number in the file which might have effects on some tests
+    # @return [Set] a set of test files which might be affected by the change in file at lineno
     def get_tests(file:, lineno:)
       tests = Set.new
-      all_tests.each do |test|
-        spectra = read_spectra(test: test)
+      @mapping.each do |test, spectra|
         lines = spectra[file]
         next unless lines
-        n = lines.bsearch { |x| x >= lineno }
-        if n == lineno
+        topmost = lines.first
+        downmost = lines.last
+        if topmost <= lineno && lineno <= downmost
           tests << test
         end
       end
@@ -43,22 +66,38 @@ module TTNT
 
     private
 
+    # Convert absolute path to relative path from the project (git repository) root.
+    #
+    # @param file [String] file name (absolute path)
+    # @return [String] normalized file path
     def normalized_path(file)
       File.expand_path(file).sub(@repo.workdir, '')
     end
 
-    def normalize_path(spectra)
+    # Normalize all file names in a spectra.
+    #
+    # @param spectra [Hash] spectra data
+    # @return [Hash] spectra whose keys (file names) are normalized
+    def normalize_paths(spectra)
       spectra.map do |filename, lines|
         [normalized_path(filename), lines]
       end.to_h
     end
 
+    # Filter out the files outside of the target project using file path.
+    #
+    # @param spectra [Hash] spectra data
+    # @return [Hash] spectra with only files inside the target project
     def select_project_files(spectra)
       spectra.select do |filename, lines|
         filename.start_with?(@repo.workdir)
       end
     end
 
+    # Generate spectra data from Ruby coverage library's data
+    #
+    # @param cov [Hash] coverage data generated using `Coverage.result`
+    # @return [Hash] spectra data
     def spectra_from_coverage(cov)
       spectra = Hash.new { |h, k| h[k] = [] }
       cov.each do |filename, executions|
@@ -69,31 +108,5 @@ module TTNT
       end
       spectra
     end
-
-    def save_spectra(test:, spectra:)
-      dir = base_savedir
-      unless File.directory?(dir)
-        FileUtils.mkdir_p(dir)
-      end
-      File.write(spectra_file(test: test), spectra.to_json)
-    end
-
-    def base_savedir
-      "#{@repo.workdir}/.ttnt/#{@sha}/test_to_code_mapping"
-    end
-
-    def spectra_file(test:)
-      filename = normalized_path(test).gsub('/', DIRECTORY_SEPARATOR_PLACEHOLDER)
-      "#{base_savedir}/#{filename}.json"
-    end
-
-    def all_tests
-      Dir["#{base_savedir}/*.json"].map do |filename|
-        filename
-          .sub(/.*\//, '')
-          .sub(/\.json\Z/, '')
-          .gsub(DIRECTORY_SEPARATOR_PLACEHOLDER, '/')
-      end
-    end
   end
 end

data/lib/ttnt/testtask.rb

@@ -0,0 +1,119 @@
+require 'rugged'
+require 'rake'
+require 'ttnt/test_selector'
+
+module TTNT
+  # TTNT version of Rake::TestTask.
+  # Uses configuration from Rake::TestTask to minimize user configuration.
+  # Defines TTNT related rake tasks when instantiated.
+  class TestTask
+    include Rake::DSL
+
+    # An instance of `Rake::TestTask` passed when TTNT::TestTask is initialized
+    attr_accessor :rake_testtask
+
+    # Create an instance of TTNT::TestTask and define TTNT rake tasks.
+    #
+    # @param rake_testtask [Rake::TestTask] an instance of Rake::TestTask after user configuration is done
+    def initialize(rake_testtask)
+      @rake_testtask = rake_testtask
+      # Since test_files is not exposed in Rake::TestTask
+      @test_files = @rake_testtask.instance_variable_get('@test_files')
+
+      @anchor_description = 'Generate test-to-code mapping' + (@rake_testtask.name == :test ? '' : " for #{@rake_testtask.name}")
+      @run_description = 'Run selected tests' + (@rake_testtask.name == :test ? '' : "for #{@rake_testtask.name}")
+      define_tasks
+    end
+
+    # Returns array of test file names.
+    #   Unlike Rake::TestTask#file_list, patterns are expanded.
+    def expanded_file_list
+      test_files = Rake::FileList[@rake_testtask.pattern].compact
+      test_files += @test_files.to_a if @test_files
+      test_files
+    end
+
+    private
+
+    # Git repository discovered from current directory
+    #
+    # @return [Rugged::Reposiotry]
+    def repo
+      @repo ||= Rugged::Repository.discover('.')
+    end
+
+    # Define TTNT tasks under namespace 'ttnt:TESTNAME'
+    #
+    # @return [void]
+    def define_tasks
+      # Task definitions are taken from Rake::TestTask
+      # https://github.com/ruby/rake/blob/e644af3/lib/rake/testtask.rb#L98-L112
+      namespace :ttnt do
+        namespace @rake_testtask.name do
+          define_run_task
+          define_anchor_task
+        end
+      end
+    end
+
+    # Define a task which runs only tests which might have affected from changes
+    # between anchored commit and TARGET_SHA.
+    #
+    # TARGET_SHA can be specified as an environment variable (defaults to HEAD).
+    #
+    # @return [void]
+    def define_run_task
+      desc @run_description
+      task 'run' do
+        target_sha = ENV['TARGET_SHA']
+        ts = TTNT::TestSelector.new(repo, target_sha, expanded_file_list)
+        tests = ts.select_tests!
+        if tests.empty?
+          STDERR.puts 'No test selected.'
+        else
+          args =
+            "#{@rake_testtask.ruby_opts_string} #{@rake_testtask.run_code} " +
+            "#{tests.to_a.join(' ')} #{@rake_testtask.option_list}"
+          run_ruby args
+        end
+      end
+    end
+
+    # Define a task which runs test files file by file, and generate and save
+    # test-to-code mapping.
+    #
+    # @return [void]
+    def define_anchor_task
+      desc @anchor_description
+      task 'anchor' do
+        # In order to make it possible to stop coverage services like Coveralls
+        # which interferes with ttnt/anchor because both use coverage library.
+        # See test/test_helper.rb
+        ENV['ANCHOR_TASK'] = '1'
+
+        Rake::FileUtilsExt.verbose(@rake_testtask.verbose) do
+          # Make it possible to require files in this gem
+          gem_root = File.expand_path('../..', __FILE__)
+          args =
+            "-I#{gem_root} -r ttnt/anchor " +
+            "#{@rake_testtask.ruby_opts_string}"
+
+          expanded_file_list.each do |test_file|
+            run_ruby "#{args} #{test_file}"
+          end
+        end
+      end
+    end
+
+    # Run ruby process with given args
+    #
+    # @param args [String] argument to pass to ruby
+    def run_ruby(args)
+      ruby "#{args}" do |ok, status|
+        if !ok && status.respond_to?(:signaled?) && status.signaled?
+          raise SignalException.new(status.termsig)
+        end
+      end
+    end
+  end
+end

data/lib/ttnt/version.rb

@@ -1,3 +1,3 @@
 module TTNT
-  VERSION = "0.0.2"
+  VERSION = "0.0.3"
 end

data/ttnt.gemspec

@@ -33,9 +33,14 @@ Gem::Specification.new do |spec|
   spec.add_development_dependency "bundler", "~> 1.10"
   spec.add_development_dependency "rake", "~> 10.0"
   spec.add_development_dependency "minitest"
+  spec.add_development_dependency "coveralls"
+  spec.add_development_dependency "yard"
 
   # Pry
   spec.add_development_dependency "hirb"
   spec.add_development_dependency "awesome_print"
   spec.add_development_dependency "pry"
+  spec.add_development_dependency "pry-doc"
+  spec.add_development_dependency "pry-byebug"
+  spec.add_development_dependency "pry-rescue"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ttnt
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
 platform: ruby
 authors:
 - Genki Sugimoto
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2015-06-07 00:00:00.000000000 Z
+date: 2015-08-01 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rugged
@@ -80,6 +80,34 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: coveralls
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: yard
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: hirb
   requirement: !ruby/object:Gem::Requirement
@@ -122,17 +150,60 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: pry-doc
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: pry-byebug
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: pry-rescue
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 description: 
 email:
 - cfhoyuk.reccos.nelg@gmail.com
-executables:
-- ttnt
-- ttnt-anchor
+executables: []
 extensions: []
 extra_rdoc_files: []
 files:
 - ".gitignore"
 - ".travis.yml"
+- ".ttnt/commit_obj.txt"
+- ".ttnt/test_to_code_mapping.json"
+- ".yardopts"
 - CODE_OF_CONDUCT.md
 - Gemfile
 - LICENSE.txt
@@ -140,12 +211,13 @@ files:
 - Rakefile
 - bin/console
 - bin/setup
-- exe/ttnt
-- exe/ttnt-anchor
 - lib/ttnt.rb
 - lib/ttnt/anchor.rb
+- lib/ttnt/metadata.rb
+- lib/ttnt/storage.rb
 - lib/ttnt/test_selector.rb
 - lib/ttnt/test_to_code_mapping.rb
+- lib/ttnt/testtask.rb
 - lib/ttnt/version.rb
 - ttnt.gemspec
 homepage: http://github.com/Genki-S/ttnt
@@ -173,3 +245,4 @@ signing_key:
 specification_version: 4
 summary: Select test cases to run based on changes in committed code
 test_files: []
+has_rdoc: 

data/exe/ttnt

@@ -1,28 +0,0 @@
-#!/usr/bin/env ruby
-
-require 'rugged'
-require_relative '../lib/ttnt/test_selector'
-
-USAGE = %{
-ttnt (Test This, Not That)
-
-Usage:
-  ttnt BASE_SHA
-
-What it does:
-  Output test files you should run based on changes between commits BASE_SHA and HEAD.
-}
-
-if ARGV.count != 1
-  puts USAGE
-  exit 1
-end
-
-repo = Rugged::Repository.discover('.')
-ts = TTNT::TestSelector.new(repo.head.target_id, ARGV[0])
-tests = ts.select_tests
-if tests.empty?
-  STDERR.puts 'No test selected.'
-else
-  puts tests.to_a.join("\n")
-end

data/exe/ttnt-anchor

@@ -1,7 +0,0 @@
-#!/usr/bin/env ruby
-
-gem_root = File.expand_path('../lib', File.dirname(File.expand_path(__FILE__)))
-argv = ARGV
-argv.unshift('ruby', '-I', gem_root, '-r', 'ttnt/anchor')
-
-exec(*argv)