mongomapper_ext 0.0.4

data/.document

@@ -0,0 +1,5 @@
+README.rdoc
+lib/**/*.rb
+bin/*
+features/**/*.feature
+LICENSE

data/.gitignore

@@ -0,0 +1,21 @@
+## MAC OS
+.DS_Store
+
+## TEXTMATE
+*.tmproj
+tmtags
+
+## EMACS
+*~
+\#*
+.\#*
+
+## VIM
+*.swp
+
+## PROJECT::GENERAL
+coverage
+rdoc
+pkg
+
+## PROJECT::SPECIFIC

data/LICENSE

@@ -0,0 +1,20 @@
+Copyright (c) 2009 David A. Cuadrado
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.rdoc

@@ -0,0 +1,19 @@
+= mongomapper_ext
+
+MongoMapper extensions.
+
+== Note on Patches/Pull Requests
+ 
+* Fork the project.
+* Make your feature addition or bug fix.
+* Add tests for it. This is important so I don't break it in a
+  future version unintentionally.
+* Commit, do not mess with rakefile, version, or history.
+  (if you want to have your own version, that is fine but
+   bump version in a commit by itself I can ignore when I pull)
+* Send me a pull request. Bonus points for topic branches.
+
+== Copyright
+
+Copyright (c) 2009 David A. Cuadrado. See LICENSE for details.
+

data/Rakefile

@@ -0,0 +1,59 @@
+require 'rubygems'
+require 'rake'
+
+begin
+  require 'jeweler'
+  Jeweler::Tasks.new do |gem|
+    gem.name = "mongomapper_ext"
+    gem.summary = %Q{MongoMapper extensions}
+    gem.description = %Q{MongoMapper extensions}
+    gem.email = "krawek@gmail.com"
+    gem.homepage = "http://github.com/dcu/mongomapper_ext"
+    gem.authors = ["David A. Cuadrado"]
+
+    gem.add_dependency('mongo_mapper', '>= 0.5.6')
+    gem.add_dependency('ruby-stemmer', '>= 0.5.3')
+    gem.add_dependency('uuidtools', '>= 2.0.0')
+
+    gem.add_development_dependency("shoulda", ">= 2.10.2")
+    gem.add_development_dependency('jnunemaker-matchy', '0.4.0')
+    gem.add_development_dependency('mocha', '>= 0.9.4')
+  end
+rescue LoadError
+  puts "Jeweler (or a dependency) not available. Install it with: sudo gem install jeweler"
+end
+
+require 'rake/testtask'
+Rake::TestTask.new(:test) do |test|
+  test.libs << 'lib' << 'test'
+  test.pattern = 'test/**/test_*.rb'
+  test.verbose = true
+end
+
+begin
+  require 'rcov/rcovtask'
+  Rcov::RcovTask.new do |test|
+    test.libs << 'test'
+    test.pattern = 'test/**/test_*.rb'
+    test.verbose = true
+  end
+rescue LoadError
+  task :rcov do
+    abort "RCov is not available. In order to run rcov, you must: sudo gem install spicycode-rcov"
+  end
+end
+
+task :test => :check_dependencies
+
+task :default => :test
+
+require 'rake/rdoctask'
+Rake::RDocTask.new do |rdoc|
+  version = File.exist?('VERSION') ? File.read('VERSION') : ""
+
+  rdoc.rdoc_dir = 'rdoc'
+  rdoc.title = "mongomapper_ext #{version}"
+  rdoc.rdoc_files.include('README*')
+  rdoc.rdoc_files.include('lib/**/*.rb')
+end
+

data/VERSION

@@ -0,0 +1 @@
+0.0.4

data/examples/filter.rb

@@ -0,0 +1,23 @@
+require './helper'
+
+class FilterEx
+  include MongoMapper::Document
+  include MongoMapperExt::Filter
+
+  key :title, String
+  key :body, String
+
+  filterable_keys :title, :body
+end
+
+FilterEx.delete_all
+o1 = FilterEx.create(:title => "A Great Title", :body => "A great Body")
+o2 = FilterEx.create(:title => "A Good Title", :body => "A good Body")
+
+
+puts "filter: #{FilterEx.filter("title").size}"
+puts "filter: #{FilterEx.filter("title", :limit => 1).inspect}"
+puts "filter: #{FilterEx.filter("great").inspect}"
+
+
+FilterEx.delete_all

