dummy 0.5

data/lib/dummy/phone_number.rb

@@ -0,0 +1,30 @@
+module Dummy
+  module PhoneNumber
+    extend self
+
+    def phone_number
+      Dummy.numerify case rand(20)
+      when      0 then "###-###-#### x#####"
+      when      1 then "###-###-#### x####"
+      when      2 then "###-###-#### x###"
+      when   3..4 then "###-###-####"
+      when      5 then "###.###.#### x#####"
+      when      6 then "###.###.#### x####"
+      when      7 then "###.###.#### x###"
+      when   8..9 then "###.###.####"
+      when     10 then "(###)###-#### x#####"
+      when     11 then "(###)###-#### x####"
+      when     12 then "(###)###-#### x###"
+      when 13..14 then "(###)###-####"
+      when     15 then "1-###-###-#### x#####"
+      when     16 then "1-###-###-#### x####"
+      when     17 then "1-###-###-#### x###"
+      when 18..19 then "1-###-###-####"
+      end
+    end
+
+    def phone_number_short
+      Dummy.numerify("###-###-####")
+    end
+  end
+end

data/lib/generators/dummy_generator.rb

@@ -0,0 +1,185 @@
+require "yaml"
+require "active_support"
+require "active_record"
+require "rails/generators"
+
+module Dummy
+  class DataGenerator < Rails::Generators::Base
+    def self.source_root
+      @source_root ||= File.expand_path("../templates", __FILE__)
+    end
+    
+    class_option :base_amount, :type => :numeric, :default => 10,
+                :desc => "The base amount of records to generate for each model."
+    class_option :growth_ratio, :type => :numeric, :default => 2.0,
+                :desc => "The growth ratio of each model, according to its associations."
+
+    def install_dummy
+      initialize_application
+      generate_dummy_data
+      copy_rake_file
+    end
+
+    private
+
+    def initialize_application
+      require File.expand_path("#{Rails.root}/config/environment.rb")
+      say_status :successful, "initialize Rails application"
+    end
+
+    def generate_dummy_data
+      get_table_names
+      gather_associations
+      predict_record_amounts
+      generate_and_write_data
+    end
+
+    def get_table_names
+      @models = Hash.new
+      Dir["app/models/*.rb"].each do |full_path|
+        model = File.basename(full_path).chomp(File.extname(full_path)).camelcase.constantize
+        @models.merge!({model => {
+          :record_amount => 0, :associations => []
+        }}) if model.respond_to?(:columns)
+      end
+    end
+
+    def gather_associations
+      @models.each_key do |model|
+        model_symbol = model.to_s.underscore.pluralize.to_sym
+        associations = model.reflect_on_all_associations(:belongs_to)
+        
+        associations.each do |assoc|
+          assoc_name = assoc.name.to_s.camelcase
+          assoc_options = assoc.options
+
+          if assoc_options.empty?
+            @models[model][:associations].push({
+              :model  => assoc_name.constantize,
+              :foreign_key => "#{assoc_name.underscore}_id"
+            })
+          elsif assoc_options.has_key?(:class_name) and assoc_options.has_key?(:foreign_key)
+            @models[model][:associations].push({
+              :model => assoc_options[:class_name].constantize, # TODO: handle class_name
+              :foreign_key => assoc_options[:foreign_key]
+            })
+          else
+            next
+          end
+
+          assoc_model = @models[model][:associations].last[:model]
+          assoc_reflections = assoc_model.reflect_on_all_associations(:has_one)
+          @models[model][:associations].pop if assoc_reflections.map(&:name).include?(model_symbol)
+        end
+      end
+    end
+
+    def predict_record_amounts
+      models = @models.dup
+      models.each do |model, info|
+        predict_record_amount(model, info, models, [])
+      end
+    end
+
+    def predict_record_amount(model, info, models, stacked_models)
+      info[:associations].each do |assoc|
+        next if stacked_models.include?(assoc[:model])
+
+        if model != assoc[:model]
+          stacked_models << assoc[:model]
+          predict_record_amount(assoc[:model], @models[assoc[:model]], models, stacked_models)
+        end
+      end
+
+      amount = options.base_amount
+      if not info[:associations].empty?
+        amount = info[:associations].map do |assoc|
+          @models[assoc[:model]][:record_amount]
+        end.max*options.growth_ratio # **info[:associations].size
+      end
+
+      @models[model][:record_amount] = amount.to_i
+      stacked_models.delete(model)
+      models.delete(model)
+    end
+
+    def generate_and_write_data
+      empty_directory "test/dummy"
+      data = Hash.new
+
+      @models.each do |model, info|
+        name = model.to_s.underscore
+        
+        (0..info[:record_amount]-1).each do |num|
+          key_value = Hash.new
+          fixture_data = Hash.new
+
+          model.columns.each do |column|
+            key_value = generate_record_data(name, info, column)
+            fixture_data.merge!(key_value) unless key_value.nil?
+          end
+          
+          data[model.table_name] = Hash.new if data[model.table_name].nil?
+          data[model.table_name].merge!({ "#{name}_#{num}" => fixture_data })
+        end
+
+        say_status :successful, "generate #{info[:record_amount]} records for '#{name}'"
+      end
+      
+      data.each do |name, fixtures|
+        content = "# '#{name}' data generated automatically by dummy at #{Time.now.strftime("%H:%M %m/%d/%Y")} (#{fixtures.size} records).\n"
+        
+        content << YAML.dump(fixtures)
+        
+        create_file "test/dummy/#{name}.yml", content
+      end
+      say_status :successful, "store fixtures"
+    end
+
+    def generate_record_data(name, info, column)
+      column_name = column.name
+      if(column_name =~ /_at$/ and column.type == :datetime) or column_name == "id"
+        return
+      end
+
+      associated_model = associated_class_name(info, column_name)
+
+      if associated_model
+        val = generate_association_data(associated_model)
+        column_name.gsub!(/_id$/, "")
+	    else
+		    val = generate_regular_data(column)
+      end
+
+	    {column_name => val}
+    end
+
+    def associated_class_name(info, name)
+      info[:associations].each do |assoc|
+        return assoc[:model] if assoc[:foreign_key] == name
+      end
+      false
+    end
+
+    def generate_association_data(associated_model)
+      random_record_num = rand(@models[associated_model][:record_amount])
+	    "#{associated_model.to_s.underscore}_#{random_record_num}"
+    end
+
+    def generate_regular_data(column)
+      val = Dummy.magic_data(column.name, column.type)
+      
+      if val
+        val
+      else
+        say_status :failed, "data generation for '#{column.name}' with type '#{column.type.to_s.downcase}'", :red
+        ""
+      end
+    end
+
+    def copy_rake_file
+      copy_file "dummy.rake", "lib/tasks/dummy.rake"
+    end
+  end
+end
+

