milkode 0.2.4 → 0.2.9

data/test/data/a_project/cdstk_yaml.rb

@@ -0,0 +1,130 @@
+# -*- coding: utf-8 -*-
+#
+# @file 
+# @brief
+# @author ongaeshi
+# @date   2011/02/20
+
+require 'yaml'
+require 'pathname'
+require 'milkode/common/dbdir'
+
+module Milkode
+  class CdstkYaml
+    class YAMLAlreadyExist < RuntimeError
+    end
+    
+    class YAMLNotExist < RuntimeError
+    end
+
+    def self.create(path = ".")
+      yf = yaml_file(path)
+      raise YAMLAlreadyExist.new if FileTest.exist? yf
+      obj = CdstkYaml.new(yf, {'contents' => [], 'version' => 0.1})
+      obj.save
+      return obj
+    end
+
+    def self.load(path = ".")
+      yf = yaml_file(path)
+      raise YAMLNotExist.new unless FileTest.exist? yf
+      open(yf) do |f|
+        return CdstkYaml.new(yf, YAML.load(f.read()))
+      end
+    end
+
+    def initialize(yaml_file, data)
+      @yaml_file = yaml_file
+      @data = data
+    end
+
+    def add(dirs)
+      contents.concat(dirs.map{|v|{'directory' => v}})
+      contents.uniq!
+    end
+
+    def remove(query)
+      r = query.select_any?(contents)
+      r.each {|v| contents.delete v}
+    end
+
+    def save
+      open(@yaml_file, "w") { |f| YAML.dump(@data, f) }
+    end
+
+    def contents
+      @data['contents']
+    end
+
+    def package_num
+      @data['contents'].size
+    end
+
+    def directorys
+      contents.map{|v|v['directory']}
+    end
+
+    def version
+      @data['version']
+    end
+
+    def list(query = nil)
+      query ? query.select_all?(contents) : contents
+    end
+
+    def exist?(shortname)
+      @data['contents'].find {|v| File.basename(v['directory']) == shortname }   
+    end
+
+    def cant_add_directory?(dir)
+      @data['contents'].find {|v|
+        v['directory'] != File.expand_path(dir) &&
+        File.basename(v['directory']) == File.basename(dir)
+      }
+    end
+
+    def cleanup
+      contents.delete_if do |v|
+        if (!File.exist? v['directory'])
+          yield v if block_given?
+          true
+        else
+          false
+        end
+      end
+    end
+
+    def package_root(dir)
+      @data['contents'].find do |v|
+        v if dir =~ /^#{v['directory']}/
+      end
+    end
+
+    def package_root_dir(dir)
+      package = package_root(dir)
+      (package) ? package['directory'] : nil
+    end
+
+    def self.yaml_file(path)
+      Dbdir.yaml_path(path)
+    end
+
+    class Query
+      def initialize(keywords)
+        @keywords = keywords
+      end
+
+      def select_any?(contents)
+        contents.find_all do |v|
+          @keywords.any? {|s| File.basename(v['directory']).include? s }
+        end
+      end
+
+      def select_all?(contents)
+        contents.find_all do |v|
+          @keywords.all? {|s| File.basename(v['directory']).include? s }
+        end
+      end
+    end
+  end
+end

data/test/data/b_project/runner.rb

@@ -0,0 +1,11 @@
+# -*- coding: utf-8 -*-
+#
+# @file 
+# @brief
+# @author ongaeshi
+# @date   2011/03/11
+
+require 'test/unit'
+Test::Unit::AutoRunner.run(true, '.')
+
+

data/test/data/b_project/test_dir.rb

@@ -0,0 +1,21 @@
+# -*- coding: utf-8 -*-
+#
+# @file 
+# @brief
+# @author ongaeshi
+# @date   2011/02/20
+
+require 'milkode/common/dir'
+require 'test/unit'
+require 'file_test_utils'
+
+class TestDir < Test::Unit::TestCase
+  include FileTestUtils
+
+  def test_emptydir?
+    assert_equal true, Dir.emptydir?(".")
+    assert_equal false, Dir.emptydir?("..")
+  end
+end
+
+

data/test/milkode_test_work.rb

