database_cleaner-spanner 0.1.0 → 0.1.1

Sign up to get free protection for your applications and to get access to all the features.
Files changed (103) hide show
  1. checksums.yaml +4 -4
  2. data/.standard.yml +3 -0
  3. data/CHANGELOG.md +4 -2
  4. data/LICENSE +21 -0
  5. data/README.md +6 -0
  6. data/database_cleaner-spanner.gemspec +3 -0
  7. data/example/rails/.gitattributes +7 -0
  8. data/example/rails/.gitignore +39 -0
  9. data/example/rails/.rspec +1 -0
  10. data/example/rails/.ruby-version +1 -0
  11. data/example/rails/Gemfile +69 -0
  12. data/example/rails/README.md +24 -0
  13. data/example/rails/Rakefile +6 -0
  14. data/example/rails/app/assets/config/manifest.js +4 -0
  15. data/example/rails/app/assets/images/.keep +0 -0
  16. data/example/rails/app/assets/stylesheets/application.css +15 -0
  17. data/example/rails/app/channels/application_cable/channel.rb +4 -0
  18. data/example/rails/app/channels/application_cable/connection.rb +4 -0
  19. data/example/rails/app/controllers/application_controller.rb +2 -0
  20. data/example/rails/app/controllers/concerns/.keep +0 -0
  21. data/example/rails/app/helpers/application_helper.rb +2 -0
  22. data/example/rails/app/javascript/application.js +3 -0
  23. data/example/rails/app/javascript/controllers/application.js +9 -0
  24. data/example/rails/app/javascript/controllers/hello_controller.js +7 -0
  25. data/example/rails/app/javascript/controllers/index.js +11 -0
  26. data/example/rails/app/jobs/application_job.rb +7 -0
  27. data/example/rails/app/mailers/application_mailer.rb +4 -0
  28. data/example/rails/app/models/album.rb +5 -0
  29. data/example/rails/app/models/application_record.rb +5 -0
  30. data/example/rails/app/models/concerns/.keep +0 -0
  31. data/example/rails/app/models/customer.rb +3 -0
  32. data/example/rails/app/models/order.rb +4 -0
  33. data/example/rails/app/models/product.rb +3 -0
  34. data/example/rails/app/models/singer.rb +4 -0
  35. data/example/rails/app/models/song.rb +15 -0
  36. data/example/rails/app/views/layouts/application.html.erb +16 -0
  37. data/example/rails/app/views/layouts/mailer.html.erb +13 -0
  38. data/example/rails/app/views/layouts/mailer.text.erb +1 -0
  39. data/example/rails/bin/bundle +114 -0
  40. data/example/rails/bin/importmap +4 -0
  41. data/example/rails/bin/rails +4 -0
  42. data/example/rails/bin/rake +4 -0
  43. data/example/rails/bin/setup +33 -0
  44. data/example/rails/config/application.rb +37 -0
  45. data/example/rails/config/boot.rb +4 -0
  46. data/example/rails/config/cable.yml +10 -0
  47. data/example/rails/config/credentials.yml.enc +1 -0
  48. data/example/rails/config/database.yml +18 -0
  49. data/example/rails/config/environment.rb +5 -0
  50. data/example/rails/config/environments/development.rb +70 -0
  51. data/example/rails/config/environments/production.rb +93 -0
  52. data/example/rails/config/environments/test.rb +60 -0
  53. data/example/rails/config/importmap.rb +7 -0
  54. data/example/rails/config/initializers/assets.rb +12 -0
  55. data/example/rails/config/initializers/content_security_policy.rb +25 -0
  56. data/example/rails/config/initializers/filter_parameter_logging.rb +8 -0
  57. data/example/rails/config/initializers/inflections.rb +16 -0
  58. data/example/rails/config/initializers/permissions_policy.rb +11 -0
  59. data/example/rails/config/locales/en.yml +33 -0
  60. data/example/rails/config/puma.rb +43 -0
  61. data/example/rails/config/routes.rb +6 -0
  62. data/example/rails/config/storage.yml +34 -0
  63. data/example/rails/config.ru +6 -0
  64. data/example/rails/db/migrate/20221113100815_create_customers.rb +9 -0
  65. data/example/rails/db/migrate/20221113101016_create_products.rb +10 -0
  66. data/example/rails/db/migrate/20221113101027_create_orders.rb +11 -0
  67. data/example/rails/db/migrate/20221113101034_create_singers.rb +10 -0
  68. data/example/rails/db/migrate/20221113101039_create_albums.rb +12 -0
  69. data/example/rails/db/migrate/20221113101044_create_songs.rb +13 -0
  70. data/example/rails/db/schema.rb +58 -0
  71. data/example/rails/db/seeds.rb +7 -0
  72. data/example/rails/lib/assets/.keep +0 -0
  73. data/example/rails/lib/tasks/.keep +0 -0
  74. data/example/rails/log/.keep +0 -0
  75. data/example/rails/public/404.html +67 -0
  76. data/example/rails/public/422.html +67 -0
  77. data/example/rails/public/500.html +66 -0
  78. data/example/rails/public/apple-touch-icon-precomposed.png +0 -0
  79. data/example/rails/public/apple-touch-icon.png +0 -0
  80. data/example/rails/public/favicon.ico +0 -0
  81. data/example/rails/public/robots.txt +1 -0
  82. data/example/rails/spec/factories/factory.rb +33 -0
  83. data/example/rails/spec/models/album_spec.rb +5 -0
  84. data/example/rails/spec/models/customer_spec.rb +5 -0
  85. data/example/rails/spec/models/order_spec.rb +5 -0
  86. data/example/rails/spec/models/product_spec.rb +5 -0
  87. data/example/rails/spec/models/singer_spec.rb +5 -0
  88. data/example/rails/spec/models/song_spec.rb +5 -0
  89. data/example/rails/spec/rails_helper.rb +70 -0
  90. data/example/rails/spec/sample_spec.rb +14 -0
  91. data/example/rails/spec/spec_helper.rb +94 -0
  92. data/example/rails/spec/support/factory_bot.rb +3 -0
  93. data/example/rails/storage/.keep +0 -0
  94. data/example/rails/tmp/.keep +0 -0
  95. data/example/rails/tmp/pids/.keep +0 -0
  96. data/example/rails/tmp/storage/.keep +0 -0
  97. data/example/rails/vendor/.keep +0 -0
  98. data/example/rails/vendor/javascript/.keep +0 -0
  99. data/lib/database_cleaner/spanner/deletion.rb +31 -5
  100. data/lib/database_cleaner/spanner/version.rb +1 -1
  101. data/scripts/performance_test.rb +127 -0
  102. data/scripts/schema.sql +93 -0
  103. metadata +125 -2