data/lib/generators/templates/dummy.rake

@@ -0,0 +1,13 @@
+require 'active_record/fixtures'
+
+namespace :dummy do
+  namespace :data do
+    desc "Load the generated dummy data into the current environment's database."
+    task :import => :environment do
+      Fixtures.reset_cache
+      fixtures_folder = File.join(Rails.root, 'test', 'dummy')
+      fixtures = Dir[File.join(fixtures_folder, '*.yml')].map {|f| File.basename(f, '.yml') }
+      Fixtures.create_fixtures(fixtures_folder, fixtures)
+    end
+  end
+end

metadata

@@ -0,0 +1,93 @@
+--- !ruby/object:Gem::Specification 
+name: dummy
+version: !ruby/object:Gem::Version 
+  prerelease: false
+  segments: 
+  - 0
+  - 5
+  version: "0.5"
+platform: ruby
+authors: 
+- "Gon\xC3\xA7alo Silva"
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2010-07-29 00:00:00 +01:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: rails
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 3
+        - 0
+        - 0
+        - rc
+        version: 3.0.0.rc
+  type: :runtime
+  version_requirements: *id001
+description: Generates test routes and consistent fake data, inserts it to the database and runs performance tests with them
+email: 
+- goncalossilva@gmail.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- lib/dummy/lorem.rb
+- lib/dummy/address.rb
+- lib/dummy/company.rb
+- lib/dummy/name.rb
+- lib/dummy/internet.rb
+- lib/dummy/geolocation.rb
+- lib/dummy/phone_number.rb
+- lib/generators/dummy_generator.rb
+- lib/dummy.rb
+- lib/generators/templates/dummy.rake
+- LICENSE
+- README.md
+- CHANGELOG.md
+has_rdoc: true
+homepage: http://github.com/goncalossilva/dummy
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 1
+      - 3
+      - 7
+      version: 1.3.7
+requirements: []
+
+rubyforge_project: dummy
+rubygems_version: 1.3.7
+signing_key: 
+specification_version: 3
+summary: Generates fake data and test routes and runs performance tests with it
+test_files: []
+