mongo_mapper_diagram 0.0.2 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: e4157c41bf11360936103a9ddb379ae8246e935d
-  data.tar.gz: 0eb1f0c9b310835d115df3bd92e546951362f811
+  metadata.gz: b832c19c5f2165438872c7bd022408971c1c0780
+  data.tar.gz: 9340dfd2ec9961115cea6f18b38da6e3906fdfb0
 SHA512:
-  metadata.gz: 118f5e12c5e183fd91894717226856926fc6a8d0ab19651d942ddc8d8d69554f196cac1fb81e21827bcbb7f51a5eeb932f287a255f9f66601becab74def52368
-  data.tar.gz: 145776bf938621367ea7b0d68d4c51157b7285cc7d0187c42c35541fc38de0918069edf671fcd636af4c8443092060b3db8fe3896e58cb342bc7f7c1cf968227
+  metadata.gz: 5fd5feb7b5077699c155b4a2c3d7ac9bc37ff70dd366f98f54fbd078d0ff4158a6300ec8e184136637bd65dcf4fda50c3b1e0cef50dc9a2c8ecdb1258801171a
+  data.tar.gz: 9acdaa70d046f6026e9f6bc68dab104ba920445576020d5b56a64f7e61935dc54ae5fe7e4e3e127b7dd9d6006015e9cc8ba73cedc0be0ca27dd305a94c9700ba

data/.travis.yml

@@ -0,0 +1,4 @@
+language: ruby
+rvm:
+  - 2.0.0-p353
+  - 2.1.0

data/README.md

@@ -1,4 +1,4 @@
-# MongoMapperDiagram
+# MongoMapperDiagram [![Build Status](https://travis-ci.org/dakatsuka/mongo_mapper_diagram.png?branch=master)](https://travis-ci.org/dakatsuka/mongo_mapper_diagram)
 
 Diagram generator for MongoMapper
 

data/Rakefile

@@ -1 +1,7 @@
 require "bundler/gem_tasks"
+require "rspec/core"
+require "rspec/core/rake_task"
+
+task default: :spec
+desc "Run all specs in spec directory"
+RSpec::Core::RakeTask.new(:spec)

data/lib/mongo_mapper_diagram/generator.rb

@@ -1,27 +1,30 @@
+require "mongo_mapper"
 require "gviz"
 
 module MongoMapperDiagram
   class Generator
-    def initialize(documents = MongoMapperDiagram::Document.all)
+    def initialize(documents = nil, gviz = nil)
       @documents = documents
+      @gviz = gviz
     end
 
     def generate(filename, extension = :png)
-      gviz = Gviz.new(:G, :digraph)
+      @documents ||= MongoMapperDiagram::Document.all
+      @gviz ||= Gviz.new(:G, :digraph)
 
       @documents.each do |doc|
-        gviz.node symbolize(doc), label: make_label(doc), shape: 'Mrecord'
+        @gviz.node symbolize(doc), label: make_label(doc), shape: 'Mrecord'
       end
 
       @documents.each do |doc|
         doc.associations.each do |k, association|
           if association.instance_of? MongoMapper::Plugins::Associations::BelongsToAssociation
-            gviz.route symbolize(doc) => symbolize(association.class_name)
+            @gviz.route symbolize(doc) => symbolize(association.class_name)
           end
         end
       end
 
-      gviz.save(filename, extension)
+      @gviz.save(filename, extension)
     end
 
     private
@@ -35,7 +38,7 @@ module MongoMapperDiagram
 
     def make_label(document)
       label =  "{#{humanize(document)}| "
-      label += document.defined_keys.map {|name, key|
+      label += document.keys.map {|name, key|
         name + " : " + key.type.to_s
       }.join('\l')
       label += "\\l}"

data/lib/mongo_mapper_diagram/version.rb

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

data/mongo_mapper_diagram.gemspec

@@ -20,6 +20,8 @@ Gem::Specification.new do |spec|
 
   spec.add_dependency "mongo_mapper", "~> 0.12"
   spec.add_dependency "gviz", "~> 0.3"
+  spec.add_development_dependency "bson_ext"
   spec.add_development_dependency "bundler", "~> 1.5"
   spec.add_development_dependency "rake"
+  spec.add_development_dependency "rspec", "~> 2.14"
 end

data/spec/mongo_mapper_diagram/generator_spec.rb

@@ -0,0 +1,46 @@
+require "spec_helper"
+
+class Author
+  include MongoMapper::Document
+  key :email, String
+  key :nickname, String
+  many :articles
+  many :comments
+end
+
+class Article
+  include MongoMapper::Document
+  key :title, String
+  key :body,  String
+  many :comments
+  belongs_to :author
+  timestamps!
+end
+
+class Comment
+  include MongoMapper::Document
+  key :body, String
+  belongs_to :author
+  belongs_to :article
+  timestamps!
+end
+
+describe MongoMapperDiagram::Generator do
+  it "generates diagram" do
+    gviz = Gviz.new(:G, :digraph)
+    gviz.should_receive(:save).once.with("/tmp/test", :svg)
+
+    generator = MongoMapperDiagram::Generator.new(nil, gviz)
+    generator.generate("/tmp/test", :svg)
+
+    nodes = gviz.instance_variable_get(:@nodes)
+    expect(nodes).to be_key :Article
+    expect(nodes).to be_key :Comment
+    expect(nodes).to be_key :Author
+
+    edges = gviz.instance_variable_get(:@edges)
+    expect(edges).to be_key :Article_Author
+    expect(edges).to be_key :Comment_Author
+    expect(edges).to be_key :Comment_Article
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,7 @@
+require "bundler/setup"
+
+Bundler.setup
+Bundler.require
+
+RSpec.configure do
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: mongo_mapper_diagram
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
 platform: ruby
 authors:
 - Dai Akatsuka
@@ -38,6 +38,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '0.3'
+- !ruby/object:Gem::Dependency
+  name: bson_ext
+  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: bundler
   requirement: !ruby/object:Gem::Requirement
@@ -66,6 +80,20 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.14'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.14'
 description: Diagram generator for MongoMapper
 email:
 - d.akatsuka@gmail.com
@@ -74,6 +102,7 @@ extensions: []
 extra_rdoc_files: []
 files:
 - ".gitignore"
+- ".travis.yml"
 - Gemfile
 - LICENSE.txt
 - README.md
@@ -85,6 +114,8 @@ files:
 - lib/mongo_mapper_diagram/railties/tasks.rake
 - lib/mongo_mapper_diagram/version.rb
 - mongo_mapper_diagram.gemspec
+- spec/mongo_mapper_diagram/generator_spec.rb
+- spec/spec_helper.rb
 homepage: ''
 licenses:
 - MIT
@@ -109,5 +140,7 @@ rubygems_version: 2.2.0
 signing_key: 
 specification_version: 4
 summary: Diagram generator for MongoMapper
-test_files: []
+test_files:
+- spec/mongo_mapper_diagram/generator_spec.rb
+- spec/spec_helper.rb
 has_rdoc: