activerecord-postgresql-expression 0.0.1
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- checksums.yaml +7 -0
- data/.gitignore +15 -0
- data/.travis.yml +12 -0
- data/Gemfile +6 -0
- data/LICENSE.txt +22 -0
- data/README.md +64 -0
- data/Rakefile +64 -0
- data/activerecord-postgresql-expression.gemspec +28 -0
- data/lib/activerecord-postgresql-expression.rb +1 -0
- data/lib/activerecord-postgresql-expression/active_record/connection_adapters/postgresql/schema_statements.rb +30 -0
- data/lib/activerecord-postgresql-expression/active_record/schema_dumper.rb +60 -0
- data/lib/activerecord/postgresql/expression.rb +15 -0
- data/lib/activerecord/postgresql/expression/base.rb +6 -0
- data/lib/activerecord/postgresql/expression/railtie.rb +13 -0
- data/lib/activerecord/postgresql/expression/version.rb +7 -0
- data/test/.gitignore +1 -0
- data/test/cases/expression_test.rb +31 -0
- data/test/cases/helper.rb +51 -0
- data/test/cases/test_case.rb +123 -0
- data/test/config.example.yml +12 -0
- data/test/config.rb +5 -0
- data/test/support/config.rb +43 -0
- data/test/support/connection.rb +18 -0
- data/test/support/connection_helper.rb +8 -0
- data/test/support/schema_dumping_helper.rb +20 -0
- metadata +148 -0
checksums.yaml
ADDED
@@ -0,0 +1,7 @@
|
|
1
|
+
---
|
2
|
+
SHA1:
|
3
|
+
metadata.gz: 50d0a7eb8c7e4a647212668965c9a1edfa46b1bd
|
4
|
+
data.tar.gz: 17a7e65e997ed922ee6732202d1ddeb2bc204ce5
|
5
|
+
SHA512:
|
6
|
+
metadata.gz: 3c4b748ba4b5f70faca61de39027ec105f34921a22cc00c921f6de43d1e9901c71ed99e142e4e644b5cd5c4da1cc42027b411298221fc2ec6f5ca1e3718236ff
|
7
|
+
data.tar.gz: 00000c7c184cf29da52c28af87f35caa1df13de7a23d8fc8b7ff232930bd91ac88b91dda2ff753b4d9049ed83326370a3d7ed771afdb02b3feef77652270fe41
|
data/.gitignore
ADDED
data/.travis.yml
ADDED
data/Gemfile
ADDED
data/LICENSE.txt
ADDED
@@ -0,0 +1,22 @@
|
|
1
|
+
Copyright (c) 2015 Naoya Murakami
|
2
|
+
|
3
|
+
MIT License
|
4
|
+
|
5
|
+
Permission is hereby granted, free of charge, to any person obtaining
|
6
|
+
a copy of this software and associated documentation files (the
|
7
|
+
"Software"), to deal in the Software without restriction, including
|
8
|
+
without limitation the rights to use, copy, modify, merge, publish,
|
9
|
+
distribute, sublicense, and/or sell copies of the Software, and to
|
10
|
+
permit persons to whom the Software is furnished to do so, subject to
|
11
|
+
the following conditions:
|
12
|
+
|
13
|
+
The above copyright notice and this permission notice shall be
|
14
|
+
included in all copies or substantial portions of the Software.
|
15
|
+
|
16
|
+
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
|
17
|
+
EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
18
|
+
MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
|
19
|
+
NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
|
20
|
+
LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
|
21
|
+
OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
|
22
|
+
WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
data/README.md
ADDED
@@ -0,0 +1,64 @@
|
|
1
|
+
# ActiveRecord::PostgreSQL::Expression
|
2
|
+
|
3
|
+
[](https://travis-ci.org/naoa/activerecord-postgresql-expression)
|
4
|
+
|
5
|
+
Adds expression to migrations for ActiveRecord PostgreSQL adapters
|
6
|
+
|
7
|
+
## Installation
|
8
|
+
|
9
|
+
Add this line to your application's Gemfile:
|
10
|
+
|
11
|
+
```ruby
|
12
|
+
gem 'activerecord-postgresql-expression'
|
13
|
+
```
|
14
|
+
|
15
|
+
And then execute:
|
16
|
+
|
17
|
+
$ bundle
|
18
|
+
|
19
|
+
Or install it yourself as:
|
20
|
+
|
21
|
+
$ gem install activerecord-postgresql-expression
|
22
|
+
|
23
|
+
## Usage
|
24
|
+
|
25
|
+
```ruby
|
26
|
+
bundle exec rails g migration CreatePosts
|
27
|
+
```
|
28
|
+
|
29
|
+
```ruby
|
30
|
+
class CreatePosts < ActiveRecord::Migration
|
31
|
+
def change
|
32
|
+
create_table :posts do |t|
|
33
|
+
t.string :title
|
34
|
+
t.text :content
|
35
|
+
t.timestamps
|
36
|
+
end
|
37
|
+
add_index :posts, :content, using: :gin, expression: 'to_tsvector(\'english\'::regconfig, (content)::text)'
|
38
|
+
end
|
39
|
+
end
|
40
|
+
```
|
41
|
+
|
42
|
+
```ruby
|
43
|
+
bundle exec rake db:migrate
|
44
|
+
```
|
45
|
+
|
46
|
+
```ruby
|
47
|
+
create_table "posts", force: :cascade do |t|
|
48
|
+
t.string "title"
|
49
|
+
t.text "content"
|
50
|
+
t.datetime "created_at"
|
51
|
+
t.datetime "updated_at"
|
52
|
+
end
|
53
|
+
|
54
|
+
add_index "posts", ["content"], name: "index_posts_on_content", using: :gin, expression: "to_tsvector('english'::regconfig, content)"
|
55
|
+
```
|
56
|
+
|
57
|
+
## Contributing
|
58
|
+
|
59
|
+
1. Fork it ( https://github.com/naoa/activerecord-postgresql-expression/fork )
|
60
|
+
2. Create your feature branch (`git checkout -b my-new-feature`)
|
61
|
+
3. Commit your changes (`git commit -am 'Add some feature'`)
|
62
|
+
4. Push to the branch (`git push origin my-new-feature`)
|
63
|
+
5. Create a new Pull Request
|
64
|
+
|
data/Rakefile
ADDED
@@ -0,0 +1,64 @@
|
|
1
|
+
require "bundler/gem_tasks"
|
2
|
+
require 'rake/testtask'
|
3
|
+
|
4
|
+
require File.expand_path(File.dirname(__FILE__)) + "/test/config"
|
5
|
+
require File.expand_path(File.dirname(__FILE__)) + "/test/support/config"
|
6
|
+
|
7
|
+
desc 'Run postgresql tests by default'
|
8
|
+
task :default => :test
|
9
|
+
|
10
|
+
desc 'Run postgresql tests'
|
11
|
+
task :test => :test_postgresql
|
12
|
+
|
13
|
+
desc 'Build PostgreSQL test databases'
|
14
|
+
namespace :db do
|
15
|
+
task :create => ['db:postgresql:build']
|
16
|
+
task :drop => ['db:postgresql:drop']
|
17
|
+
end
|
18
|
+
|
19
|
+
%w( postgresql ).each do |adapter|
|
20
|
+
namespace :test do
|
21
|
+
Rake::TestTask.new(adapter => "#{adapter}:env") { |t|
|
22
|
+
t.libs << 'test'
|
23
|
+
t.test_files = Dir.glob( "test/cases/**/*_test.rb" ).sort
|
24
|
+
|
25
|
+
t.warning = true
|
26
|
+
t.verbose = true
|
27
|
+
}
|
28
|
+
end
|
29
|
+
|
30
|
+
namespace adapter do
|
31
|
+
task :test => "test_#{adapter}"
|
32
|
+
|
33
|
+
# Set the connection environment for the adapter
|
34
|
+
task(:env) { ENV['ARCONN'] = adapter }
|
35
|
+
end
|
36
|
+
|
37
|
+
# Make sure the adapter test evaluates the env setting task
|
38
|
+
task "test_#{adapter}" => ["#{adapter}:env", "test:#{adapter}"]
|
39
|
+
end
|
40
|
+
|
41
|
+
namespace :db do
|
42
|
+
namespace :postgresql do
|
43
|
+
desc 'Build the PostgreSQL test databases'
|
44
|
+
task :build do
|
45
|
+
config = ARTest.config['connections']['postgresql']
|
46
|
+
%x( createdb -U #{config['arunit']['username']} -E UTF8 -T template0 #{config['arunit']['database']} )
|
47
|
+
%x( createdb -U #{config['arunit']['username']} -E UTF8 -T template0 #{config['arunit2']['database']} )
|
48
|
+
end
|
49
|
+
|
50
|
+
desc 'Drop the PostgreSQL test databases'
|
51
|
+
task :drop do
|
52
|
+
config = ARTest.config['connections']['postgresql']
|
53
|
+
%x( dropdb #{config['arunit']['database']} )
|
54
|
+
%x( dropdb #{config['arunit2']['database']} )
|
55
|
+
end
|
56
|
+
|
57
|
+
desc 'Rebuild the PostgreSQL test databases'
|
58
|
+
task :rebuild => [:drop, :build]
|
59
|
+
end
|
60
|
+
end
|
61
|
+
|
62
|
+
task :build_postgresql_databases => 'db:postgresql:build'
|
63
|
+
task :drop_postgresql_databases => 'db:postgresql:drop'
|
64
|
+
task :rebuild_postgresql_databases => 'db:postgresql:rebuild'
|
@@ -0,0 +1,28 @@
|
|
1
|
+
# coding: utf-8
|
2
|
+
lib = File.expand_path('../lib', __FILE__)
|
3
|
+
$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
|
4
|
+
require 'activerecord/postgresql/expression/version'
|
5
|
+
|
6
|
+
Gem::Specification.new do |spec|
|
7
|
+
spec.name = "activerecord-postgresql-expression"
|
8
|
+
spec.version = ActiveRecord::PostgreSQL::Expression::VERSION
|
9
|
+
spec.authors = ["Naoya Murakami"]
|
10
|
+
spec.email = ["naoya@createfield.com"]
|
11
|
+
spec.summary = %q{Adds expression to migrations for ActiveRecord PostgreSQL adapters}
|
12
|
+
spec.description = %q{Adds expression to migrations for ActiveRecord PostgreSQL adapters}
|
13
|
+
spec.homepage = "https://github.com/naoa/activerecord-postgresql-expression"
|
14
|
+
spec.license = "MIT"
|
15
|
+
|
16
|
+
spec.files = `git ls-files -z`.split("\x0")
|
17
|
+
spec.executables = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
|
18
|
+
spec.test_files = spec.files.grep(%r{^(test|spec|features)/})
|
19
|
+
spec.require_paths = ["lib"]
|
20
|
+
|
21
|
+
spec.required_ruby_version = '>= 2.0.0'
|
22
|
+
|
23
|
+
spec.add_development_dependency "bundler", "~> 1.7"
|
24
|
+
spec.add_development_dependency "rake", "~> 10.0"
|
25
|
+
spec.add_runtime_dependency "activesupport", "~> 4.0"
|
26
|
+
spec.add_runtime_dependency "activerecord", "~> 4.0"
|
27
|
+
spec.add_runtime_dependency "pg"
|
28
|
+
end
|
@@ -0,0 +1 @@
|
|
1
|
+
require 'activerecord/postgresql/expression'
|
@@ -0,0 +1,30 @@
|
|
1
|
+
require 'active_record/connection_adapters/postgresql/schema_statements'
|
2
|
+
|
3
|
+
module ActiveRecord
|
4
|
+
module PostgreSQL
|
5
|
+
module Expression
|
6
|
+
module SchemaStatements
|
7
|
+
def add_index_options(table_name, column_name, options = {})
|
8
|
+
if options.key?(:expression)
|
9
|
+
expression = options[:expression]
|
10
|
+
options.delete(:expression)
|
11
|
+
end
|
12
|
+
|
13
|
+
index_name, index_type, index_columns, index_options, algorithm, using = super
|
14
|
+
|
15
|
+
if expression.present?
|
16
|
+
index_columns = expression
|
17
|
+
end
|
18
|
+
|
19
|
+
[index_name, index_type, index_columns, index_options, algorithm, using]
|
20
|
+
end
|
21
|
+
end
|
22
|
+
end
|
23
|
+
end
|
24
|
+
|
25
|
+
module ConnectionAdapters
|
26
|
+
class PostgreSQLAdapter < AbstractAdapter
|
27
|
+
prepend ActiveRecord::PostgreSQL::Expression::SchemaStatements
|
28
|
+
end
|
29
|
+
end
|
30
|
+
end
|
@@ -0,0 +1,60 @@
|
|
1
|
+
require 'active_record/schema_dumper'
|
2
|
+
|
3
|
+
module ActiveRecord
|
4
|
+
module PostgreSQL
|
5
|
+
module Expression
|
6
|
+
module SchemaDumper
|
7
|
+
private
|
8
|
+
def indexes(table, stream)
|
9
|
+
buf = StringIO.new
|
10
|
+
super(table, buf)
|
11
|
+
buf = buf.string.chomp
|
12
|
+
output = add_index_with_expression(table) + "\n"
|
13
|
+
stream.print buf
|
14
|
+
stream.print output
|
15
|
+
stream
|
16
|
+
end
|
17
|
+
|
18
|
+
def add_index_with_expression(table)
|
19
|
+
result = @connection.query(<<-SQL, 'SCHEMA')
|
20
|
+
SELECT distinct i.relname, pg_get_expr(d.indexprs, d.indrelid)
|
21
|
+
FROM pg_class t
|
22
|
+
INNER JOIN pg_index d ON t.oid = d.indrelid
|
23
|
+
INNER JOIN pg_class i ON d.indexrelid = i.oid
|
24
|
+
WHERE i.relkind = 'i'
|
25
|
+
AND d.indisprimary = 'f'
|
26
|
+
AND t.relname = '#{table}'
|
27
|
+
AND i.relnamespace IN (SELECT oid FROM pg_namespace WHERE nspname = ANY (current_schemas(false)) )
|
28
|
+
AND d.indexprs IS NOT NULL
|
29
|
+
ORDER BY i.relname
|
30
|
+
SQL
|
31
|
+
add_index_statements = result.map do |row|
|
32
|
+
index_name = row[0]
|
33
|
+
expression = row[1]
|
34
|
+
column_name = guess_column_name(index_name)
|
35
|
+
statement_parts = [
|
36
|
+
" add_index #{remove_prefix_and_suffix(table).inspect}",
|
37
|
+
"#{column_name}",
|
38
|
+
"name: #{index_name.inspect}",
|
39
|
+
"using: :gin",
|
40
|
+
"expression: #{expression.inspect}"
|
41
|
+
]
|
42
|
+
" #{statement_parts.join(', ')}"
|
43
|
+
end
|
44
|
+
add_index_statements.sort.join("\n")
|
45
|
+
end
|
46
|
+
def guess_column_name(index_name)
|
47
|
+
if matched = index_name.match(/_on_(?<name>.*)$/)
|
48
|
+
column_name = "[#{matched[:name].inspect}]"
|
49
|
+
else
|
50
|
+
column_name = "''"
|
51
|
+
end
|
52
|
+
end
|
53
|
+
end
|
54
|
+
end
|
55
|
+
end
|
56
|
+
|
57
|
+
class SchemaDumper #:nodoc:
|
58
|
+
prepend PostgreSQL::Expression::SchemaDumper
|
59
|
+
end
|
60
|
+
end
|
@@ -0,0 +1,15 @@
|
|
1
|
+
require 'active_support'
|
2
|
+
|
3
|
+
begin
|
4
|
+
require 'rails'
|
5
|
+
rescue LoadError
|
6
|
+
# nothing to do! yay!
|
7
|
+
end
|
8
|
+
|
9
|
+
if defined? Rails
|
10
|
+
require 'activerecord/postgresql/expression/railtie'
|
11
|
+
else
|
12
|
+
ActiveSupport.on_load :active_record do
|
13
|
+
require 'activerecord/postgresql/expression/base'
|
14
|
+
end
|
15
|
+
end
|
@@ -0,0 +1,6 @@
|
|
1
|
+
if ActiveRecord::VERSION::MAJOR == 4
|
2
|
+
require 'activerecord-postgresql-expression/active_record/schema_dumper'
|
3
|
+
require 'activerecord-postgresql-expression/active_record/connection_adapters/postgresql/schema_statements'
|
4
|
+
else
|
5
|
+
raise "activerecord-postgresql-expression supports activerecord ~> 4.x"
|
6
|
+
end
|
@@ -0,0 +1,13 @@
|
|
1
|
+
module ActiveRecord
|
2
|
+
module PostgreSQL
|
3
|
+
module Expression
|
4
|
+
class Railtie < Rails::Railtie
|
5
|
+
initializer 'activerecord-postgresql-expression' do
|
6
|
+
ActiveSupport.on_load :active_record do
|
7
|
+
require 'activerecord/postgresql/expression/base'
|
8
|
+
end
|
9
|
+
end
|
10
|
+
end
|
11
|
+
end
|
12
|
+
end
|
13
|
+
end
|
data/test/.gitignore
ADDED
@@ -0,0 +1 @@
|
|
1
|
+
/config.yml
|
@@ -0,0 +1,31 @@
|
|
1
|
+
require 'cases/helper'
|
2
|
+
require 'support/schema_dumping_helper'
|
3
|
+
|
4
|
+
class ExpressionTest < ActiveRecord::TestCase
|
5
|
+
include SchemaDumpingHelper
|
6
|
+
|
7
|
+
class Expression < ActiveRecord::Base
|
8
|
+
end
|
9
|
+
|
10
|
+
setup do
|
11
|
+
@connection = ActiveRecord::Base.connection
|
12
|
+
@connection.create_table("posts", force: true) do |t|
|
13
|
+
t.string "title"
|
14
|
+
t.text "content"
|
15
|
+
end
|
16
|
+
@connection.add_index :posts, :content, using: :gin, expression: 'to_tsvector(\'english\'::regconfig, (content)::text)'
|
17
|
+
@connection.add_index :posts, :content, name: 'hogehoge', using: :gin, expression: 'to_tsvector(\'english\'::regconfig, (content)::text)'
|
18
|
+
@connection.add_index :posts, :title
|
19
|
+
end
|
20
|
+
|
21
|
+
teardown do
|
22
|
+
@connection.drop_table "posts"
|
23
|
+
end
|
24
|
+
|
25
|
+
def test_schema_dump_expression
|
26
|
+
schema = dump_table_schema "posts"
|
27
|
+
assert_match %r{add_index\s+\"posts\",\s+\[\"title\"\],\s+name:\s+\"index_posts_on_title\",\s+using:\s+:btree$}, schema
|
28
|
+
assert_match %r{add_index\s+\"posts\",\s+\[\"content\"\],\s+name:\s+\"index_posts_on_content\",\s+using:\s+:gin,\s+expression:\s+\"to_tsvector\('english'::regconfig,\s+content\)\"$}, schema
|
29
|
+
assert_match %r{add_index\s+\"posts\",\s+'',\s+name:\s+\"hogehoge\",\s+using:\s+:gin,\s+expression:\s+\"to_tsvector\('english'::regconfig,\s+content\)\"$}, schema
|
30
|
+
end
|
31
|
+
end
|
@@ -0,0 +1,51 @@
|
|
1
|
+
require 'bundler'
|
2
|
+
Bundler.setup
|
3
|
+
|
4
|
+
require 'activerecord-postgresql-expression'
|
5
|
+
require 'config'
|
6
|
+
|
7
|
+
require 'active_support/testing/autorun'
|
8
|
+
require 'stringio'
|
9
|
+
|
10
|
+
require 'active_record'
|
11
|
+
require 'cases/test_case'
|
12
|
+
require 'active_support/dependencies'
|
13
|
+
require 'active_support/logger'
|
14
|
+
require 'active_support/core_ext/string/strip'
|
15
|
+
|
16
|
+
require 'support/config'
|
17
|
+
require 'support/connection'
|
18
|
+
|
19
|
+
# TODO: Move all these random hacks into the ARTest namespace and into the support/ dir
|
20
|
+
|
21
|
+
Thread.abort_on_exception = true
|
22
|
+
|
23
|
+
# Show backtraces for deprecated behavior for quicker cleanup.
|
24
|
+
ActiveSupport::Deprecation.debug = true
|
25
|
+
|
26
|
+
# Enable raise errors in after_commit and after_rollback.
|
27
|
+
ActiveRecord::Base.tap do |klass|
|
28
|
+
klass.raise_in_transactional_callbacks = true if klass.respond_to?(:raise_in_transactional_callbacks=)
|
29
|
+
end
|
30
|
+
|
31
|
+
# Connect to the database
|
32
|
+
ARTest.connect
|
33
|
+
|
34
|
+
def current_adapter?(*types)
|
35
|
+
types.any? do |type|
|
36
|
+
ActiveRecord::ConnectionAdapters.const_defined?(type) &&
|
37
|
+
ActiveRecord::Base.connection.is_a?(ActiveRecord::ConnectionAdapters.const_get(type))
|
38
|
+
end
|
39
|
+
end
|
40
|
+
|
41
|
+
def mysql_56?
|
42
|
+
current_adapter?(:Mysql2Adapter) &&
|
43
|
+
ActiveRecord::Base.connection.send(:version).join(".") >= "5.6.0"
|
44
|
+
end
|
45
|
+
|
46
|
+
# FIXME: we have tests that depend on run order, we should fix that and
|
47
|
+
# remove this method call.
|
48
|
+
require 'active_support/test_case'
|
49
|
+
ActiveSupport::TestCase.tap do |klass|
|
50
|
+
klass.test_order = :sorted if klass.respond_to?(:test_order=)
|
51
|
+
end
|
@@ -0,0 +1,123 @@
|
|
1
|
+
require 'active_support/test_case'
|
2
|
+
|
3
|
+
module ActiveRecord
|
4
|
+
# = Active Record Test Case
|
5
|
+
#
|
6
|
+
# Defines some test assertions to test against SQL queries.
|
7
|
+
class TestCase < ActiveSupport::TestCase #:nodoc:
|
8
|
+
def teardown
|
9
|
+
SQLCounter.clear_log
|
10
|
+
end
|
11
|
+
|
12
|
+
def assert_date_from_db(expected, actual, message = nil)
|
13
|
+
assert_equal expected.to_s, actual.to_s, message
|
14
|
+
end
|
15
|
+
|
16
|
+
def capture(stream)
|
17
|
+
stream = stream.to_s
|
18
|
+
captured_stream = Tempfile.new(stream)
|
19
|
+
stream_io = eval("$#{stream}")
|
20
|
+
origin_stream = stream_io.dup
|
21
|
+
stream_io.reopen(captured_stream)
|
22
|
+
|
23
|
+
yield
|
24
|
+
|
25
|
+
stream_io.rewind
|
26
|
+
return captured_stream.read
|
27
|
+
ensure
|
28
|
+
captured_stream.close
|
29
|
+
captured_stream.unlink
|
30
|
+
stream_io.reopen(origin_stream)
|
31
|
+
end
|
32
|
+
|
33
|
+
def capture_sql
|
34
|
+
SQLCounter.clear_log
|
35
|
+
yield
|
36
|
+
SQLCounter.log_all.dup
|
37
|
+
end
|
38
|
+
|
39
|
+
def assert_sql(*patterns_to_match)
|
40
|
+
capture_sql { yield }
|
41
|
+
ensure
|
42
|
+
failed_patterns = []
|
43
|
+
patterns_to_match.each do |pattern|
|
44
|
+
failed_patterns << pattern unless SQLCounter.log_all.any?{ |sql| pattern === sql }
|
45
|
+
end
|
46
|
+
assert failed_patterns.empty?, "Query pattern(s) #{failed_patterns.map(&:inspect).join(', ')} not found.#{SQLCounter.log.size == 0 ? '' : "\nQueries:\n#{SQLCounter.log.join("\n")}"}"
|
47
|
+
end
|
48
|
+
|
49
|
+
def assert_queries(num = 1, options = {})
|
50
|
+
ignore_none = options.fetch(:ignore_none) { num == :any }
|
51
|
+
SQLCounter.clear_log
|
52
|
+
x = yield
|
53
|
+
the_log = ignore_none ? SQLCounter.log_all : SQLCounter.log
|
54
|
+
if num == :any
|
55
|
+
assert_operator the_log.size, :>=, 1, "1 or more queries expected, but none were executed."
|
56
|
+
else
|
57
|
+
mesg = "#{the_log.size} instead of #{num} queries were executed.#{the_log.size == 0 ? '' : "\nQueries:\n#{the_log.join("\n")}"}"
|
58
|
+
assert_equal num, the_log.size, mesg
|
59
|
+
end
|
60
|
+
x
|
61
|
+
end
|
62
|
+
|
63
|
+
def assert_no_queries(options = {}, &block)
|
64
|
+
options.reverse_merge! ignore_none: true
|
65
|
+
assert_queries(0, options, &block)
|
66
|
+
end
|
67
|
+
|
68
|
+
def assert_column(model, column_name, msg=nil)
|
69
|
+
assert has_column?(model, column_name), msg
|
70
|
+
end
|
71
|
+
|
72
|
+
def assert_no_column(model, column_name, msg=nil)
|
73
|
+
assert_not has_column?(model, column_name), msg
|
74
|
+
end
|
75
|
+
|
76
|
+
def has_column?(model, column_name)
|
77
|
+
model.reset_column_information
|
78
|
+
model.column_names.include?(column_name.to_s)
|
79
|
+
end
|
80
|
+
end
|
81
|
+
|
82
|
+
class SQLCounter
|
83
|
+
class << self
|
84
|
+
attr_accessor :ignored_sql, :log, :log_all
|
85
|
+
def clear_log; self.log = []; self.log_all = []; end
|
86
|
+
end
|
87
|
+
|
88
|
+
self.clear_log
|
89
|
+
|
90
|
+
self.ignored_sql = [/^PRAGMA/, /^SELECT currval/, /^SELECT CAST/, /^SELECT @@IDENTITY/, /^SELECT @@ROWCOUNT/, /^SAVEPOINT/, /^ROLLBACK TO SAVEPOINT/, /^RELEASE SAVEPOINT/, /^SHOW max_identifier_length/, /^BEGIN/, /^COMMIT/]
|
91
|
+
|
92
|
+
# FIXME: this needs to be refactored so specific database can add their own
|
93
|
+
# ignored SQL, or better yet, use a different notification for the queries
|
94
|
+
# instead examining the SQL content.
|
95
|
+
oracle_ignored = [/^select .*nextval/i, /^SAVEPOINT/, /^ROLLBACK TO/, /^\s*select .* from all_triggers/im, /^\s*select .* from all_constraints/im, /^\s*select .* from all_tab_cols/im]
|
96
|
+
mysql_ignored = [/^SHOW TABLES/i, /^SHOW FULL FIELDS/, /^SHOW CREATE TABLE /i, /^SHOW VARIABLES /]
|
97
|
+
postgresql_ignored = [/^\s*select\b.*\bfrom\b.*pg_namespace\b/im, /^\s*select tablename\b.*from pg_tables\b/im, /^\s*select\b.*\battname\b.*\bfrom\b.*\bpg_attribute\b/im, /^SHOW search_path/i]
|
98
|
+
sqlite3_ignored = [/^\s*SELECT name\b.*\bFROM sqlite_master/im]
|
99
|
+
|
100
|
+
[oracle_ignored, mysql_ignored, postgresql_ignored, sqlite3_ignored].each do |db_ignored_sql|
|
101
|
+
ignored_sql.concat db_ignored_sql
|
102
|
+
end
|
103
|
+
|
104
|
+
attr_reader :ignore
|
105
|
+
|
106
|
+
def initialize(ignore = Regexp.union(self.class.ignored_sql))
|
107
|
+
@ignore = ignore
|
108
|
+
end
|
109
|
+
|
110
|
+
def call(name, start, finish, message_id, values)
|
111
|
+
sql = values[:sql]
|
112
|
+
|
113
|
+
# FIXME: this seems bad. we should probably have a better way to indicate
|
114
|
+
# the query was cached
|
115
|
+
return if 'CACHE' == values[:name]
|
116
|
+
|
117
|
+
self.class.log_all << sql
|
118
|
+
self.class.log << sql unless ignore =~ sql
|
119
|
+
end
|
120
|
+
end
|
121
|
+
|
122
|
+
ActiveSupport::Notifications.subscribe('sql.active_record', SQLCounter.new)
|
123
|
+
end
|
data/test/config.rb
ADDED
@@ -0,0 +1,43 @@
|
|
1
|
+
require 'yaml'
|
2
|
+
require 'erb'
|
3
|
+
require 'fileutils'
|
4
|
+
require 'pathname'
|
5
|
+
|
6
|
+
module ARTest
|
7
|
+
class << self
|
8
|
+
def config
|
9
|
+
@config ||= read_config
|
10
|
+
end
|
11
|
+
|
12
|
+
private
|
13
|
+
|
14
|
+
def config_file
|
15
|
+
Pathname.new(ENV['ARCONFIG'] || TEST_ROOT + '/config.yml')
|
16
|
+
end
|
17
|
+
|
18
|
+
def read_config
|
19
|
+
unless config_file.exist?
|
20
|
+
FileUtils.cp TEST_ROOT + '/config.example.yml', config_file
|
21
|
+
end
|
22
|
+
|
23
|
+
erb = ERB.new(config_file.read)
|
24
|
+
expand_config(YAML.parse(erb.result(binding)).transform)
|
25
|
+
end
|
26
|
+
|
27
|
+
def expand_config(config)
|
28
|
+
config['connections'].each do |adapter, connection|
|
29
|
+
dbs = [['arunit', 'activerecord_unittest'], ['arunit2', 'activerecord_unittest2']]
|
30
|
+
dbs.each do |name, dbname|
|
31
|
+
unless connection[name].is_a?(Hash)
|
32
|
+
connection[name] = { 'database' => connection[name] }
|
33
|
+
end
|
34
|
+
|
35
|
+
connection[name]['database'] ||= dbname
|
36
|
+
connection[name]['adapter'] ||= adapter
|
37
|
+
end
|
38
|
+
end
|
39
|
+
|
40
|
+
config
|
41
|
+
end
|
42
|
+
end
|
43
|
+
end
|
@@ -0,0 +1,18 @@
|
|
1
|
+
require 'active_support/logger'
|
2
|
+
|
3
|
+
module ARTest
|
4
|
+
def self.connection_name
|
5
|
+
ENV['ARCONN'] || config['default_connection']
|
6
|
+
end
|
7
|
+
|
8
|
+
def self.connection_config
|
9
|
+
config['connections'][connection_name]
|
10
|
+
end
|
11
|
+
|
12
|
+
def self.connect
|
13
|
+
puts "Using #{connection_name}"
|
14
|
+
ActiveRecord::Base.logger = ActiveSupport::Logger.new("debug.log", 0, 100 * 1024 * 1024)
|
15
|
+
ActiveRecord::Base.configurations = connection_config
|
16
|
+
ActiveRecord::Base.establish_connection :arunit
|
17
|
+
end
|
18
|
+
end
|
@@ -0,0 +1,20 @@
|
|
1
|
+
module SchemaDumpingHelper
|
2
|
+
def dump_table_schema(table, connection = ActiveRecord::Base.connection)
|
3
|
+
old_ignore_tables = ActiveRecord::SchemaDumper.ignore_tables
|
4
|
+
ActiveRecord::SchemaDumper.ignore_tables = connection.tables - [table]
|
5
|
+
stream = StringIO.new
|
6
|
+
ActiveRecord::SchemaDumper.dump(ActiveRecord::Base.connection, stream)
|
7
|
+
stream.string
|
8
|
+
ensure
|
9
|
+
ActiveRecord::SchemaDumper.ignore_tables = old_ignore_tables
|
10
|
+
end
|
11
|
+
|
12
|
+
def dump_all_table_schema(ignore_tables)
|
13
|
+
old_ignore_tables, ActiveRecord::SchemaDumper.ignore_tables = ActiveRecord::SchemaDumper.ignore_tables, ignore_tables
|
14
|
+
stream = StringIO.new
|
15
|
+
ActiveRecord::SchemaDumper.dump(ActiveRecord::Base.connection, stream)
|
16
|
+
stream.string
|
17
|
+
ensure
|
18
|
+
ActiveRecord::SchemaDumper.ignore_tables = old_ignore_tables
|
19
|
+
end
|
20
|
+
end
|
metadata
ADDED
@@ -0,0 +1,148 @@
|
|
1
|
+
--- !ruby/object:Gem::Specification
|
2
|
+
name: activerecord-postgresql-expression
|
3
|
+
version: !ruby/object:Gem::Version
|
4
|
+
version: 0.0.1
|
5
|
+
platform: ruby
|
6
|
+
authors:
|
7
|
+
- Naoya Murakami
|
8
|
+
autorequire:
|
9
|
+
bindir: bin
|
10
|
+
cert_chain: []
|
11
|
+
date: 2015-02-23 00:00:00.000000000 Z
|
12
|
+
dependencies:
|
13
|
+
- !ruby/object:Gem::Dependency
|
14
|
+
name: bundler
|
15
|
+
requirement: !ruby/object:Gem::Requirement
|
16
|
+
requirements:
|
17
|
+
- - "~>"
|
18
|
+
- !ruby/object:Gem::Version
|
19
|
+
version: '1.7'
|
20
|
+
type: :development
|
21
|
+
prerelease: false
|
22
|
+
version_requirements: !ruby/object:Gem::Requirement
|
23
|
+
requirements:
|
24
|
+
- - "~>"
|
25
|
+
- !ruby/object:Gem::Version
|
26
|
+
version: '1.7'
|
27
|
+
- !ruby/object:Gem::Dependency
|
28
|
+
name: rake
|
29
|
+
requirement: !ruby/object:Gem::Requirement
|
30
|
+
requirements:
|
31
|
+
- - "~>"
|
32
|
+
- !ruby/object:Gem::Version
|
33
|
+
version: '10.0'
|
34
|
+
type: :development
|
35
|
+
prerelease: false
|
36
|
+
version_requirements: !ruby/object:Gem::Requirement
|
37
|
+
requirements:
|
38
|
+
- - "~>"
|
39
|
+
- !ruby/object:Gem::Version
|
40
|
+
version: '10.0'
|
41
|
+
- !ruby/object:Gem::Dependency
|
42
|
+
name: activesupport
|
43
|
+
requirement: !ruby/object:Gem::Requirement
|
44
|
+
requirements:
|
45
|
+
- - "~>"
|
46
|
+
- !ruby/object:Gem::Version
|
47
|
+
version: '4.0'
|
48
|
+
type: :runtime
|
49
|
+
prerelease: false
|
50
|
+
version_requirements: !ruby/object:Gem::Requirement
|
51
|
+
requirements:
|
52
|
+
- - "~>"
|
53
|
+
- !ruby/object:Gem::Version
|
54
|
+
version: '4.0'
|
55
|
+
- !ruby/object:Gem::Dependency
|
56
|
+
name: activerecord
|
57
|
+
requirement: !ruby/object:Gem::Requirement
|
58
|
+
requirements:
|
59
|
+
- - "~>"
|
60
|
+
- !ruby/object:Gem::Version
|
61
|
+
version: '4.0'
|
62
|
+
type: :runtime
|
63
|
+
prerelease: false
|
64
|
+
version_requirements: !ruby/object:Gem::Requirement
|
65
|
+
requirements:
|
66
|
+
- - "~>"
|
67
|
+
- !ruby/object:Gem::Version
|
68
|
+
version: '4.0'
|
69
|
+
- !ruby/object:Gem::Dependency
|
70
|
+
name: pg
|
71
|
+
requirement: !ruby/object:Gem::Requirement
|
72
|
+
requirements:
|
73
|
+
- - ">="
|
74
|
+
- !ruby/object:Gem::Version
|
75
|
+
version: '0'
|
76
|
+
type: :runtime
|
77
|
+
prerelease: false
|
78
|
+
version_requirements: !ruby/object:Gem::Requirement
|
79
|
+
requirements:
|
80
|
+
- - ">="
|
81
|
+
- !ruby/object:Gem::Version
|
82
|
+
version: '0'
|
83
|
+
description: Adds expression to migrations for ActiveRecord PostgreSQL adapters
|
84
|
+
email:
|
85
|
+
- naoya@createfield.com
|
86
|
+
executables: []
|
87
|
+
extensions: []
|
88
|
+
extra_rdoc_files: []
|
89
|
+
files:
|
90
|
+
- ".gitignore"
|
91
|
+
- ".travis.yml"
|
92
|
+
- Gemfile
|
93
|
+
- LICENSE.txt
|
94
|
+
- README.md
|
95
|
+
- Rakefile
|
96
|
+
- activerecord-postgresql-expression.gemspec
|
97
|
+
- lib/activerecord-postgresql-expression.rb
|
98
|
+
- lib/activerecord-postgresql-expression/active_record/connection_adapters/postgresql/schema_statements.rb
|
99
|
+
- lib/activerecord-postgresql-expression/active_record/schema_dumper.rb
|
100
|
+
- lib/activerecord/postgresql/expression.rb
|
101
|
+
- lib/activerecord/postgresql/expression/base.rb
|
102
|
+
- lib/activerecord/postgresql/expression/railtie.rb
|
103
|
+
- lib/activerecord/postgresql/expression/version.rb
|
104
|
+
- test/.gitignore
|
105
|
+
- test/cases/expression_test.rb
|
106
|
+
- test/cases/helper.rb
|
107
|
+
- test/cases/test_case.rb
|
108
|
+
- test/config.example.yml
|
109
|
+
- test/config.rb
|
110
|
+
- test/support/config.rb
|
111
|
+
- test/support/connection.rb
|
112
|
+
- test/support/connection_helper.rb
|
113
|
+
- test/support/schema_dumping_helper.rb
|
114
|
+
homepage: https://github.com/naoa/activerecord-postgresql-expression
|
115
|
+
licenses:
|
116
|
+
- MIT
|
117
|
+
metadata: {}
|
118
|
+
post_install_message:
|
119
|
+
rdoc_options: []
|
120
|
+
require_paths:
|
121
|
+
- lib
|
122
|
+
required_ruby_version: !ruby/object:Gem::Requirement
|
123
|
+
requirements:
|
124
|
+
- - ">="
|
125
|
+
- !ruby/object:Gem::Version
|
126
|
+
version: 2.0.0
|
127
|
+
required_rubygems_version: !ruby/object:Gem::Requirement
|
128
|
+
requirements:
|
129
|
+
- - ">="
|
130
|
+
- !ruby/object:Gem::Version
|
131
|
+
version: '0'
|
132
|
+
requirements: []
|
133
|
+
rubyforge_project:
|
134
|
+
rubygems_version: 2.2.2
|
135
|
+
signing_key:
|
136
|
+
specification_version: 4
|
137
|
+
summary: Adds expression to migrations for ActiveRecord PostgreSQL adapters
|
138
|
+
test_files:
|
139
|
+
- test/.gitignore
|
140
|
+
- test/cases/expression_test.rb
|
141
|
+
- test/cases/helper.rb
|
142
|
+
- test/cases/test_case.rb
|
143
|
+
- test/config.example.yml
|
144
|
+
- test/config.rb
|
145
|
+
- test/support/config.rb
|
146
|
+
- test/support/connection.rb
|
147
|
+
- test/support/connection_helper.rb
|
148
|
+
- test/support/schema_dumping_helper.rb
|