entity 0.0.1 → 0.0.2

data/README.md

@@ -11,3 +11,9 @@ c = Company.new(name: 'Acme')
 c.attributes # => { name: 'Acme', address: nil } 
 ```
 
+## Installation
+
+```ruby
+gem 'entity'
+# or gem 'entity', github: 'redstonelabs/entity'
+```

data/init.rb

@@ -1,21 +1,4 @@
 require 'ostruct'
 require 'set'
-
-class Entity < OpenStruct
-  alias attributes marshal_dump
-  @@ensured_attributes = Set.new
-
-  def initialize(attributes = {})
-    super(attributes)
-    @@ensured_attributes.each do |key|
-      unless attributes.has_key?(key)
-        self.public_send("#{key}=", nil)
-      end
-    end
-  end
-
-  def self.ensure_attributes(*attribute_names)
-    @@ensured_attributes.merge(attribute_names)
-  end
-end
+require 'entity'
 

data/lib/entity.rb

@@ -0,0 +1,24 @@
+class Entity < OpenStruct
+  alias attributes marshal_dump
+
+  def initialize(attributes = {})
+    super(attributes)
+    self.class.ensured_attributes.each do |key|
+      unless attributes.has_key?(key)
+        self.public_send("#{key}=", nil)
+      end
+    end
+  end
+  
+  class << self
+    def ensure_attributes(*attribute_names)
+      @ensured_attributes ||= Set.new
+      @ensured_attributes.merge(attribute_names)
+    end
+
+    def ensured_attributes
+      @ensured_attributes ||= Set.new
+    end
+  end
+end
+

data/spec/entity_spec.rb

@@ -6,17 +6,17 @@ end
 
 describe 'Entity' do
   it 'assigns attributes' do
-    entity = Entity.new(foo: 'foo')
+    entity = Foo.new(foo: 'foo')
     entity.foo.should == 'foo'
   end
 
   it 'ensures ensured attributes' do
-    entity = Entity.new()
+    entity = Foo.new()
     entity.attributes.has_key?(:foo).should be_true
   end
 
   it 'returns all attributes' do
-    entity = Entity.new(foo: 'moo')
+    entity = Foo.new(foo: 'moo')
     entity.attributes.should == { foo: 'moo', bar: nil, buz: nil }
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: entity
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
   prerelease: 
 platform: ruby
 authors:
@@ -33,6 +33,7 @@ executables: []
 extensions: []
 extra_rdoc_files: []
 files:
+- lib/entity.rb
 - spec/entity_spec.rb
 - spec/spec_helper.rb
 - Gemfile