population 0.0.3

data/.gitignore

@@ -0,0 +1,20 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp
+*.swp
+*.swo
+.DS_Store

data/.rspec

@@ -0,0 +1 @@
+--color

data/Gemfile

@@ -0,0 +1,4 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in population.gemspec
+gemspec

data/LICENSE

@@ -0,0 +1,22 @@
+Copyright (c) 2012 David Tuite
+
+MIT License
+
+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.md

@@ -0,0 +1,80 @@
+# Population
+
+Population is a gem which provides a simple syntax for
+populating and resetting your database tables.
+
+ - [Public Development Trello](https://trello.com/board/populations/4ffc4c6d685b0d606219fec6)
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+    gem 'population'
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install population
+
+## Usage
+
+First define some population recipes in `lib/populations.rb`.
+
+```ruby
+# lib/populations.rb
+Population.define do
+  populate :users do
+    create!(name: "David Tuite", phone: '34893')
+    create!(name: "Peter Pan", phone: '093489')
+  end
+
+  populate :states do
+    create!(name: 'Alabama')
+  end
+end
+```
+
+Now you can use the `Population` class to populate and reset your
+database tables.
+
+**Populating**
+
+```
+# From the console:
+$ rails c
+: Population.populate(:users)
+
+# From a task
+# lib/tasts/populate.rake
+namespace :db do
+  desc "Populate the DB with sample data"
+  task :populate => :environment do
+    tables = [:users, :states]
+    Population.populate(*tables)
+  end
+end
+
+$ rake db:populate
+```
+
+**Resetting**
+
+Resetting will empty the passed in tables and reset any auto-increment
+columns to 1.
+
+```
+# From the console:
+$ rails c
+: Population.reset(:users)
+```
+
+## Contributing
+
+1. Fork it
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Add some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create new Pull Request

data/Rakefile

@@ -0,0 +1,2 @@
+#!/usr/bin/env rake
+require "bundler/gem_tasks"

data/lib/population/configuration.rb

@@ -0,0 +1,9 @@
+module Population
+  class Configuration
+    attr_reader :populations
+
+    def initialize
+      @populations = {}
+    end
+  end
+end

data/lib/population/database.rb

@@ -0,0 +1,39 @@
+module Population
+  class Database
+
+    # Destroy all records belonging to a class and reset the
+    # auto-increment ID to 1.
+    def self.reset_table_belonging_to(klass)
+      klass.destroy_all
+      reset_command = reset_command_for(klass)
+      ActiveRecord::Base.connection.execute(reset_command)
+    end
+
+    def self.reset_command_for(klass)
+      table_name = klass.name.tableize
+
+      if sqlite?
+        "delete from sqlite_sequence where name='#{table_name}';" 
+      elsif mysql?
+        "ALTER TABLE #{table_name} AUTO_INCREMENT = 1;" 
+      elsif postgres?
+        "SELECT setval('public.#{table_name}_id_seq', 1, false)" 
+      end
+    end
+
+    # Check if we're dealing with a SQLite database.
+    def self.sqlite?
+      ActiveRecord::Base.connection.adapter_name == 'SQLite'
+    end
+
+    # Check if we're dealing with a PostgreSQL database.
+    def self.postgres?
+      ActiveRecord::Base.connection.adapter_name == 'PostgreSQL'
+    end
+
+    # Check if we're dealing with a MySQL database.
+    def self.mysql?
+      ActiveRecord::Base.connection.adapter_name == 'MySQL'
+    end
+  end
+end

data/lib/population/definition.rb

@@ -0,0 +1,20 @@
+module Population
+  module Definition
+    def define(&block)
+      DSL.run(block)
+    end
+
+    class DSL
+      def populate(name, &block)
+        populate = Populate.new(name, &block)
+        Population.register_populate(populate)
+      end
+
+      def self.run(block)
+        new.instance_eval(&block)
+      end
+    end
+  end
+
+  extend Definition
+end

data/lib/population/find_definitions.rb

@@ -0,0 +1,22 @@
+module Population
+  class << self
+    attr_accessor :definition_file_paths
+  end
+
+  self.definition_file_paths = %w(populations lib/populations)
+
+  def self.find_definitions
+    absolute_definition_file_paths = 
+      definition_file_paths.map {|path| File.expand_path(path) }
+
+    absolute_definition_file_paths.uniq.each do |path|
+      load("#{path}.rb") if File.exists?("#{path}.rb")
+
+      if File.directory? path
+        Dir[File.join(path, '**', '*.rb')].sort.each do |file|
+          load file
+        end
+      end
+    end
+  end
+end

data/lib/population/populate.rb

@@ -0,0 +1,19 @@
+module Population
+  class Populate
+    attr_reader :name, :klass, :block
+
+    def initialize(name, &block)
+      @name = name
+      @klass = name.to_s.classify.constantize
+      @block = block
+    end
+
+    def up
+      @klass.instance_eval(&block)
+    end
+
+    def down
+      Population::Database.reset_table_belonging_to(klass)
+    end
+  end
+end

data/lib/population/railtie.rb

@@ -0,0 +1,16 @@
+require "rails"
+
+module Population
+  class Railtie < Rails::Railtie
+    initializer "population.set_population_paths" do
+      Population.definition_file_paths = [
+          File.join(Rails.root, 'populations'),
+          File.join(Rails.root, 'lib', 'populations')
+      ]
+    end
+
+    config.after_initialize do
+      Population.find_definitions
+    end
+  end
+end

data/lib/population/version.rb

@@ -0,0 +1,3 @@
+module Population
+  VERSION = "0.0.3"
+end

data/lib/population.rb

@@ -0,0 +1,66 @@
+require 'active_support/core_ext/module/delegation'
+require "active_record"
+
+require "population/populate"
+require "population/configuration"
+require "population/database"
+require "population/definition"
+require "population/find_definitions"
+require "population/railtie"
+require "population/version"
+
+module Population
+
+  def self.configuration
+    @configuration ||= Configuration.new
+  end
+
+  # Delegate these method to configuration so they can be used
+  # as if they were class methods on Population.
+  class << self
+    delegate :populations, to: :configuration
+  end
+
+  # Register a populate in the populations register.
+  def self.register_populate(populate)
+    # Register the populate definition in a hash of all populates.
+    populations[populate.name] = populate
+    populate
+  end
+
+  # Reset one or many tables specified by their class Constant.
+  #
+  # Arguments:
+  #
+  #  - klasses - One or many class constants.
+  #
+  # Examples:
+  #
+  #  Population.reset(User)
+  #
+  #  Population.reset(User, Article)
+  #
+  def self.reset(*table_names)
+    table_names.each do |name|
+      populations[name].down
+    end
+  end
+
+  # Populate one or many tables specified by their table name.
+  #
+  # Arguments:
+  #
+  #  - klasses - One or many table names.
+  #
+  # Examples:
+  #
+  #  Population.populate(:users)
+  #
+  #  Population.populate(:users, :articles)
+  #
+  def self.populate(*table_names)
+    table_names.each do |name|
+      populations[name].up
+    end
+  end
+end

data/population.gemspec

@@ -0,0 +1,25 @@
+# -*- encoding: utf-8 -*-
+require File.expand_path('../lib/population/version', __FILE__)
+
+Gem::Specification.new do |gem|
+  gem.authors       = ["David Tuite"]
+  gem.email         = ["dtuite@gmail.com"]
+  gem.description   = %q{Population for rails apps.}
+  gem.summary       = %q{Populate or reset your AcriveRecord models based on FactoryGirl like definitions.}
+  gem.homepage      = ""
+
+  gem.files         = `git ls-files`.split($\)
+  gem.executables   = gem.files.grep(%r{^bin/}).map{ |f| File.basename(f) }
+  gem.test_files    = gem.files.grep(%r{^(test|spec|features)/})
+  gem.name          = "population"
+  gem.require_paths = ["lib"]
+  gem.version       = Population::VERSION
+
+  gem.add_dependency "activesupport"
+  gem.add_dependency "activerecord"
+  gem.add_dependency "railties", ">= 3.0.0"
+
+  gem.add_development_dependency "rspec"
+  gem.add_development_dependency "sqlite3"
+  gem.add_development_dependency "rails", "3.0.7"
+end

data/spec/integration/population_spec.rb

@@ -0,0 +1,28 @@
+require "spec_helper"
+
+describe "Population" do
+  before :all do
+    Population.define do
+      populate :users do
+        create!(name: "Peter")
+        create!(name: "John")
+      end
+    end
+  end
+
+  describe "populate" do
+    it "should populate database tables" do
+      expect do
+        Population.populate(:users)
+      end.to change(User, :count).by(2)
+    end
+  end
+
+  describe "reset" do
+    it "should reset database tables" do
+      expect do
+        Population.reset(:users)
+      end.to change(User, :count).by(-2)
+    end
+  end
+end

data/spec/population/database_spec.rb

@@ -0,0 +1,70 @@
+require "spec_helper"
+
+describe Population::Database do
+  subject { Population::Database }
+
+  describe "reset_table_belonging_to" do
+    # Prevent commands from actually running on the DB.
+    before { ActiveRecord::Base.connection.stub(:execute) }
+
+    it "should call :destroy_all on the class" do
+      User.should_receive(:destroy_all)
+      subject.reset_table_belonging_to(User)
+    end
+
+    it "should delegate to reset_command_for" do
+      subject.should_receive(:reset_command_for).with(User)
+      subject.reset_table_belonging_to(User)
+    end
+
+    it "should execute the command" do
+      command = "hello"
+      subject.stub(:reset_command_for) { command }
+      ActiveRecord::Base.connection.should_receive(:execute)
+        .with(command)
+      subject.reset_table_belonging_to(User)
+    end
+  end
+
+  describe "sqlite?" do
+    let(:connection) { ActiveRecord::Base.connection }
+
+    it "should return true if the adapter_name is Sqlite" do
+      connection.stub(:adapter_name) { 'SQLite' }
+      subject.should be_sqlite
+    end
+
+    it "should return false if the adapter_name is not Sqlite" do
+      connection.stub(:adapter_name) { 'Grinnick' }
+      subject.should_not be_sqlite
+    end
+  end
+
+  describe "mysql?" do
+    let(:connection) { ActiveRecord::Base.connection }
+
+    it "should return true if the adapter_name is MySQL" do
+      connection.stub(:adapter_name) { 'MySQL' }
+      subject.should be_mysql
+    end
+
+    it "should return false if the adapter_name is not MySQL" do
+      connection.stub(:adapter_name) { 'Grinnick' }
+      subject.should_not be_mysql
+    end
+  end
+
+  describe "postgres?" do
+    let(:connection) { ActiveRecord::Base.connection }
+
+    it "should return true if the adapter_name is PostgreSQL" do
+      connection.stub(:adapter_name) { 'PostgreSQL' }
+      subject.should be_postgres
+    end
+
+    it "should return false if the adapter_name is not PostgreSQL" do
+      connection.stub(:adapter_name) { 'Grinnick' }
+      subject.should_not be_postgres
+    end
+  end
+end

data/spec/population/definition_spec.rb

@@ -0,0 +1,40 @@
+require "spec_helper"
+
+describe Population::Definition do
+  describe "define" do
+    it "should delegate to DSL.run" do
+      block = Proc.new { "Hello" }
+      Population::Definition::DSL.should_receive(:run).with(block)
+      Population.define(&block)
+    end
+  end
+
+  describe "populate" do
+    it "should create a Populate with a name" do
+      block = Proc.new { "Hello" }
+      Population.define do
+        populate(:users, &block)
+      end
+      Population.populations[:users].name.should == :users
+    end
+
+    it "should pass the block to the Populate instance" do
+      block = Proc.new { "Hello" }
+      Population.define do
+        populate(:users, &block)
+      end
+      Population.populations[:users].block.should == block
+    end
+
+    it "should register the populate" do
+      pop = stub
+      Population::Populate.stub(:new) { pop }
+      Population.should_receive(:register_populate).with(pop)
+
+      block = Proc.new { "Hello" }
+      Population.define do
+        populate(:users, &block)
+      end
+    end
+  end
+end

data/spec/population/populate_spec.rb

@@ -0,0 +1,40 @@
+require "spec_helper"
+
+describe Population::Populate do
+  subject { Population::Populate }
+
+  it "should have a name" do
+    pop = subject.new(:users)
+    pop.name.should == :users
+  end
+
+  it "should have a klass" do
+    pop = subject.new(:users)
+    pop.klass.should == User
+  end
+
+  it "should take a block" do
+    block = Proc.new { p "hello" }
+    pop = subject.new(:users, &block)
+    pop.block.should == block
+  end
+
+  describe "up" do
+    it "should call the block" do
+      User.should_receive(:create!).with(name: "Peter")
+      pop = subject.new :users do
+        create!(name: "Peter") 
+      end
+      pop.up
+    end
+  end
+
+  describe "down" do
+    it "should reset the table" do
+      Population::Database.should_receive(:reset_table_belonging_to).with(User)
+      block = Proc.new { p "hello" }
+      pop = subject.new(:users, &block)
+      pop.down
+    end
+  end
+end

data/spec/population_spec.rb

@@ -0,0 +1,19 @@
+require "spec_helper"
+
+describe Population do
+  it "should have a configuration" do
+    Population.configuration.should be_instance_of(Population::Configuration)
+  end
+
+  it "should have access to populates" do
+    Population.populations.should be_instance_of(Hash)
+  end
+
+  describe "register_populate" do
+    it "should register a populate" do
+      pop = Population::Populate.new(:users)
+      Population.register_populate(pop)
+      Population.populations[:users].should == pop
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,2 @@
+require "population"
+require "support/active_record"

data/spec/support/active_record.rb

@@ -0,0 +1,10 @@
+ActiveRecord::Base.establish_connection(adapter: 'sqlite3', database: ':memory:')
+
+class User < ActiveRecord::Base
+  attr_accessible :name
+end
+
+ActiveRecord::Migration.create_table :users do |t|
+  t.string :name
+end
+

metadata

@@ -0,0 +1,140 @@
+--- !ruby/object:Gem::Specification
+name: population
+version: !ruby/object:Gem::Version
+  version: 0.0.3
+  prerelease: 
+platform: ruby
+authors:
+- David Tuite
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-07-10 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: activesupport
+  requirement: &70332471984980 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *70332471984980
+- !ruby/object:Gem::Dependency
+  name: activerecord
+  requirement: &70332471983600 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *70332471983600
+- !ruby/object:Gem::Dependency
+  name: railties
+  requirement: &70332471981500 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 3.0.0
+  type: :runtime
+  prerelease: false
+  version_requirements: *70332471981500
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: &70332471979140 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: *70332471979140
+- !ruby/object:Gem::Dependency
+  name: sqlite3
+  requirement: &70332471994300 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: *70332471994300
+- !ruby/object:Gem::Dependency
+  name: rails
+  requirement: &70332471993060 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - =
+      - !ruby/object:Gem::Version
+        version: 3.0.7
+  type: :development
+  prerelease: false
+  version_requirements: *70332471993060
+description: Population for rails apps.
+email:
+- dtuite@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- .rspec
+- Gemfile
+- LICENSE
+- README.md
+- Rakefile
+- lib/population.rb
+- lib/population/configuration.rb
+- lib/population/database.rb
+- lib/population/definition.rb
+- lib/population/find_definitions.rb
+- lib/population/populate.rb
+- lib/population/railtie.rb
+- lib/population/version.rb
+- population.gemspec
+- spec/integration/population_spec.rb
+- spec/population/database_spec.rb
+- spec/population/definition_spec.rb
+- spec/population/populate_spec.rb
+- spec/population_spec.rb
+- spec/spec_helper.rb
+- spec/support/active_record.rb
+homepage: ''
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.16
+signing_key: 
+specification_version: 3
+summary: Populate or reset your AcriveRecord models based on FactoryGirl like definitions.
+test_files:
+- spec/integration/population_spec.rb
+- spec/population/database_spec.rb
+- spec/population/definition_spec.rb
+- spec/population/populate_spec.rb
+- spec/population_spec.rb
+- spec/spec_helper.rb
+- spec/support/active_record.rb