118 lines
3.0 KiB
Ruby
118 lines
3.0 KiB
Ruby
ENV["RACK_ENV"] = "test"
|
|
|
|
require 'rubygems'
|
|
require 'bundler'
|
|
Bundler.require
|
|
|
|
require_relative '../liquor-cabinet'
|
|
require 'minitest/autorun'
|
|
require 'rack/test'
|
|
require 'purdytest'
|
|
require 'riak'
|
|
require "redis"
|
|
require "rest_client"
|
|
require "minitest/stub_any_instance"
|
|
require "ostruct"
|
|
|
|
def app
|
|
LiquorCabinet
|
|
end
|
|
|
|
app.set :environment, :test
|
|
|
|
def wait_a_second
|
|
now = Time.now.to_i
|
|
while Time.now.to_i == now; end
|
|
end
|
|
|
|
def write_last_response_to_file(filename = "last_response.html")
|
|
File.open(filename, "w") do |f|
|
|
f.write last_response.body
|
|
end
|
|
end
|
|
|
|
alias context describe
|
|
|
|
if app.settings.respond_to? :redis
|
|
def redis
|
|
@redis ||= Redis.new(host: app.settings.redis["host"], port: app.settings.redis["port"])
|
|
end
|
|
|
|
def purge_redis
|
|
redis.keys("rs_*").each do |key|
|
|
redis.del key
|
|
end
|
|
end
|
|
end
|
|
|
|
if app.settings.respond_to? :riak
|
|
::Riak.disable_list_keys_warnings = true
|
|
|
|
def client
|
|
@client ||= ::Riak::Client.new(:host => app.settings.riak['host'],
|
|
:http_port => app.settings.riak['http_port'])
|
|
end
|
|
|
|
def data_bucket
|
|
@data_bucket ||= begin
|
|
bucket = client.bucket(app.settings.riak['buckets']['data'])
|
|
bucket.allow_mult = false
|
|
bucket
|
|
end
|
|
end
|
|
|
|
def directory_bucket
|
|
@directory_bucket ||= begin
|
|
bucket = client.bucket(app.settings.riak['buckets']['directories'])
|
|
bucket.allow_mult = false
|
|
bucket
|
|
end
|
|
end
|
|
|
|
def auth_bucket
|
|
@auth_bucket ||= begin
|
|
bucket = client.bucket(app.settings.riak['buckets']['authorizations'])
|
|
bucket.allow_mult = false
|
|
bucket
|
|
end
|
|
end
|
|
|
|
def opslog_bucket
|
|
@opslog_bucket ||= begin
|
|
bucket = client.bucket(app.settings.riak['buckets']['opslog'])
|
|
bucket.allow_mult = false
|
|
bucket
|
|
end
|
|
end
|
|
|
|
def cs_credentials
|
|
@cs_credentials ||= begin
|
|
credentials = File.read(app.settings.riak['riak_cs']['credentials_file'])
|
|
JSON.parse(credentials)
|
|
end
|
|
end
|
|
|
|
def cs_client
|
|
@cs_client ||= Fog::Storage.new({
|
|
:provider => 'AWS',
|
|
:aws_access_key_id => cs_credentials['key_id'],
|
|
:aws_secret_access_key => cs_credentials['key_secret'],
|
|
:endpoint => app.settings.riak['riak_cs']['endpoint']
|
|
})
|
|
end
|
|
|
|
def cs_binary_bucket
|
|
@cs_binary_bucket ||= cs_client.directories.create(:key => app.settings.riak['buckets']['cs_binaries'])
|
|
end
|
|
|
|
def purge_all_buckets
|
|
[data_bucket, directory_bucket, auth_bucket, opslog_bucket].each do |bucket|
|
|
bucket.keys.each {|key| bucket.delete key}
|
|
end
|
|
|
|
cs_binary_bucket.files.each do |file|
|
|
file.destroy
|
|
end
|
|
end
|
|
end
|