1 class DatabaseController < ApplicationController
2 skip_before_filter :find_object_by_uuid
3 skip_before_filter :render_404_if_no_object
4 before_filter :admin_required
6 raise ArvadosModel::PermissionDeniedError unless Rails.env == 'test'
8 # Sanity check: If someone has actually logged in here, this might
9 # not really be a throwaway database. Client test suites should
10 # use @example.com email addresses when creating user records, so
11 # we can tell they're not valuable.
13 where('email is null or email not like ?', '%@example.com').
16 YAML::load_file(File.expand_path('../../../test/fixtures/users.yml',
18 values.collect { |u| u['uuid'] }
19 unexpected_uuids = user_uuids - fixture_uuids
20 if unexpected_uuids.any?
21 logger.error("Running in test environment, but non-fixture users exist: " +
22 "#{unexpected_uuids}")
23 raise ArvadosModel::PermissionDeniedError
26 require 'active_record/fixtures'
28 # What kinds of fixtures do we have?
29 fixturesets = Dir.glob(Rails.root.join('test', 'fixtures', '*.yml')).
30 collect { |yml| yml.match(/([^\/]*)\.yml$/)[1] }
32 ActiveRecord::Base.transaction do
33 # Avoid deadlock by locking all tables before doing anything
35 table_names = '"' + fixturesets.sort.join('","') + '"'
36 ActiveRecord::Base.connection.execute \
37 "LOCK TABLE #{table_names} IN ACCESS EXCLUSIVE MODE"
39 # Delete existing fixtures (and everything else) from fixture
41 fixturesets.each do |x|
42 x.classify.constantize.unscoped.delete_all
45 # create_fixtures() is a no-op for cached fixture sets, so
47 ActiveRecord::Fixtures.reset_cache
48 ActiveRecord::Fixtures.
49 create_fixtures(Rails.root.join('test', 'fixtures'), fixturesets)
51 # Dump cache of permissions etc.
53 ActiveRecord::Base.connection.clear_query_cache
55 # Reload database seeds
60 render json: {success: true}