@@ -0,0 +1,78 @@
+# -*- coding: utf-8 -*-
+#
+# @file 
+# @brief Milkodeテスト用のワーク領域確保
+# @author ongaeshi
+# @date 2011/12/14
+
+require 'test_helper'
+require 'rubygems'
+require 'groonga'
+require 'pathname'
+require 'fileutils'
+require 'milkode/cdstk/cdstk'
+require 'milkode/common/dbdir'
+
+class MilkodeTestWork
+  def initialize(option = nil)
+    @option = option
+
+    create_tmp_dir
+
+    if (@option[:default_db])
+      @old_path = Dbdir.milkode_db_dir
+      path = expand_path(".milkode_db_dir")
+      Dbdir.set_milkode_db_dir path
+      open(Dbdir.milkode_db_dir, "w") {|f| f.print expand_path("db1") }
+    end
+
+    init_db("db1")
+  end
+
+  def init_db(name)
+    dbdir = expand_path(name)
+    FileUtils.mkdir_p dbdir
+    Dir.chdir(dbdir) { cdstk.init }
+  end
+
+  def add_package(name, package_path)
+    dbdir = expand_path(name)
+    
+    Dir.chdir(dbdir) do
+      cdstk.add [package_path]
+    end
+  end
+
+  def teardown
+    FileUtils.rm_rf(@tmp_dir.to_s)
+    Dbdir.set_milkode_db_dir @old_path if (@option[:default_db])
+  end
+
+  def path(path)
+    File.join @tmp_dir.to_s, path
+  end
+  
+  def expand_path(path)
+    File.expand_path path(path)
+  end
+
+  def pwd
+    cdstk.pwd({})
+  end
+  
+  private
+  
+  def create_tmp_dir
+    @tmp_dir = Pathname(File.dirname(__FILE__)) + "milkode_test_work"
+    FileUtils.rm_rf(@tmp_dir.to_s)
+    FileUtils.mkdir_p(@tmp_dir.to_s)
+  end
+
+  def cdstk
+    Cdstk.new(StringIO.new, Dbdir.select_dbdir)
+  end
+
+end
+
+
+

data/test/test_cdstk.rb

@@ -50,10 +50,20 @@ class TestCdstk < Test::Unit::TestCase
       io.puts('--- add notfound ---')
       obj.add(['notfound.html'])
 
-      io.puts('--- update ---')
+      io.puts('--- update_all ---')
       FileUtils.touch('packages/zip/abc/c.txt')
       FileUtils.touch('packages/zip/abc/d.txt')
-      obj.update
+      obj.update_all
+
+      io.puts('--- update --all ---')
+      FileUtils.touch('packages/zip/abc/e.txt')
+      obj.update([], {:all => true})
+
+      io.puts('--- update in package dir ---')
+      Dir.chdir('packages/zip/abc') do
+        # setdbコマンドが無いので上手く動かせない
+        # obj.update([], {})
+      end
 
       io.puts('--- remove ---')
       obj.remove(['findgrep'], {:force => true})

data/test/test_cdstk_command.rb

@@ -0,0 +1,50 @@
+# -*- coding: utf-8 -*-
+#
+# @file 
+# @brief
+# @author ongaeshi
+# @date   2011/02/20
+
+require 'milkode_test_work'
+require 'milkode/cdstk/cdstk_command'
+
+class TestCdstkCommand < Test::Unit::TestCase
+  def setup
+    @work = MilkodeTestWork.new({:default_db => true})
+  end
+
+  def test_main
+    t_setdb_set
+    t_setdb_reset
+  end
+  
+  def teardown
+    @work.teardown
+  end
+
+  private
+
+  def t_setdb_set
+    # デフォルトデータベースの切り替え
+    CdstkCommand.setdb_set(@work.expand_path "db1")
+
+    # デフォルトデータベースの切り替え
+    # @work.init_db( "db2" )
+    # CdstkCommand.setdb_set(@work.expand_path "db2")
+    
+    # 存在していないデータベースを指定するとエラー
+    assert_raise(Milkode::CdstkCommand::NotExistDatabase) { CdstkCommand.setdb_set("not_found") } 
+  end
+
+  def t_setdb_reset
+    path = @work.expand_path '.milkode_db_dir'
+
+    CdstkCommand.setdb_set(@work.expand_path "db1")
+    assert File.exist?(path)
+
+    CdstkCommand.setdb_reset
+    assert !File.exist?(path)
+  end
+end
+
+

data/test/test_cdstk_yaml.rb

@@ -73,14 +73,14 @@ class TestCdstkYaml < Test::Unit::TestCase
     contents = [{d => 'key'}, {d => 'keyword'}, {d => 'not'}]
 
     query = CdstkYaml::Query.new(['key'])
-    assert_equal [{d => 'key'}, {d => 'keyword'}], query.select(contents)
+    assert_equal [{d => 'key'}, {d => 'keyword'}], query.select_any?(contents)
 
     query = CdstkYaml::Query.new(['word'])
-    assert_equal [{d => 'keyword'}], query.select(contents)
+    assert_equal [{d => 'keyword'}], query.select_any?(contents)
 
     contents = [{d => 'a/dir'}, {d => 'b/dia'}]
     query = CdstkYaml::Query.new(['a'])
-    assert_equal [{d => 'b/dia'}], query.select(contents) # ディレクトリ名は含めない
+    assert_equal [{d => 'b/dia'}], query.select_any?(contents) # ディレクトリ名は含めない
   end
 
   def test_list
@@ -142,6 +142,23 @@ EOF
     assert_nil yaml.exist?('dir123')
     assert_nil yaml.exist?('dir')
   end