data/examples/helper.rb

@@ -0,0 +1,6 @@
+
+$:.unshift File.dirname(__FILE__)+"/../lib"
+require 'mongomapper_ext'
+
+MongoMapper.database = 'test'
+

data/examples/slugizer.rb

@@ -0,0 +1,15 @@
+require './helper'
+
+class SlugizerEx
+  include MongoMapper::Document
+  include MongoMapperExt::Slugizer
+
+  slug_key :title
+
+  key :title
+end
+
+u = SlugizerEx.create(:title => "The title of the blogpost!!!")
+
+puts u.slug
+puts SlugizerEx.by_slug(u.slug).inspect

data/examples/storage.rb

@@ -0,0 +1,22 @@
+require './helper'
+
+class StorageEx
+  include MongoMapper::Document
+  include MongoMapperExt::Storage
+
+  file_key :my_file
+end
+
+file = StringIO.new("file content")
+file2 = StringIO.new("file content 2")
+
+s = StorageEx.new
+s.put_file("filename.txt", file)
+s.my_file = file2
+
+s.save
+
+from_db = StorageEx.find(s.id)
+
+puts from_db.fetch_file("filename.txt").read
+puts from_db.my_file.read

data/examples/types.rb

@@ -0,0 +1,32 @@
+require './helper'
+
+class TypeEx
+  include MongoMapper::Document
+
+  key :time, Timestamp
+  key :config, OpenStruct
+  key :set, Set
+end
+
+
+type = TypeEx.new(:set => Set.new, :config => OpenStruct.new)
+
+type.config.name = "Alan"
+type.config.last_name = "Turing"
+
+
+type.time = Time.now
+type.set += [1,1,2,3,2,2]
+
+type.save
+
+from_db = TypeEx.find(type.id)
+
+puts from_db.config.inspect
+puts from_db.set.inspect
+
+Time.zone = "Hawaii"
+puts from_db.time.inspect
+
+from_db.destroy
+

data/examples/update.rb

@@ -0,0 +1,14 @@
+require './helper'
+
+class UpdateEx
+  include MongoMapper::Document
+
+  key :a
+  key :b
+  key :c
+end
+
+u = UpdateEx.create(:a => 1, :b => 0, :c => 1)
+
+u.safe_update(%w[a c], {"a" => 3, "b" => 3, "c" => 3})
+puts u.attributes.inspect

data/lib/mongomapper_ext/file.rb

@@ -0,0 +1,35 @@
+module MongoMapperExt
+  class File < GridFS::GridStore
+    attr_reader :id, :attributes
+
+    def initialize(owner, attrs = {})
+      @owner = owner
+      @id = attrs.delete("_id")
+
+      class_eval do
+        attrs.each do |k,v|
+          define_method(k) do
+            v
+          end
+        end
+      end
+
+      super(@owner.class.database, attrs["filename"], "r", :root => @owner.collection.name)
+    end
+
+    def [](name)
+      @attributes[name.to_s]
+    end
+
+    def self.fetch(owner, filename)
+      db = owner.class.database
+      criteria, options = MongoMapper::FinderOptions.new(owner.class, :filename => filename, :metadata => {:_id => owner.id}, :limit => 1).to_a
+
+      obj = db.collection("#{owner.collection.name}.files").find(criteria, options).next_object
+
+      if obj
+        self.new(owner, obj)
+      end
+    end
+  end
+end

data/lib/mongomapper_ext/filter.rb

