ascheink-active_yaml 0.0.3 → 0.0.4

data/VERSION

@@ -1 +1 @@
-0.0.3
+0.0.5

data/active_yaml.gemspec

@@ -2,7 +2,7 @@
 
 Gem::Specification.new do |s|
   s.name = %q{active_yaml}
-  s.version = "0.0.3"
+  s.version = "0.0.5"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["Andrei Scheinkman"]
@@ -23,7 +23,9 @@ Gem::Specification.new do |s|
      "lib/active_yaml.rb",
      "test/active_yaml_test.rb",
      "test/app/models/city.rb",
+     "test/app/models/state.rb",
      "test/db/yaml/city.yml",
+     "test/db/yaml/state.yml",
      "test/test_helper.rb"
   ]
   s.has_rdoc = true
@@ -35,6 +37,7 @@ Gem::Specification.new do |s|
   s.test_files = [
     "test/active_yaml_test.rb",
      "test/app/models/city.rb",
+     "test/app/models/state.rb",
      "test/test_helper.rb"
   ]
 

data/lib/active_yaml.rb

@@ -8,6 +8,7 @@ class ActiveYAML
   
   def self.inherited(p)
     data = YAML.load_file("#{RAILS_ROOT}/db/yaml/#{p.name.underscore}.yml")
+    p.extend(Enumerable)
     p.class_eval do
       @instances = []
       fields = []
@@ -26,6 +27,7 @@ class ActiveYAML
       class_eval "def self.find(id); find_by_id(id) or raise ActiveRecord::RecordNotFound; end"
       class_eval "def self.first; @instances.first; end"
       class_eval "def self.last; @instances.last; end"
+      class_eval "def self.each; @instances.each { |b| yield b }; end"
       s = "#{p.to_s}(#{fields.join(', ')})"
       class_eval "def self.to_s; \"#{s}\"; end"
       def self.find_all

data/test/active_yaml_test.rb

@@ -4,6 +4,7 @@ class ActiveYamlTest < Test::Unit::TestCase
   
   ActiveYAML::RAILS_ROOT = 'test'
   require 'app/models/city.rb'
+  require 'app/models/state.rb'
   
   context "class" do
     should "create find_all" do
@@ -27,6 +28,14 @@ class ActiveYamlTest < Test::Unit::TestCase
       assert_respond_to City, :find_all_by_population
     end
     
+    should "support each" do
+      assert_respond_to City, :each
+    end
+    
+    should "be Enumerable" do
+      assert_respond_to City, :any?
+    end
+    
     should "print out fields in to_s" do
       assert_equal "City(id, population, display_name)", City.to_s
     end
@@ -42,7 +51,24 @@ class ActiveYamlTest < Test::Unit::TestCase
       assert_respond_to @i, :display_name
       assert_respond_to @i, :population
     end
-    
   end
-    
+
+  context "data" do
+    should "load all records" do
+      assert_equal 3, City.find_all.size
+    end
+    should "load fields" do
+      nyc = City.find('NYC')
+      assert_equal "New York", nyc.display_name
+      assert_equal 8274527, nyc.population
+    end
+    should "not confuse records from different models" do
+      assert_equal 3, City.find_all.size
+      assert_equal 2, State.find_all.size
+      assert_equal 3, City.find_all.size
+    end
+    should "support enumerable" do
+      assert_equal 14951385, City.inject(0) { |a, c| a + c.population }
+    end
+  end
 end

data/test/app/models/state.rb

@@ -0,0 +1,4 @@
+require 'lib/active_yaml'
+
+class State < ActiveYAML
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: ascheink-active_yaml
 version: !ruby/object:Gem::Version 
-  version: 0.0.3
+  version: 0.0.4
 platform: ruby
 authors: 
 - Andrei Scheinkman
@@ -73,4 +73,5 @@ summary: ActiveYAML
 test_files: 
 - test/active_yaml_test.rb
 - test/app/models/city.rb
+- test/app/models/state.rb
 - test/test_helper.rb