← Back to branch summary

~ubuntu-branches/ubuntu/utopic/ruby-database-cleaner/utopic-proposed

« back to all changes in this revision

Viewing changes to examples/features/support/env.rb

  • Committer: Package Import Robot
  • Author(s): Pirate Praveen
  • Date: 2014-05-24 01:23:06 UTC
  • Revision ID: package-import@ubuntu.com-20140524012306-1nn3zh7s867f2gn9
Tags: upstream-1.2.0
Import upstream version 1.2.0

Show diffs side-by-side

added added

removed removed

Lines of Context:
examples/features/support/env.rb
 
1
#Hilarious as it seems, this is necessary so bundle exec cucumber works for mongoid cukeage (I'm assuming mongomapper is automatically present because its a git repo)
 
2
Object.send(:remove_const, 'MongoMapper')  if defined?(::MongoMapper)
 
3
 
 
4
require 'rubygems'
 
5
require 'bundler'
 
6
 
 
7
Bundler.setup
 
8
require 'rspec/expectations'
 
9
#require 'ruby-debug'
 
10
 
 
11
DB_DIR = "#{File.dirname(__FILE__)}/../../db"
 
12
 
 
13
orm         = ENV['ORM']
 
14
another_orm = ENV['ANOTHER_ORM']
 
15
strategy    = ENV['STRATEGY']
 
16
multiple_db = ENV['MULTIPLE_DBS']
 
17
 
 
18
config = YAML::load(File.open("#{File.dirname(__FILE__)}/../../config/redis.yml"))
 
19
ENV['REDIS_URL'] = config['test']['url']
 
20
ENV['REDIS_URL_ONE'] = config['one']['url']
 
21
ENV['REDIS_URL_TWO'] = config['two']['url']
 
22
 
 
23
if orm && strategy
 
24
  $:.unshift(File.dirname(__FILE__) + '/../../../lib')
 
25
  require 'database_cleaner'
 
26
  require 'database_cleaner/cucumber'
 
27
  require "#{File.dirname(__FILE__)}/../../lib/#{orm.downcase}_models"
 
28
 
 
29
  if another_orm
 
30
    require "#{File.dirname(__FILE__)}/../../lib/#{another_orm.downcase}_models"
 
31
  end
 
32
 
 
33
 
 
34
 
 
35
 
 
36
  if multiple_db
 
37
    DatabaseCleaner.app_root = "#{File.dirname(__FILE__)}/../.."
 
38
    orm_sym = orm.gsub(/(.)([A-Z]+)/,'\1_\2').downcase.to_sym
 
39
 
 
40
    case orm_sym
 
41
    when :mongo_mapper
 
42
      DatabaseCleaner[ orm_sym, {:connection => 'database_cleaner_test_one'} ].strategy = strategy.to_sym
 
43
      DatabaseCleaner[ orm_sym, {:connection => 'database_cleaner_test_two'} ].strategy = strategy.to_sym
 
44
    when :redis, :ohm
 
45
      DatabaseCleaner[ orm_sym, {:connection => ENV['REDIS_URL_ONE']} ].strategy = strategy.to_sym
 
46
      DatabaseCleaner[ orm_sym, {:connection => ENV['REDIS_URL_TWO']} ].strategy = strategy.to_sym
 
47
    when :active_record
 
48
      DatabaseCleaner[:active_record, {:model => ActiveRecordWidgetUsingDatabaseOne} ].strategy = strategy.to_sym
 
49
      DatabaseCleaner[:active_record, {:model => ActiveRecordWidgetUsingDatabaseTwo} ].strategy = strategy.to_sym
 
50
    else
 
51
      DatabaseCleaner[ orm_sym, {:connection => :one} ].strategy = strategy.to_sym
 
52
      DatabaseCleaner[ orm_sym, {:connection => :two} ].strategy = strategy.to_sym
 
53
    end
 
54
 
 
55
  elsif another_orm
 
56
    DatabaseCleaner[         orm.gsub(/(.)([A-Z]+)/,'\1_\2').downcase.to_sym ].strategy = strategy.to_sym
 
57
    DatabaseCleaner[ another_orm.gsub(/(.)([A-Z]+)/,'\1_\2').downcase.to_sym ].strategy = strategy.to_sym
 
58
  else
 
59
    DatabaseCleaner.strategy = strategy.to_sym unless strategy == "default"
 
60
  end
 
61
 
 
62
else
 
63
  raise "Run 'ORM=ActiveRecord|DataMapper|MongoMapper|CouchPotato|Ohm|Redis [ANOTHER_ORM=...] [MULTIPLE_DBS=true] STRATEGY=transaction|truncation|default cucumber examples/features'"
 
64
end