@@ -0,0 +1,67 @@
1
+ <!DOCTYPE html>
2
+ <html>
3
+ <head>
4
+ <title>The page you were looking for doesn't exist (404)</title>
5
+ <meta name="viewport" content="width=device-width,initial-scale=1">
6
+ <style>
7
+ .rails-default-error-page {
8
+ background-color: #EFEFEF;
9
+ color: #2E2F30;
10
+ text-align: center;
11
+ font-family: arial, sans-serif;
12
+ margin: 0;
13
+ }
14
+
15
+ .rails-default-error-page div.dialog {
16
+ width: 95%;
17
+ max-width: 33em;
18
+ margin: 4em auto 0;
19
+ }
20
+
21
+ .rails-default-error-page div.dialog > div {
22
+ border: 1px solid #CCC;
23
+ border-right-color: #999;
24
+ border-left-color: #999;
25
+ border-bottom-color: #BBB;
26
+ border-top: #B00100 solid 4px;
27
+ border-top-left-radius: 9px;
28
+ border-top-right-radius: 9px;
29
+ background-color: white;
30
+ padding: 7px 12% 0;
31
+ box-shadow: 0 3px 8px rgba(50, 50, 50, 0.17);
32
+ }
33
+
34
+ .rails-default-error-page h1 {
35
+ font-size: 100%;
36
+ color: #730E15;
37
+ line-height: 1.5em;
38
+ }
39
+
40
+ .rails-default-error-page div.dialog > p {
41
+ margin: 0 0 1em;
42
+ padding: 1em;
43
+ background-color: #F7F7F7;
44
+ border: 1px solid #CCC;
45
+ border-right-color: #999;
46
+ border-left-color: #999;
47
+ border-bottom-color: #999;
48
+ border-bottom-left-radius: 4px;
49
+ border-bottom-right-radius: 4px;
50
+ border-top-color: #DADADA;
51
+ color: #666;
52
+ box-shadow: 0 3px 8px rgba(50, 50, 50, 0.17);
53
+ }
54
+ </style>
55
+ </head>
56
+
57
+ <body class="rails-default-error-page">
58
+ <!-- This file lives in public/404.html -->
59
+ <div class="dialog">
60
+ <div>
61
+ <h1>The page you were looking for doesn't exist.</h1>
62
+ <p>You may have mistyped the address or the page may have moved.</p>
63
+ </div>
64
+ <p>If you are the application owner check the logs for more information.</p>
65
+ </div>
66
+ </body>
67
+ </html>
@@ -0,0 +1,67 @@
1
+ <!DOCTYPE html>
2
+ <html>
3
+ <head>
4
+ <title>The change you wanted was rejected (422)</title>
5
+ <meta name="viewport" content="width=device-width,initial-scale=1">
6
+ <style>
7
+ .rails-default-error-page {
8
+ background-color: #EFEFEF;
9
+ color: #2E2F30;
10
+ text-align: center;
11
+ font-family: arial, sans-serif;
12
+ margin: 0;
13
+ }
14
+
15
+ .rails-default-error-page div.dialog {
16
+ width: 95%;
17
+ max-width: 33em;
18
+ margin: 4em auto 0;
19
+ }
20
+
21
+ .rails-default-error-page div.dialog > div {
22
+ border: 1px solid #CCC;
23
+ border-right-color: #999;
24
+ border-left-color: #999;
25
+ border-bottom-color: #BBB;
26
+ border-top: #B00100 solid 4px;
27
+ border-top-left-radius: 9px;
28
+ border-top-right-radius: 9px;
29
+ background-color: white;
30
+ padding: 7px 12% 0;
31
+ box-shadow: 0 3px 8px rgba(50, 50, 50, 0.17);
32
+ }
33
+
34
+ .rails-default-error-page h1 {
35
+ font-size: 100%;
36
+ color: #730E15;
37
+ line-height: 1.5em;
38
+ }
39
+
40
+ .rails-default-error-page div.dialog > p {
41
+ margin: 0 0 1em;
42
+ padding: 1em;
43
+ background-color: #F7F7F7;
44
+ border: 1px solid #CCC;
45
+ border-right-color: #999;
46
+ border-left-color: #999;
47
+ border-bottom-color: #999;
48
+ border-bottom-left-radius: 4px;
49
+ border-bottom-right-radius: 4px;
50
+ border-top-color: #DADADA;
51
+ color: #666;
52
+ box-shadow: 0 3px 8px rgba(50, 50, 50, 0.17);
53
+ }
54
+ </style>
55
+ </head>
56
+
57
+ <body class="rails-default-error-page">
58
+ <!-- This file lives in public/422.html -->
59
+ <div class="dialog">
60
+ <div>
61
+ <h1>The change you wanted was rejected.</h1>
62
+ <p>Maybe you tried to change something you didn't have access to.</p>
63
+ </div>
64
+ <p>If you are the application owner check the logs for more information.</p>
65
+ </div>
66
+ </body>
67
+ </html>
@@ -0,0 +1,66 @@
1
+ <!DOCTYPE html>
2
+ <html>
3
+ <head>
4
+ <title>We're sorry, but something went wrong (500)</title>
5
+ <meta name="viewport" content="width=device-width,initial-scale=1">
6
+ <style>
7
+ .rails-default-error-page {
8
+ background-color: #EFEFEF;
9
+ color: #2E2F30;
10
+ text-align: center;
11
+ font-family: arial, sans-serif;
12
+ margin: 0;
13
+ }
14
+
15
+ .rails-default-error-page div.dialog {
16
+ width: 95%;
17
+ max-width: 33em;
18
+ margin: 4em auto 0;
19
+ }
20
+
21
+ .rails-default-error-page div.dialog > div {
22
+ border: 1px solid #CCC;
23
+ border-right-color: #999;
24
+ border-left-color: #999;
25
+ border-bottom-color: #BBB;
26
+ border-top: #B00100 solid 4px;
27
+ border-top-left-radius: 9px;
28
+ border-top-right-radius: 9px;
29
+ background-color: white;
30
+ padding: 7px 12% 0;
31
+ box-shadow: 0 3px 8px rgba(50, 50, 50, 0.17);
32
+ }
33
+
34
+ .rails-default-error-page h1 {
35
+ font-size: 100%;
36
+ color: #730E15;
37
+ line-height: 1.5em;
38
+ }
39
+
40
+ .rails-default-error-page div.dialog > p {
41
+ margin: 0 0 1em;
42
+ padding: 1em;
43
+ background-color: #F7F7F7;
44
+ border: 1px solid #CCC;
45
+ border-right-color: #999;
46
+ border-left-color: #999;
47
+ border-bottom-color: #999;
48
+ border-bottom-left-radius: 4px;
49
+ border-bottom-right-radius: 4px;
50
+ border-top-color: #DADADA;
51
+ color: #666;
52
+ box-shadow: 0 3px 8px rgba(50, 50, 50, 0.17);
53
+ }
54
+ </style>
55
+ </head>
56
+
57
+ <body class="rails-default-error-page">
58
+ <!-- This file lives in public/500.html -->
59
+ <div class="dialog">
60
+ <div>
61
+ <h1>We're sorry, but something went wrong.</h1>
62
+ </div>
63
+ <p>If you are the application owner check the logs for more information.</p>
64
+ </div>
65
+ </body>
66
+ </html>
File without changes
File without changes
@@ -0,0 +1 @@
1
+ # See https://www.robotstxt.org/robotstxt.html for documentation on how to use the robots.txt file
@@ -0,0 +1,33 @@
1
+ FactoryBot.define do
2
+ factory :customer do
3
+ name { "customer" }
4
+ end
5
+
6
+ factory :product do
7
+ name { "product" }
8
+ price { 1.0 }
9
+ end
10
+
11
+ factory :order do
12
+ customer
13
+ product
14
+ quantity { 1 }
15
+ end
16
+
17
+ factory :singer do
18
+ name { "singer" }
19
+ end
20
+
21
+ factory :album do
22
+ singer
23
+ albumid { 1 }
24
+ title { "album" }
25
+ end
26
+
27
+ factory :song do
28
+ singer
29
+ album
30
+ songid { 1 }
31
+ title { "song" }
32
+ end
33
+ end
@@ -0,0 +1,5 @@
1
+ require 'rails_helper'
2
+
3
+ RSpec.describe Album, type: :model do
4
+ pending "add some examples to (or delete) #{__FILE__}"
5
+ end
@@ -0,0 +1,5 @@
1
+ require 'rails_helper'
2
+
3
+ RSpec.describe Customer, type: :model do
4
+ pending "add some examples to (or delete) #{__FILE__}"
5
+ end
@@ -0,0 +1,5 @@
1
+ require 'rails_helper'
2
+
3
+ RSpec.describe Order, type: :model do
4
+ pending "add some examples to (or delete) #{__FILE__}"
5
+ end
@@ -0,0 +1,5 @@
1
+ require 'rails_helper'
2
+
3
+ RSpec.describe Product, type: :model do
4
+ pending "add some examples to (or delete) #{__FILE__}"
5
+ end
@@ -0,0 +1,5 @@
1
+ require 'rails_helper'
2
+
3
+ RSpec.describe Singer, type: :model do
4
+ pending "add some examples to (or delete) #{__FILE__}"
5
+ end
@@ -0,0 +1,5 @@
1
+ require 'rails_helper'
2
+
3
+ RSpec.describe Song, type: :model do
4
+ pending "add some examples to (or delete) #{__FILE__}"
5
+ end
@@ -0,0 +1,70 @@
1
+ # This file is copied to spec/ when you run 'rails generate rspec:install'
2
+ require 'spec_helper'
3
+ ENV['RAILS_ENV'] ||= 'test'
4
+ require_relative '../config/environment'
5
+ # Prevent database truncation if the environment is production
6
+ abort("The Rails environment is running in production mode!") if Rails.env.production?
7
+ require 'rspec/rails'
8
+ # Add additional requires below this line. Rails is not loaded until this point!
9
+
10
+ # Requires supporting ruby files with custom matchers and macros, etc, in
11
+ # spec/support/ and its subdirectories. Files matching `spec/**/*_spec.rb` are
12
+ # run as spec files by default. This means that files in spec/support that end
13
+ # in _spec.rb will both be required and run as specs, causing the specs to be
14
+ # run twice. It is recommended that you do not name files matching this glob to
15
+ # end with _spec.rb. You can configure this pattern with the --pattern
16
+ # option on the command line or in ~/.rspec, .rspec or `.rspec-local`.
17
+ #
18
+ # The following line is provided for convenience purposes. It has the downside
19
+ # of increasing the boot-up time by auto-requiring all files in the support
20
+ # directory. Alternatively, in the individual `*_spec.rb` files, manually
21
+ # require only the support files necessary.
22
+ #
23
+
24
+ Dir[Rails.root.join('spec', 'support', '**', '*.rb')].sort.each { |f| require f }
25
+
26
+ # Checks for pending migrations and applies them before tests are run.
27
+ # If you are not using ActiveRecord, you can remove these lines.
28
+ # begin
29
+ # ActiveRecord::Migration.maintain_test_schema!
30
+ # rescue ActiveRecord::PendingMigrationError => e
31
+ # abort e.to_s.strip
32
+ # end
33
+ RSpec.configure do |config|
34
+ # Remove this line if you're not using ActiveRecord or ActiveRecord fixtures
35
+ config.fixture_path = "#{::Rails.root}/spec/fixtures"
36
+
37
+ # If you're not using ActiveRecord, or you'd prefer not to run each of your
38
+ # examples within a transaction, remove the following line or assign false
39
+ # instead of true.
40
+ config.use_transactional_fixtures = false
41
+
42
+ # You can uncomment this line to turn off ActiveRecord support entirely.
43
+ # config.use_active_record = false
44
+
45
+ # RSpec Rails can automatically mix in different behaviours to your tests
46
+ # based on their file location, for example enabling you to call `get` and
47
+ # `post` in specs under `spec/controllers`.
48
+ #
49
+ # You can disable this behaviour by removing the line below, and instead
50
+ # explicitly tag your specs with their type, e.g.:
51
+ #
52
+ # RSpec.describe UsersController, type: :controller do
53
+ # # ...
54
+ # end
55
+ #
56
+ # The different available types are documented in the features, such as in
57
+ # https://relishapp.com/rspec/rspec-rails/docs
58
+ config.infer_spec_type_from_file_location!
59
+
60
+ # Filter lines from Rails gems in backtraces.
61
+ config.filter_rails_from_backtrace!
62
+ # arbitrary gems may also be filtered via:
63
+ # config.filter_gems_from_backtrace("gem name")
64
+
65
+ config.around(:each) do |example|
66
+ DatabaseCleaner[:spanner].cleaning do
67
+ example.run
68
+ end
69
+ end
70
+ end
@@ -0,0 +1,14 @@
1
+ RSpec.describe "sample" do
2
+ before do
3
+ customer = create(:customer)
4
+ product = create(:product)
5
+ create(:order, customer: customer, product: product)
6
+ singer = create(:singer)
7
+ album = create(:album, singer: singer)
8
+ create(:song, singer: singer, album: album)
9
+ end
10
+
11
+ it "sample" do
12
+ expect(true).to be true
13
+ end
14
+ end
@@ -0,0 +1,94 @@
1
+ # This file was generated by the `rails generate rspec:install` command. Conventionally, all
2
+ # specs live under a `spec` directory, which RSpec adds to the `$LOAD_PATH`.
3
+ # The generated `.rspec` file contains `--require spec_helper` which will cause
4
+ # this file to always be loaded, without a need to explicitly require it in any
5
+ # files.
6
+ #
7
+ # Given that it is always loaded, you are encouraged to keep this file as
8
+ # light-weight as possible. Requiring heavyweight dependencies from this file
9
+ # will add to the boot time of your test suite on EVERY test run, even for an
10
+ # individual file that may not need all of that loaded. Instead, consider making
11
+ # a separate helper file that requires the additional dependencies and performs
12
+ # the additional setup, and require it from the spec files that actually need
13
+ # it.
14
+ #
15
+ # See https://rubydoc.info/gems/rspec-core/RSpec/Core/Configuration
16
+ RSpec.configure do |config|
17
+ # rspec-expectations config goes here. You can use an alternate
18
+ # assertion/expectation library such as wrong or the stdlib/minitest
19
+ # assertions if you prefer.
20
+ config.expect_with :rspec do |expectations|
21
+ # This option will default to `true` in RSpec 4. It makes the `description`
22
+ # and `failure_message` of custom matchers include text for helper methods
23
+ # defined using `chain`, e.g.:
24
+ # be_bigger_than(2).and_smaller_than(4).description
25
+ # # => "be bigger than 2 and smaller than 4"
26
+ # ...rather than:
27
+ # # => "be bigger than 2"
28
+ expectations.include_chain_clauses_in_custom_matcher_descriptions = true
29
+ end
30
+
31
+ # rspec-mocks config goes here. You can use an alternate test double
32
+ # library (such as bogus or mocha) by changing the `mock_with` option here.
33
+ config.mock_with :rspec do |mocks|
34
+ # Prevents you from mocking or stubbing a method that does not exist on
35
+ # a real object. This is generally recommended, and will default to
36
+ # `true` in RSpec 4.
37
+ mocks.verify_partial_doubles = true
38
+ end
39
+
40
+ # This option will default to `:apply_to_host_groups` in RSpec 4 (and will
41
+ # have no way to turn it off -- the option exists only for backwards
42
+ # compatibility in RSpec 3). It causes shared context metadata to be
43
+ # inherited by the metadata hash of host groups and examples, rather than
44
+ # triggering implicit auto-inclusion in groups with matching metadata.
45
+ config.shared_context_metadata_behavior = :apply_to_host_groups
46
+
47
+ # The settings below are suggested to provide a good initial experience
48
+ # with RSpec, but feel free to customize to your heart's content.
49
+ =begin
50
+ # This allows you to limit a spec run to individual examples or groups
51
+ # you care about by tagging them with `:focus` metadata. When nothing
52
+ # is tagged with `:focus`, all examples get run. RSpec also provides
53
+ # aliases for `it`, `describe`, and `context` that include `:focus`
54
+ # metadata: `fit`, `fdescribe` and `fcontext`, respectively.
55
+ config.filter_run_when_matching :focus
56
+
57
+ # Allows RSpec to persist some state between runs in order to support
58
+ # the `--only-failures` and `--next-failure` CLI options. We recommend
59
+ # you configure your source control system to ignore this file.
60
+ config.example_status_persistence_file_path = "spec/examples.txt"
61
+
62
+ # Limits the available syntax to the non-monkey patched syntax that is
63
+ # recommended. For more details, see:
64
+ # https://relishapp.com/rspec/rspec-core/docs/configuration/zero-monkey-patching-mode
65
+ config.disable_monkey_patching!
66
+
67
+ # Many RSpec users commonly either run the entire suite or an individual
68
+ # file, and it's useful to allow more verbose output when running an
69
+ # individual spec file.
70
+ if config.files_to_run.one?
71
+ # Use the documentation formatter for detailed output,
72
+ # unless a formatter has already been configured
73
+ # (e.g. via a command-line flag).
74
+ config.default_formatter = "doc"
75
+ end
76
+
77
+ # Print the 10 slowest examples and example groups at the
78
+ # end of the spec run, to help surface which specs are running
79
+ # particularly slow.
80
+ config.profile_examples = 10
81
+
82
+ # Run specs in random order to surface order dependencies. If you find an
83
+ # order dependency and want to debug it, you can fix the order by providing
84
+ # the seed, which is printed after each run.
85
+ # --seed 1234
86
+ config.order = :random
87
+
88
+ # Seed global randomization in this process using the `--seed` CLI option.
89
+ # Setting this allows you to use `--seed` to deterministically reproduce
90
+ # test failures related to randomization by passing the same `--seed` value
91
+ # as the one that triggered the failure.
92
+ Kernel.srand config.seed
93
+ =end
94
+ end
@@ -0,0 +1,3 @@
1
+ RSpec.configure do |config|
2
+ config.include FactoryBot::Syntax::Methods
3
+ end
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
@@ -33,26 +33,40 @@ module DatabaseCleaner
33
33
  def initialize(
34
34
  only: [],
35
35
  except: [],
36
+ batch_deletion: false,
36
37
  cache_tables: true
37
38
  )