+
+  def test_package_root
+    src = <<EOF
+version: 0.1
+contents: 
+- directory: /a/dir1
+- directory: /path/to/dir
+- directory: /a/b/c
+EOF
+
+    yaml = CdstkYaml.new('dummy.yaml', YAML.load(src))
+
+    assert_equal nil           , yaml.package_root_dir('/not_dir')
+    assert_equal "/a/dir1"     , yaml.package_root_dir('/a/dir1/dir3')
+    assert_equal nil           , yaml.package_root_dir('/hoge/a/dir1/dir3')
+    assert_equal '/path/to/dir', yaml.package_root_dir('/path/to/dir')
+  end
   
   def teardown
     FileUtils.cd(@prev_dir)

data/test/test_cli_cdstk.rb

@@ -0,0 +1,71 @@
+# -*- coding: utf-8 -*-
+#
+# @file 
+# @brief
+# @author ongaeshi
+# @date   2011/12/03
+
+require 'milkode/cdstk/cli_cdstk.rb'
+require 'milkode/common/dbdir'
+require 'test_helper'
+require 'milkode/cdstk/cdstk_command'
+require 'milkode_test_work'
+
+class TestCLI_Cdstk < Test::Unit::TestCase
+  def setup
+    @first_default_dir = Dbdir.default_dir
+    @work = MilkodeTestWork.new({:default_db => true})
+    @work.add_package "db1", @work.expand_path("../data/a_project")
+  end
+
+  def test_main
+    t_grep
+    t_mcd
+    t_setdb
+  end
+
+  def teardown
+    @work.teardown
+  end
+
+  private
+
+  def t_grep
+    command("grep")
+    command("grep not_found")
+    command("grep require -a")
+  end
+
+  def t_mcd
+    assert_match /mcd/, command("mcd")
+  end
+  
+  def t_setdb
+    # 引数無しで現在の値を表示
+    assert_equal @work.expand_path("db1") + "\n", command("setdb")
+    
+    # .milkode_db_dir を書き換えてテスト
+    open(@work.path(".milkode_db_dir"), "w") {|f| f.print "/a/custom/db" }
+    assert_equal "/a/custom/db\n", command("setdb")
+
+    # データベースではないディレクトリに切り替ようとするとエラー
+    assert_match(/fatal:/, command("setdb /a/write/test"))
+    
+    # 切り替え
+    @work.init_db("db2")
+    assert_match "Set default db", command("setdb #{@work.path "db2"}")
+
+    # リセット
+    assert_not_equal @first_default_dir, Dbdir.default_dir
+    command("setdb --reset")
+    assert_equal @first_default_dir, Dbdir.default_dir
+  end
+
+  private
+
+  def command(arg)
+    io = StringIO.new
+    CLI_Cdstk.execute(io, arg.split)
+    io.string
+  end
+end

data/test/test_cli_grep.rb

@@ -0,0 +1,79 @@
+# -*- coding: utf-8 -*-
+#
+# @file 
+# @brief
+# @author ongaeshi
+# @date   2011/12/03
+
+require 'milkode/grep/cli_grep.rb'
+require 'test_helper'
+require 'milkode_test_work'
+
+class TestCLI_Grep < Test::Unit::TestCase
+  def setup
+    @work = MilkodeTestWork.new({:default_db => true})
+    @work.add_package "db1", @work.expand_path("../data/a_project")
+    @work.add_package "db1", @work.expand_path("../data/b_project")
+  end
+
+  def test_main
+    # 全てを test_* にすると、毎回setup, teardown が走ってデータベースを生成する時間がもったいないので
+    t_basic
+    t_not_found_package
+    t_not_package_root
+    t_exec_onlysuffix
+    t_cache
+  end
+
+  def teardown
+    @work.teardown
+  end
+
+  private
+
+  def t_basic
+    io = StringIO.new
+
+    CLI_Grep.execute(io, "".split)
+    CLI_Grep.execute(io, "-a test".split)
+    CLI_Grep.execute(io, "-p a_project test".split)
+
+    # puts io.string
+  end
+
+  def t_not_found_package
+    io = StringIO.new
+    CLI_Grep.execute(io, "-p c_project test".split)
+    assert_match "fatal:", io.string
+  end
+
+  def t_not_package_root
+    Dir.chdir(File.join(File.dirname(__FILE__))) do
+      io = StringIO.new
+      CLI_Grep.execute(io, "require".split)
+      assert_match "fatal:", io.string
+    end
+    
+    Dir.chdir(File.join(File.dirname(__FILE__), "data/a_project")) do
+      io = StringIO.new
+      CLI_Grep.execute(io, "require".split)
+      assert_no_match /fatal:/, io.string
+    end
+  end
+
+  def t_exec_onlysuffix
+    io = StringIO.new
+    CLI_Grep.execute(io, "-p a_project -s rb".split)
+    assert_match "a_project/cdstk.rb\n", io.string
+  end
+
+  def t_cache
+    io = StringIO.new
+    CLI_Grep.execute(io, "-p a_project test --cache".split)
+    assert_match "grenfiletest", io.string
+  end
+end
+
+
+
+