Changeset 19115

Timestamp:

09/10/08 11:31:41 (4 months ago)

Author:

authorNari

Message:

write readme. add spec test.

Location:

lang/ruby/ymldot

Files:

• README (added)
• lib/ymldot.rb (modified) (5 diffs)
• sample (added)
• sample/sample.dot (added)
• sample/sample.png (added)
• sample/sample.yml (added)
• sample/sample_jp.dot (added)
• sample/sample_jp.png (added)
• sample/sample_jp.yml (added)
• spec/test_config_none.yml (added)
• spec/test_many_relation.yml (added)
• spec/ymldot_spec.rb (modified) (2 diffs)

lang/ruby/ymldot/lib/ymldot.rb

@@ -14 +14 @@
   end
 
-  def to_s
+  def to_dot
     res = ""
     res << %Q!"#{@name}" [#{@dependent? 'shape=Mrecord, ' : 'shape=record, '}label=\"{#{@name}|!
@@ -45 +45 @@
     code += <<"EOS"
 digraph #{@file_name} {
-#{add_2_tab(config_to_s)}
-#{add_2_tab(entities_to_s)}
-#{add_2_tab(relations_to_s)}
+#{add_2_tab(config_to_dot)}
+#{add_2_tab(entities_to_dot)}
+#{add_2_tab(relations_to_dot)}
 }
 EOS
@@ -59 +59 @@
   end
 
-  def config_to_s
+  def config_to_dot
     res = ""
     res << "graph[overlap=false, splines=true]\n"
     res << %Q!node [fontname="#{@config["font"]}"]! if @config["font"]
+    res
   end
 
-  def entities_to_s
+  def entities_to_dot
     res = ""
     s = []
     @entities.each_value{|e| s << e }
-    s.sort{|a, b| a.name <=> b.name}.each{|e| res << e.to_s << "\n"}
+    s.sort{|a, b| a.name <=> b.name}.each{|e| res << e.to_dot << "\n"}
     res
   end
 
-  def relations_to_s
+  def relations_to_dot
     res = ""
     @one_relations.sort{|a, b| a[:self].name <=> b[:self].name}.each do |r|
@@ -85 +86 @@
 
   def eval
-    @config = @node["config"]
+    @config = @node["config"]? @node["config"] : {}
     @node["tables"].each {|t| eval_entity(t) }
     @node["tables"].each {|t| eval_relation(t) }
@@ -98 +99 @@
     foreignkeys = table["foreignkeys"]
     tname = table["name"]
-    case
-    when foreignkeys["belongs_to"]
-      foreignkeys["belongs_to"].each do |rel|
-        @entities[tname].foreignkeys << "#{rel}ID(FK)"
-      end
-    when foreignkeys["has_many"]
-      foreignkeys["has_many"].each do |rel|
-        @many_relations << {:self => @entities[tname], :have => @entities[rel]}
-      end
-    when foreignkeys["has_one"]
-      foreignkeys["has_one"].each do |rel|
-        @one_relations << {:self => @entities[tname], :have => @entities[rel]}
-      end
-    when foreignkeys["has_many_and_belongs_to"]
-      foreignkeys["has_many_and_belongs_to"].each do |rel|
-        join_tname = "#{rel}_#{tname}"
-        return if @entities.has_key? join_tname
-        join_tname = "#{tname}_#{rel}"
-        @entities[join_tname] ||= Entity.new(join_tname, true)
-        @entities[join_tname].foreignkeys << "#{tname}ID(FK)"
-        @entities[join_tname].foreignkeys << "#{rel}ID(FK)"
-        @many_relations << {:self => @entities[tname], :have => @entities[join_tname]}
-        @many_relations << {:self => @entities[rel], :have => @entities[join_tname]}
-      end
+    eval_relation_belongs_to(foreignkeys["belongs_to"], tname) if foreignkeys["belongs_to"]
+    eval_relation_has_many(foreignkeys["has_many"], tname) if foreignkeys["has_many"]
+    eval_relation_has_one(foreignkeys["has_one"], tname) if foreignkeys["has_one"]
+    eval_relation_hmabt(foreignkeys["has_many_and_belongs_to"], tname) if foreignkeys["has_many_and_belongs_to"]
+  end
+
+  def eval_relation_belongs_to(keys, tname)
+    keys.each do |rel|
+      @entities[tname].foreignkeys << "#{rel}ID(FK)"
+    end
+  end
+
+  def eval_relation_has_many(keys, tname)
+    keys.each do |rel|
+      @many_relations << {:self => @entities[tname], :have => @entities[rel]}
+    end
+  end
+
+  def eval_relation_has_one(keys, tname)
+    keys.each do |rel|
+      @one_relations << {:self => @entities[tname], :have => @entities[rel]}
+    end
+  end
+
+  def eval_relation_hmabt(keys, tname)
+    keys.each do |rel|
+      join_tname = "#{rel}_#{tname}"
+      return if @entities.has_key? join_tname
+      join_tname = "#{tname}_#{rel}"
+      @entities[join_tname] ||= Entity.new(join_tname, true)
+      @entities[join_tname].foreignkeys << "#{tname}ID(FK)"
+      @entities[join_tname].foreignkeys << "#{rel}ID(FK)"
+      @many_relations << {:self => @entities[tname], :have => @entities[join_tname]}
+      @many_relations << {:self => @entities[rel], :have => @entities[join_tname]}
     end
   end

lang/ruby/ymldot/spec/ymldot_spec.rb

@@ -3 +3 @@
 
 class Ymldot
-  attr_accessor :node, :entities, :one_relations, :many_relations, :file_name
+  attr_accessor :node, :entities, :one_relations, :many_relations, :file_name, :config
 end
 
@@ -137 +137 @@
   end
 end
+
+describe Ymldot, "で読み込んだ.ymlにconfigがない場合" do
+  before do
+    @ymldot = Ymldot.new("test_config_none.yml")
+  end
+
+  it "#config は空のHashが格納されていること" do
+    @ymldot.config.should be_empty
+  end
+end
+
+describe Ymldot, "で読み込んだ.ymlにて一つのに複数の関係を持っている場合" do
+  before do
+    @ymldot = Ymldot.new("test_many_relation.yml")
+  end
+
+  it "#many_relations には'bar'から'baf'へのリレーションがあること" do
+    res = nil
+    @ymldot.many_relations.each{|e| res = e if e[:self].name == 'bar' and e[:have].name == 'baf' }
+    res.should_not be_nil
+  end
+
+  it "#entities の'baf'には'bar'へのforeginkeyがあること" do
+     @ymldot.entities["baf"].foreignkeys[0].should == "barID(FK)"
+  end
+end