38
39
  @only = only
39
40
  @except = except
41
+ @batch_deletion = batch_deletion
40
42
  @cache_tables = cache_tables
41
43
 
42
44
  @deletable_tables = {}
43
45
  end
44
46
 
45
47
  def clean
46
- sorted_table_groups.each do |group|
47
- clean_table_group(group)
48
+ if @batch_deletion
49
+ clean_as_batch
50
+ else
51
+ clean_each
48
52
  end
49
53
  end
50
54
 
51
55
  private
52
56
 
53
- def clean_table_group(group)
54
- group.each do |table|
55
- client.delete(table) if deletable?(table)
57
+ def clean_as_batch
58
+ client.transaction do |tx|
59
+ tx.batch_update do |b|
60
+ each_deletable_table do |table|
61
+ b.batch_update("DELETE #{table} WHERE TRUE")
62
+ end
63
+ end
64
+ end
65
+ end
66
+
67
+ def clean_each
68
+ each_deletable_table do |table|
69
+ client.delete(table)
56
70
  end
57
71
  end
58
72
 
@@ -69,6 +83,18 @@ module DatabaseCleaner
69
83
  end
70
84
  end
71
85
 
86
+ def each_deletable_table(&block)
87
+ each_group do |group|
88
+ group.each do |table|
89
+ block.call(table) if deletable?(table)
90
+ end
91
+ end
92
+ end
93
+
94
+ def each_group(&block)
95
+ sorted_table_groups.each(&block)
96
+ end
97
+
72
98
  def sorted_table_groups
73
99
  return @sorted_table_groups if @cache_tables && @sorted_table_groups
74
100
 
@@ -2,6 +2,6 @@
2
2
 
3
3
  module DatabaseCleaner
4
4
  module Spanner
5
- VERSION = "0.1.0"
5
+ VERSION = "0.1.1"
6
6
  end
7
7
  end