@@ -0,0 +1,76 @@
+module MongoMapperExt
+  module Filter
+    def self.included(klass)
+      require 'lingua/stemmer'
+
+      klass.class_eval do
+        extend ClassMethods
+
+        key :_keywords, Array
+        ensure_index :_keywords
+
+        before_save :_update_keywords
+      end
+    end
+
+    module ClassMethods
+      def filterable_keys(*keys)
+        @filterable_keys ||= Set.new
+        @filterable_keys += keys
+
+        @filterable_keys
+      end
+
+      def language(lang = 'en')
+        @language ||= lang
+      end
+
+      def filter(query, opts = {})
+        q = query.downcase.split.map do |k|
+          Regexp.escape(k)
+        end.join("|")
+        if opts[:per_page]
+          self.paginate(opts.deep_merge(:conditions => {:_keywords => /^(#{q}).*/ }))
+        else
+          self.find(:all, opts.deep_merge(:conditions => {:_keywords => /^(#{q}).*/ }))
+        end
+      end
+    end
+
+    protected
+    def _update_keywords
+      lang = self.class.language
+      if lang.kind_of?(Symbol)
+        lang = send(lang)
+      elsif lang.kind_of?(Proc)
+        lang = lang.call(self)
+      end
+
+      s = Lingua::Stemmer.new(:language => lang)
+
+      self._keywords = []
+      self.class.filterable_keys.each do |key|
+        self._keywords += keywords_for_value(s, read_attribute(key))
+      end
+    end
+
+    private
+    def keywords_for_value(stemmer, val)
+      if val.kind_of?(String)
+        val.downcase.split.map do |word|
+          stem = stemmer.stem(word)
+          if stem != word
+            [stem, word]
+          else
+            word
+          end
+        end.flatten
+      elsif val.kind_of?(Array)
+        val.map { |e| keywords_for_value(stemmer, e) }.flatten
+      else
+        [val]
+      end
+    end
+  end
+end
+

data/lib/mongomapper_ext/slugizer.rb

@@ -0,0 +1,46 @@
+module MongoMapperExt
+  module Slugizer
+    def self.included(klass)
+      klass.class_eval do
+        extend ClassMethods
+        extend Finder
+
+        key :slug, String
+        ensure_index :slug
+
+        before_validation_on_create :generate_slug
+      end
+    end
+
+    def to_param
+      if self.slug.blank?
+        self.id
+      else
+        self.slug
+      end
+    end
+
+    protected
+    def generate_slug
+      if self.slug.blank?
+        key = UUIDTools::UUID.random_create.hexdigest[0,4] #optimize
+        self.slug = key+"-"+self[self.class.slug_key].gsub(/[^A-Za-z0-9\s\-]/, "")[0,20].strip.gsub(/\s+/, "-").downcase
+      end
+    end
+
+    module ClassMethods
+      def slug_key(key = :name)
+        @slug_key ||= key
+      end
+    end
+
+    module Finder
+      def by_slug(id)
+        self.find_by_slug(id) || self.find_by_id(id)
+      end
+      alias :find_by_slug_or_id :by_slug
+    end
+  end
+end
+
+MongoMapper::Associations::Proxy.send(:include, MongoMapperExt::Slugizer::Finder)

data/lib/mongomapper_ext/storage.rb

@@ -0,0 +1,73 @@
+module MongoMapperExt
+  module Storage
+    def self.included(model)
+      model.class_eval do
+        extend ClassMethods
+        after_create :_sync_pending_files
+      end
+    end
+
+    # FIXME: enable metadata. re http://jira.mongodb.org/browse/SERVER-377
+    def put_file(filename, io, metadata = {})
+      if !new?
+        # :metadata => metadata.deep_merge({:_id => self.id})
+        GridFS::GridStore.open(self.class.database, filename, "w",
+                               :root => self.collection.name,
+                               :metadata => {:_id => self.id}) do |f|
+          while data = io.read(256)
+            f.write(data)
+          end
+          io.close
+        end
+      else
+        (@_pending_files ||= {})[filename] = io
+      end
+    end
+
+    def fetch_file(filename)
+      if !new?
+        MongoMapperExt::File.fetch(self, filename)
+      end
+    end
+
+    def files
+      criteria, options = MongoMapper::FinderOptions.new(self.class, :metadata => {:_id => self.id}).to_a
+      coll = self.class.database.collection("#{self.collection.name}.files")
+      @files = coll.find(criteria, options).map do |a|
+        MongoMapperExt::File.new(self, a)
+      end
+    end
+
+    protected
+    def _sync_pending_files
+      if @_pending_files
+        @_pending_files.each do |filename, data|
+          put_file(filename, data)
+        end
+        @_pending_files = nil
+      end
+    end
+
+    module ClassMethods
+      def file_key(name)
+        define_method("#{name}=") do |file|
+          file_id = UUIDTools::UUID.random_create.hexdigest
+          filename = name
+
+          if file.respond_to?(:original_filename)
+            filename = file.original_filename
+          elsif file.respond_to?(:path)
+            filename = file.path
+          end
+
+          put_file(file_id, file, :original_filename => filename)
+          self["_#{name}"] = file_id
+        end
+
+        define_method(name) do
+          fetch_file(self["_#{name}"])
+        end
+      end
+    end
+  end
+end

data/lib/mongomapper_ext/types/open_struct.rb

@@ -0,0 +1,19 @@
+require 'ostruct'
+
+class OpenStruct
+  def self.to_mongo(value)
+    if value.kind_of?(self)
+      value.send(:table)
+    else
+      value
+    end
+  end
+
+  def self.from_mongo(value)
+    if value.kind_of?(self)
+      value
+    else
+      OpenStruct.new(value || {})
+    end
+  end
+end

data/lib/mongomapper_ext/types/set.rb

@@ -0,0 +1,9 @@
+class Set
+  def self.to_mongo(value)
+    value.to_a
+  end
+
+  def self.from_mongo(value)
+    Set.new(value)
+  end
+end

data/lib/mongomapper_ext/types/timestamp.rb

@@ -0,0 +1,15 @@
+class Timestamp
+  def self.from_mongo(value)
+    if value.nil? || value == ''
+      nil
+    else
+      Time.zone.at(value.to_i)
+    end
+  end
+
+  def self.to_mongo(value)
+    value.to_i
+  end
+end
+
+Time.zone ||= 'UTC'

data/lib/mongomapper_ext/update.rb

@@ -0,0 +1,10 @@
+module MongoMapperExt
+  module Update
+    def safe_update(white_list, values)
+      white_list.each do |key|
+        send("#{key}=", values[key]) if values.has_key?(key)
+      end
+    end
+  end
+end
+MongoMapper::EmbeddedDocument.send(:include, MongoMapperExt::Update)

data/lib/mongomapper_ext.rb

@@ -0,0 +1,23 @@
+$:.unshift File.dirname(__FILE__)
+
+require 'mongo/gridfs'
+require 'mongo_mapper'
+require 'uuidtools'
+
+# types
+require 'mongomapper_ext/types/open_struct'
+require 'mongomapper_ext/types/timestamp'
+require 'mongomapper_ext/types/set'
+
+# storage
+require 'mongomapper_ext/file'
+require 'mongomapper_ext/storage'
+
+# update
+require 'mongomapper_ext/update'
+
+# filter
+require 'mongomapper_ext/filter'
+
+# slug
+require 'mongomapper_ext/slugizer'

data/test/helper.rb

@@ -0,0 +1,23 @@
+require 'rubygems'
+
+gem 'jnunemaker-matchy'
+
+require 'matchy'
+require 'shoulda'
+require 'timecop'
+require 'mocha'
+require 'pp'
+
+require 'support/custom_matchers'
+
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+
+require 'mongomapper_ext'
+require 'models'
+
+class Test::Unit::TestCase
+  include CustomMatchers
+end
+
+MongoMapper.database = 'test'

data/test/models.rb

@@ -0,0 +1,52 @@
+
+class Event # for safe_update, and Timestamp
+  include MongoMapper::Document
+
+  key :start_date, Timestamp
+  key :end_date, Timestamp
+
+  key :password, String
+end
+
+class Recipe # for Set
+  include MongoMapper::Document
+  include MongoMapperExt::Filter
+
+  language Proc.new { |d| d.language }
+  filterable_keys :language
+
+  key :ingredients, Set
+  key :description, String
+  key :language, String, :default => 'en'
+end
+
+class Avatar # for Storage and File
+  include MongoMapper::Document
+  include MongoMapperExt::Storage
+
+  file_key :data
+end
+
+class UserConfig #for OpenStruct
+  include MongoMapper::Document
+  key :entries, OpenStruct
+end
+
+class BlogPost # for Slug and Filter
+  include MongoMapper::Document
+  include MongoMapperExt::Filter
+  include MongoMapperExt::Slugizer
+
+  filterable_keys :title, :body, :tags, :date
+  slug_key :title
+  language :find_language
+
+  key :title, String
+  key :body, String
+  key :tags, Array
+  key :date, Time
+
+  def find_language
+    'en'
+  end
+end

data/test/support/custom_matchers.rb

@@ -0,0 +1,55 @@
+module CustomMatchers
+  custom_matcher :be_nil do |receiver, matcher, args|
+    matcher.positive_failure_message = "Expected #{receiver} to be nil but it wasn't"
+    matcher.negative_failure_message = "Expected #{receiver} not to be nil but it was"
+    receiver.nil?
+  end
+  
+  custom_matcher :be_blank do |receiver, matcher, args|
+    matcher.positive_failure_message = "Expected #{receiver} to be blank but it wasn't"
+    matcher.negative_failure_message = "Expected #{receiver} not to be blank but it was"
+    receiver.blank?
+  end
+  
+  custom_matcher :be_true do |receiver, matcher, args|
+    matcher.positive_failure_message = "Expected #{receiver} to be true but it wasn't"
+    matcher.negative_failure_message = "Expected #{receiver} not to be true but it was"
+    receiver.eql?(true)
+  end
+  
+  custom_matcher :be_false do |receiver, matcher, args|
+    matcher.positive_failure_message = "Expected #{receiver} to be false but it wasn't"
+    matcher.negative_failure_message = "Expected #{receiver} not to be false but it was"
+    receiver.eql?(false)
+  end
+
+  custom_matcher :be_valid do |receiver, matcher, args|
+    matcher.positive_failure_message = "Expected to be valid but it was invalid #{receiver.errors.inspect}"
+    matcher.negative_failure_message = "Expected to be invalid but it was valid #{receiver.errors.inspect}"
+    receiver.valid?
+  end
+
+  custom_matcher :have_error_on do |receiver, matcher, args|
+    receiver.valid?
+    attribute = args[0]
+    expected_message = args[1]
+    
+    if expected_message.nil?
+      matcher.positive_failure_message = "#{receiver} had no errors on #{attribute}"
+      matcher.negative_failure_message = "#{receiver} had errors on #{attribute} #{receiver.errors.inspect}"
+      !receiver.errors.on(attribute).blank?
+    else
+      actual = receiver.errors.on(attribute)
+      matcher.positive_failure_message = %Q(Expected error on #{attribute} to be "#{expected_message}" but was "#{actual}")
+      matcher.negative_failure_message = %Q(Expected error on #{attribute} not to be "#{expected_message}" but was "#{actual}")
+      actual == expected_message
+    end
+  end
+  
+  custom_matcher :have_index do |receiver, matcher, args|
+    index_name = args[0]
+    matcher.positive_failure_message = "#{receiver} does not have index named #{index_name}, but should"
+    matcher.negative_failure_message = "#{receiver} does have index named #{index_name}, but should not"
+    !receiver.collection.index_information.detect { |index| index[0] == index_name }.nil?
+  end
+end

data/test/test_filter.rb

@@ -0,0 +1,43 @@
+require 'helper'
+
+class TestFilter < Test::Unit::TestCase
+  context "filtering data" do
+    setup do
+      BlogPost.delete_all
+      @blogpost = BlogPost.create(:title => "%How dOEs tHIs Work?!",
+                                  :body => "HeRe is tHe Body of the bLog pOsT",
+                                  :tags => ["my", "list", "of", "tags"],
+                                  :date => Time.parse('01-01-2009'))
+    end
+
+    should "be case insensitive" do
+      BlogPost.filter("body").should == [@blogpost]
+    end
+
+    should "be able to find by title" do
+      BlogPost.filter("this").should == [@blogpost]
+    end
+
+    should "be able to find by body" do
+      BlogPost.filter("blog").should == [@blogpost]
+    end
+
+    should "be able to find by tags" do
+      BlogPost.filter("list").should == [@blogpost]
+    end
+
+    should "be able to find by title or body" do
+      BlogPost.filter("work blog").should == [@blogpost]
+    end
+
+    should "ignore inexistant words" do
+      BlogPost.filter("work lalala").should == [@blogpost]
+    end
+
+    should "allow to paginate results" do
+      results = BlogPost.filter("tag", :per_page => 1, :page => 1)
+      results.should == [@blogpost]
+      results.total_pages.should == 1
+    end
+  end
+end

data/test/test_slugizer.rb

@@ -0,0 +1,40 @@
+require 'helper'
+
+class TestSlugizer < Test::Unit::TestCase
+  context "working with slugs" do
+    setup do
+      BlogPost.delete_all
+      @blogpost = BlogPost.create(:title => "%bLog pOSt tiTLe!",
+                                  :body => "HeRe is tHe Body of the bLog pOsT")
+    end
+
+    should "generate the slug" do
+      @blogpost.slug.should =~ /\w+-blog-post-title/
+    end
+
+    should "return the slug as param" do
+      @blogpost.to_param =~ /\w+-blog-post-title/
+    end
+
+    should "return the id if slug was not generated" do
+      @blogpost.slug = nil
+      @blogpost.to_param.should == @blogpost.id
+    end
+  end
+
+  context "finding objects" do
+    setup do
+      BlogPost.delete_all
+      @blogpost = BlogPost.create(:title => "%bLog pOSt tiTLe!",
+                                  :body => "HeRe is tHe Body of the bLog pOsT")
+    end
+
+    should "be able to find by slug" do
+      BlogPost.by_slug(@blogpost.slug).should == @blogpost
+    end
+
+    should "be able to find by id" do
+      BlogPost.by_slug(@blogpost.id).should == @blogpost
+    end
+  end
+end

data/test/test_storage.rb

@@ -0,0 +1,64 @@
+require 'helper'
+
+class StorageTest < Test::Unit::TestCase
+  context "Storing files" do
+    setup do
+      @avatar = Avatar.create
+      @data = StringIO.new("my avatar image")
+    end
+
+    should "store the file" do
+      @avatar.put_file("an_avatar.png", @data)
+      data = Avatar.find(@avatar.id).fetch_file("an_avatar.png").read
+      data.should == "my avatar image"
+    end
+
+    should "close the file after storing" do
+      @avatar.put_file("an_avatar.png", @data)
+      @data.should be_closed
+    end
+
+    context "in attributes" do
+      should "store the given file" do
+        @avatar.data = @data
+        @avatar.save!
+        @avatar.data.should_not be_nil
+        @avatar.data.read.should == "my avatar image"
+      end
+    end
+
+    context "with new objects" do
+      setup do
+        @avatar = Avatar.new
+      end
+
+      should "store the file after saving" do
+        @avatar.put_file("an_avatar.png", @data)
+        @avatar.save
+        @avatar.fetch_file("an_avatar.png").read.should == "my avatar image"
+      end
+
+      should "store not the file if object is new" do
+        @avatar.put_file("an_avatar.png", @data)
+        @avatar.fetch_file("an_avatar.png").should be_nil
+      end
+    end
+  end
+
+  context "Fetching files" do
+    setup do
+      @avatar = Avatar.create
+      @data = StringIO.new("my avatar image")
+    end
+
+    should "fetch the list of files" do
+      @avatar.put_file("file1", StringIO.new("data1"))
+      @avatar.put_file("file2", StringIO.new("data2"))
+      @avatar.put_file("file3", StringIO.new("data3"))
+      file_names = @avatar.files.map { |f| f.filename }
+      file_names.should include("file1")
+      file_names.should include("file2")
+      file_names.should include("file3")
+    end
+  end
+end

data/test/test_update.rb

@@ -0,0 +1,16 @@
+require 'helper'
+
+class UpdateTest < Test::Unit::TestCase
+  should "only update the given white listed attributes" do
+    event = Event.new(:password => "original")
+    start_date = Time.zone.now
+    end_date = start_date.tomorrow
+
+    event.safe_update(%w[start_date end_date], {"start_date" => start_date,
+                                                "end_date" => end_date,
+                                                "password" => "hacked"})
+    event.password.should == "original"
+    event.start_date.to_s.should == start_date.to_s
+    event.end_date.to_s.should == end_date.to_s
+  end
+end

data/test/types/test_open_struct.rb

@@ -0,0 +1,19 @@
+require 'helper'
+
+class OpenStructTest < Test::Unit::TestCase
+  def from_db
+    UserConfig.find(@config.id)
+  end
+
+  context "working with sets" do
+    setup do
+      @config = UserConfig.create!(:entries => OpenStruct.new({}))
+    end
+
+    should "allow to add new keys" do
+      @config.entries.new_key = "my new key"
+      @config.save!
+      from_db.entries.new_key.should == "my new key"
+    end
+  end
+end

data/test/types/test_set.rb

@@ -0,0 +1,26 @@
+require 'helper'
+
+class SetTest < Test::Unit::TestCase
+  def from_db
+    Recipe.find(@recipe.id)
+  end
+
+  context "working with sets" do
+    setup do
+      @recipe = Recipe.create!(:ingredients => %w[salt sugar water salt sugar water])
+    end
+
+    should "not have duplicates" do
+      from_db.ingredients.size.should == 3
+      from_db.ingredients.should include("salt")
+      from_db.ingredients.should include("sugar")
+      from_db.ingredients.should include("water")
+    end
+
+    should "not add duplicates" do
+      original_size = @recipe.ingredients.size
+      @recipe.ingredients << "salt"
+      @recipe.ingredients.size.should == original_size
+    end
+  end
+end

data/test/types/test_timestamp.rb

@@ -0,0 +1,40 @@
+require 'helper'
+
+class TimestampTest < Test::Unit::TestCase
+  def from_db
+    Event.find(@event.id)
+  end
+
+  context "working with timestamps" do
+    setup do
+      Event.delete_all
+
+      Time.zone = 'UTC'
+      @start_time = Time.zone.parse('01-01-2009')
+      @end_time = @start_time.tomorrow
+
+      @event = Event.create!(:start_date => @start_time, :end_date => @end_time)
+    end
+
+    should "store the date" do
+      from_db.start_date.to_s.should == @start_time.to_s
+    end
+
+    should "be able to convert the time to the given timezone" do
+      Time.zone = 'Hawaii'
+      from_db.start_date.to_s.should == "2008-12-31 14:00:00 -1000"
+    end
+
+    should "be able to compare dates" do
+      start_time = @start_time.tomorrow.tomorrow
+      end_time = start_time.tomorrow
+
+      @event2 = Event.create!(:start_date => start_time, :end_datime => end_time)
+
+      Event.count.should == 2
+      events = Event.find(:all, :$where => ("this.start_date >= %d && this.start_date <= %d" % [@event.start_date.yesterday.to_i, @event2.start_date.yesterday.to_i]))
+
+      events.should == [@event]
+    end
+  end
+end

metadata

@@ -0,0 +1,160 @@
+--- !ruby/object:Gem::Specification 
+name: mongomapper_ext
+version: !ruby/object:Gem::Version 
+  version: 0.0.4
+platform: ruby
+authors: 
+- David A. Cuadrado
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2009-10-25 00:00:00 -05:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: mongo_mapper
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 0.5.6
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: ruby-stemmer
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 0.5.3
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: uuidtools
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 2.0.0
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: shoulda
+  type: :development
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 2.10.2
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: jnunemaker-matchy
+  type: :development
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - "="
+      - !ruby/object:Gem::Version 
+        version: 0.4.0
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: mocha
+  type: :development
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 0.9.4
+    version: 
+description: MongoMapper extensions
+email: krawek@gmail.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: 
+- LICENSE
+- README.rdoc
+files: 
+- .document
+- .gitignore
+- LICENSE
+- README.rdoc
+- Rakefile
+- VERSION
+- examples/filter.rb
+- examples/helper.rb
+- examples/slugizer.rb
+- examples/storage.rb
+- examples/types.rb
+- examples/update.rb
+- lib/mongomapper_ext.rb
+- lib/mongomapper_ext/file.rb
+- lib/mongomapper_ext/filter.rb
+- lib/mongomapper_ext/slugizer.rb
+- lib/mongomapper_ext/storage.rb
+- lib/mongomapper_ext/types/open_struct.rb
+- lib/mongomapper_ext/types/set.rb
+- lib/mongomapper_ext/types/timestamp.rb
+- lib/mongomapper_ext/update.rb
+- test/helper.rb
+- test/models.rb
+- test/support/custom_matchers.rb
+- test/test_filter.rb
+- test/test_slugizer.rb
+- test/test_storage.rb
+- test/test_update.rb
+- test/types/test_open_struct.rb
+- test/types/test_set.rb
+- test/types/test_timestamp.rb
+has_rdoc: true
+homepage: http://github.com/dcu/mongomapper_ext
+licenses: []
+
+post_install_message: 
+rdoc_options: 
+- --charset=UTF-8
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.5
+signing_key: 
+specification_version: 3
+summary: MongoMapper extensions
+test_files: 
+- test/test_slugizer.rb
+- test/test_filter.rb
+- test/test_storage.rb
+- test/test_update.rb
+- test/support/custom_matchers.rb
+- test/models.rb
+- test/helper.rb
+- test/types/test_open_struct.rb
+- test/types/test_timestamp.rb
+- test/types/test_set.rb
+- examples/types.rb
+- examples/filter.rb
+- examples/update.rb
+- examples/storage.rb
+- examples/helper.rb
+- examples/slugizer.rb