snapbot 0.1.1 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: aa9a6b0cbc5334ac7078254f385f9c859c0ea22d7ca4821741bade0b4c32cca6
-  data.tar.gz: 8e69a33b99977466a92b9772abaa2e2d889a0d7ff200ed81c5effb01581eb014
+  metadata.gz: e57537e104190501dc2c47d19b8f76a95e91dd2b7f189f2686b4159c171ad101
+  data.tar.gz: d8dfe6118e506a0a10ad5d8dc86c8a8ba041e3fa99253d64c2a29062026f853f
 SHA512:
-  metadata.gz: 1de519ae9f8e26a44d081682b83ca89d101daf00f5665c7d04a4ed1bb7c8ee701a72b476b45d9d5760ff2621e498d8bbe0ddfb3bde386fc4f188a16cee3c82db
-  data.tar.gz: 6f549408fbffc23fc5f96f620dc0be49df614b84ce0e5c913fb375422bcec052f36e065a63080f03be3a2992c7f7a519873ab0efcb31bd9610e5b2f2bdf2340e
+  metadata.gz: 3c3d84a4445afa197bafa704cb1d2ac1535b8bdf12c8a19155734af415ca03aa9bc1fa2c28ba44903ddbcfd15fea0e1bcee244bdd0d99e4aaada3c6fbbcbb0c0
+  data.tar.gz: 3f19483aa69878acbdf035ef05d65be3db10f0d5f0bb4a79658020b86e6ed5580f2f366b4bb204fa312d99a7f197c021b1717852be828591f5fb74bb525a4a17

data/lib/snapbot/diagram/dot_generator.rb

@@ -40,7 +40,7 @@ module Snapbot
       def collect_lets(example)
         @lets_by_value = RSpec::Lets.new(example).collect.each_with_object({}) do |sym, lets_by_value|
           value = example.send(sym) unless @ignore_lets.include?(sym)
-          lets_by_value[value] = sym if value.is_a?(Reflector::BASE_ACTIVERECORD_CLASS)
+          lets_by_value[value] = sym if value.is_a?(reflector.base_activerecord_class)
         end
       end
 

data/lib/snapbot/reflector.rb

@@ -5,12 +5,14 @@ require "active_record"
 module Snapbot
   # Reflect models and instances in a way that's useful for generating a diagram
   class Reflector
-    BASE_ACTIVERECORD_CLASS = defined?(ApplicationRecord) ? ApplicationRecord : ActiveRecord::Base
+    def base_activerecord_class
+      defined?(ApplicationRecord) ? ApplicationRecord : ActiveRecord::Base
+    end
 
     def models(only_with_records: true)
       @models ||= begin
         Rails.application.eager_load! if defined?(Rails)
-        BASE_ACTIVERECORD_CLASS.descendants.reject do |c|
+        base_activerecord_class.descendants.reject do |c|
           c.to_s == "Schema" || (only_with_records && c.count.zero?)
         end
       end

data/lib/snapbot/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Snapbot
-  VERSION = "0.1.1"
+  VERSION = "0.1.2"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: snapbot
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.2
 platform: ruby
 authors:
 - Russell Garner