click 0.0.2 → 0.0.3

data/lib/click/clicker.rb

@@ -1,3 +1,5 @@
+require 'click/indeterminable'
+
 module Click
   class Clicker
     def click!
@@ -9,9 +11,9 @@ module Click
       ObjectSpace.each_object do |object|
         begin
           klass = object.class
-          next unless klass.is_a?(Class)
+          klass = Click::Indeterminable unless klass.is_a?(Class)
         rescue NoMethodError
-          next
+          klass = Click::Indeterminable
         end
 
         @state[klass] += 1

data/lib/click/database.rb

@@ -19,12 +19,14 @@ module Click
       private
       def _with_db(db)
         ensure_tables!(db)
-        Sequel::Model.db = db
-        require 'click/database/models'
+        assign_db_to_models(db)
         yield db
-      ensure
-        db = nil
-        Sequel::Model.db = nil
+      end
+
+      def assign_db_to_models(db)
+        require 'click/database/models'
+        Click::Database::Models::Snapshot.db = db
+        Click::Database::Models::ObjectCount.db = db
       end
 
       def ensure_tables!(db)

data/lib/click/indeterminable.rb

@@ -0,0 +1,4 @@
+module Click
+  class Indeterminable < BasicObject
+  end
+end

data/lib/click/version.rb

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

data/spec/clicker_spec.rb

@@ -31,6 +31,35 @@ describe Click::Clicker do
         clicker.instance_count(Symbol)
       }.by_at_least(difference)
     end
+
+    it 'counts BasicObjects as indeterminable' do
+      difference = 100
+      stuff = []
+
+      expect {
+        difference.times { |i| stuff << BasicObject.new }
+      }.to change {
+        clicker.click!
+        clicker.instance_count(Click::Indeterminable)
+      }.by_at_least(difference)
+    end
+
+    it 'counts objects that have overridden #class as indeterminable' do
+      difference = 100
+      stuff = []
+      klass = Class.new do
+        def class
+          nil
+        end
+      end
+
+      expect {
+        difference.times { |i| stuff << klass.new }
+      }.to change {
+        clicker.click!
+        clicker.instance_count(Click::Indeterminable)
+      }.by_at_least(difference)
+    end
   end
 
   describe '#object_counts' do

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: click
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
   prerelease: 
 platform: ruby
 authors:
@@ -131,6 +131,7 @@ files:
 - lib/click/database/models/object_count.rb
 - lib/click/database/models/snapshot.rb
 - lib/click/database/writer.rb
+- lib/click/indeterminable.rb
 - lib/click/observer.rb
 - lib/click/version.rb
 - spec/click_spec.rb