require_relative 'test_helper' require 'json' require 'set' redis = Gaptool.redis describe 'Test API' do before(:all) do ENV['DRYRUN'] = 'true' ENV['GAPTOOL_CHECK_CLIENT_VERSION'] = nil end after(:all) do ENV['DRYRUN'] = nil ENV['GAPTOOL_CHECK_CLIENT_VERSION'] = nil end before(:each) do header 'X-GAPTOOL-USER', 'test' header 'X-GAPTOOL-KEY', 'test' header 'X-GAPTOOL-VERSION', version redis.flushall DH.useradd('test', 'test') $time = Time.now end def add_and_register_server(data = nil) data ||= host_data post '/init', data.to_json expect(last_response.status).to eq(200) res = JSON.parse(last_response.body) id = res['instance'] # there is no API to get the secret, get it from the database. secret = Gaptool::Data.get_server_data(id)['secret'] put '/register', { 'role' => data['role'], 'environment' => data['environment'], 'secret' => secret, 'zone' => data['zone'] }.to_json expect(last_response.status).to eq(200) res end def host_data { 'security_group' => 'mysg', 'role' => 'testrole', 'environment' => 'testenv', 'ami' => 'ami-1234567', 'chef_runlist' => ['recipe[myrecipe]'], 'terminable' => true, 'zone' => 'my-zone-1a', 'itype' => 'm1.type', 'hostname' => 'fake.hostname.gild.com', 'launch_time' => $time.to_i } end def expanded_runlist ['role[base]', 'recipe[myrecipe]'] end def version @v ||= File.read( File.realpath(File.join(File.dirname(__FILE__), '..', 'VERSION') )).strip end it 'should return 200' do get '/' expect(last_response.status).to eq(200) expect(last_response.content_type).to include('text/html') header 'Accept', 'application/json' get '/' expect(last_response.status).to eq(200) expect(last_response.content_type).to include('application/json') end it 'should create a instance' do post '/init', host_data.to_json expect(last_response.status).to eq(200) expect(JSON.parse(last_response.body).keys).to eq(%w(instance ami role hostname launch_time environment secret terminable security_group)) end it 'should get the runlist from the role' do DH.save_role_data(host_data['role'], chef_runlist: ['recipe[other]'].to_json) id = add_and_register_server(host_data.reject { |k, _v| k == 'chef_runlist' })['instance'] get "/host/testrole/testenv/#{id}" resp = JSON.parse(last_response.body) expect(resp.keys).to include('chef_runlist') expect(resp['chef_runlist']).to eq(['recipe[other]']) end it 'should get the security group from the role' do DH.save_role_data(host_data['role'], 'sg' => { host_data['environment'] => 'mysg-in-role' }) id = add_and_register_server(host_data.reject { |k, _v| k == 'security_group' })['instance'] get "/host/testrole/testenv/#{id}" resp = JSON.parse(last_response.body) expect(resp.keys).to include('security_group') expect(resp['security_group']).to eq('mysg-in-role') end it 'should get the default security group' do id = add_and_register_server(host_data.reject { |k, _v| k == 'security_group' })['instance'] get "/host/testrole/testenv/#{id}" resp = JSON.parse(last_response.body) expect(resp.keys).to include('security_group') expect(resp['security_group']).to eq("#{host_data['role']}-#{host_data['environment']}") end it 'should set instance unterminable' do id = add_and_register_server['instance'] get "/instance/#{id}" expect(last_response.status).to eq(200) resp = JSON.parse(last_response.body) expect(resp.keys).not_to include('terminable') patch "/instance/#{id}", { terminable: false }.to_json expect(last_response.status).to eq(200) resp = JSON.parse(last_response.body) expect(resp['terminable']).to eq(false) get "/instance/#{id}" expect(last_response.status).to eq(200) resp = JSON.parse(last_response.body) expect(resp['terminable']).to eq(false) post '/terminate', { 'id' => id }.to_json expect(last_response.status).to eq(409) patch "/instance/#{id}", { terminable: true }.to_json expect(last_response.status).to eq(200) resp = JSON.parse(last_response.body) expect(resp['terminable']).to eq(true) post '/terminate', { 'id' => id }.to_json expect(last_response.status).to eq(200) end it 'should set an instance as hidden' do id = add_and_register_server['instance'] get "/instance/#{id}" expect(last_response.status).to eq(200) resp = JSON.parse(last_response.body) expect(resp.keys).not_to include('hidden') patch "/instance/#{id}", { hidden: false }.to_json expect(last_response.status).to eq(200) resp = JSON.parse(last_response.body) expect(resp.keys).not_to include('hidden') get "/instance/#{id}" expect(last_response.status).to eq(200) resp = JSON.parse(last_response.body) expect(resp.keys).not_to include('hidden') patch "/instance/#{id}", { hidden: true }.to_json expect(last_response.status).to eq(200) resp = JSON.parse(last_response.body) expect(resp['hidden']).to eq(true) get "/instance/#{id}" expect(last_response.status).to eq(200) resp = JSON.parse(last_response.body) expect(resp['hidden']).to eq(true) get "/host/#{host_data['role']}/#{host_data['environment']}/#{id}" expect(last_response.status).to eq(200) resp = JSON.parse(last_response.body) expect(resp['hidden']).to eq(true) get '/hosts' expect(last_response.status).to eq(200) expect(JSON.parse(last_response.body)).to eq([]) get "/hosts/#{host_data['role']}" expect(last_response.status).to eq(200) expect(JSON.parse(last_response.body)).to eq([]) get "/hosts/#{host_data['role']}/#{host_data['environment']}" expect(last_response.status).to eq(200) expect(JSON.parse(last_response.body)).to eq([]) get "/hosts/ALL/#{host_data['environment']}" expect(last_response.status).to eq(200) expect(JSON.parse(last_response.body)).to eq([]) end it 'should remove default runlist' do id = add_and_register_server(host_data.merge('chef_runlist' => ['role[base]']))['instance'] get "/host/testrole/testenv/#{id}" resp = JSON.parse(last_response.body) expect(resp.keys).not_to include('chef_runlist') end it 'should remove default runlist from role' do DH.save_role_data(host_data['role'], chef_runlist: ['role[base]'].to_json) id = add_and_register_server(host_data.reject { |k, _v| k == 'chef_runlist' })['instance'] get "/host/testrole/testenv/#{id}" resp = JSON.parse(last_response.body) expect(resp.keys).not_to include('chef_runlist') expect(resp['chef_runlist']).to be(nil) end it 'should get the ami from the role' do DH.save_role_data(host_data['role'], 'amis' => { 'my-zone-1' => 'ami-other' }) res = add_and_register_server(host_data.reject { |k, _v| k == 'ami' }) expect(res['ami']).to eq('ami-other') get "/host/testrole/testenv/#{res['instance']}" resp = JSON.parse(last_response.body) expect(resp.keys).to_not include('ami') end it 'should get the ami from global conf' do DH.amis('my-zone-1' => 'ami-default') res = add_and_register_server(host_data.reject { |k, _v| k == 'ami' }) expect(res['ami']).to eq('ami-default') get "/host/testrole/testenv/#{res['instance']}" resp = JSON.parse(last_response.body) expect(resp.keys).to_not include('ami') end it 'should fail to parse client data' do post '/init', host_data expect(last_response.status).to eq(400) end it 'should fail to create a instance' do %w(role environment zone itype).each do |req| post '/init', host_data.reject { |k, _v| k == req }.to_json expect(last_response.status).to eq(400) expect(JSON.parse(last_response.body).keys).to eq(%w(result message)) end end it 'should fail for missing parameter to terminate' do post '/terminate' expect(last_response.status).to eq(400) end it 'should fail to terminate non-existing instance' do post '/terminate', { 'id' => 'i-1234567' }.to_json expect(last_response.status).to eq(404) end it 'should fail to terminate non-terminable instance' do post '/init', host_data.merge('terminable' => false).to_json expect(last_response.status).to eq(200) id = JSON.parse(last_response.body)['instance'] post '/terminate', { 'id' => id }.to_json expect(last_response.status).to eq(409) end it 'should terminate instance' do post '/init', host_data.to_json expect(last_response.status).to eq(200) id = JSON.parse(last_response.body)['instance'] post '/terminate', { 'id' => id }.to_json expect(last_response.status).to eq(200) expect(JSON.parse(last_response.body)).to eq(id => { 'status' => 'terminated' }) end it 'should fail to register a server' do required_keys = %w(role zone environment secret) hdata = host_data.select { |k, _v| required_keys.include?(k) } hdata['secret'] = 'mysecret' required_keys.each do |key| put '/register', hdata.select { |k, _v| k != key }.to_json expect(last_response.status).to eq(400) end put '/register', hdata.to_json expect(last_response.status).to eq(403) end it 'should register the server' do add_and_register_server expect(last_response.body).to include("-E #{host_data['environment']}") end it 'should init with the default runlist' do hdata = host_data.select{ |k, _v| k != 'chef_runlist' } add_and_register_server(hdata) expect(last_response.body).to include('"run_list":["role[base]"],') end it 'should init with the host runlist' do add_and_register_server expect(last_response.body).to include('"run_list":["role[base]","recipe[myrecipe]"],') end it 'should not find any hosts' do post '/init', host_data.to_json expect(last_response.status).to eq(200) get '/hosts' expect(last_response.status).to eq(200) expect(JSON.parse(last_response.body)).to eq([]) get "/hosts/#{host_data['role']}" expect(last_response.status).to eq(200) expect(JSON.parse(last_response.body)).to eq([]) end it 'should find an host' do id = add_and_register_server['instance'] get '/hosts' expect(last_response.status).to eq(200) exp_data = host_data.reject { |k, _v| k == 'terminable' }.merge('chef_runlist' => expanded_runlist, 'apps' => []) exp_data['instance'] = id expect(JSON.parse(last_response.body)).to eq([exp_data]) end it 'should find two hosts' do id1 = add_and_register_server['instance'] id2 = add_and_register_server['instance'] get '/hosts' expect(last_response.status).to eq(200) exp_data = [ host_data.reject { |k, _v| k == 'terminable' }.merge('instance' => id1, 'chef_runlist' => expanded_runlist, 'apps' => []), host_data.reject { |k, _v| k == 'terminable' }.merge('instance' => id2, 'chef_runlist' => expanded_runlist, 'apps' => []) ].to_set expect(JSON.parse(last_response.body).to_set).to eq(exp_data) end it 'should find one hidden host' do id1 = add_and_register_server['instance'] id2 = add_and_register_server['instance'] patch "/instance/#{id1}", { hidden: true }.to_json %W(/hosts /hosts/#{host_data['role']} /hosts/#{host_data['role']}/#{host_data['environment']} /hosts/ALL/#{host_data['environment']}).each do |url| get url expect(last_response.status).to eq(200) res = JSON.parse(last_response.body).map { |x| x.delete('apps'); x }.to_set exp_data = [host_data.reject { |k, _v| k == 'terminable' }.merge('instance' => id2, 'chef_runlist' => expanded_runlist)].to_set expect(res).to eq(exp_data) get url, hidden: true expect(last_response.status).to eq(200) exp_data = [ host_data.reject { |k, _v| k == 'terminable' }.merge('instance' => id1, 'chef_runlist' => expanded_runlist, 'hidden' => true), host_data.reject { |k, _v| k == 'terminable' }.merge('instance' => id2, 'chef_runlist' => expanded_runlist) ].to_set expect(JSON.parse(last_response.body).map { |x| x.delete('apps'); x }.to_set).to eq(exp_data) end end it 'should find an host by role' do other = host_data.merge('role' => 'otherrole') id1 = add_and_register_server(other)['instance'] add_and_register_server['instance'] get '/hosts/otherrole' expect(last_response.status).to eq(200) exp_data = [other.reject { |k, _v| k == 'terminable' }.merge('instance' => id1, 'chef_runlist' => expanded_runlist, 'apps' => [])] expect(JSON.parse(last_response.body)).to eq(exp_data) end it 'should find an host by id' do id = add_and_register_server['instance'] ["/instance/#{id}", "/host/testrole/testenv/#{id}", "/host/FAKE/FAKE/#{id}"].each do |url| get url expect(last_response.status).to eq(200) exp_data = host_data.reject { |k, _v| k == 'terminable' }.merge('instance' => id, 'chef_runlist' => expanded_runlist, 'apps' => []) expect(JSON.parse(last_response.body)).to eq(exp_data) end end it 'should return the host attr json' do DH.set_config('chef_repo', 'FAKECHEFREPO') DH.set_config('chef_branch', 'chefbranch') DH.set_config('url', 'localhost:666') id = add_and_register_server['instance'] get "/instance/#{id}/attrs" expect(last_response.status).to eq(200) expect(JSON.parse(last_response.body)).to eq( 'apps' => [], 'branch' => 'master', 'chefbranch' => 'chefbranch', 'chefrepo' => 'FAKECHEFREPO', 'deploy_apps' => [], 'environment' => host_data['environment'], 'gaptool' => { 'user' => 'test', 'key' => 'test', 'url' => 'localhost:666' }, 'role' => host_data['role'], 'run_list' => expanded_runlist ) end it 'old clients should receive apps as strings' do header 'X-GAPTOOL-VERSION', '0.7.0' id = add_and_register_server['instance'] ["/instance/#{id}", "/host/testrole/testenv/#{id}", "/host/FAKE/FAKE/#{id}"].each do |url| get url expect(last_response.status).to eq(200) exp_data = host_data.reject { |k, _v| k == 'terminable' }.merge('instance' => id, 'chef_runlist' => expanded_runlist, 'apps' => '[]') expect(JSON.parse(last_response.body)).to eq(exp_data) end end it 'clients with no version should be handled as old' do header 'X-GAPTOOL-VERSION', nil id = add_and_register_server['instance'] ["/instance/#{id}", "/host/testrole/testenv/#{id}", "/host/FAKE/FAKE/#{id}"].each do |url| get url expect(last_response.status).to eq(200) exp_data = host_data.reject { |k, _v| k == 'terminable' }.merge('instance' => id, 'chef_runlist' => expanded_runlist, 'apps' => '[]') expect(JSON.parse(last_response.body)).to eq(exp_data) end end it 'should not find an host by id' do add_and_register_server['instance'] get '/instance/other' expect(last_response.status).to eq(404) expect(JSON.parse(last_response.body)).to eq( 'result' => 'error', 'message' => "instance with id 'other' not found" ) end it 'should find all hosts by environment and role' do id1 = add_and_register_server['instance'] id2 = add_and_register_server['instance'] add_and_register_server(host_data.merge('environment' => 'otherenv'))['instance'] id4 = add_and_register_server(host_data.merge('role' => 'otherrole'))['instance'] get '/hosts/testrole/testenv' expect(last_response.status).to eq(200) d = host_data.reject { |k, _v| k == 'terminable' }.merge('chef_runlist' => expanded_runlist, 'apps' => []) exp_data = [ d.merge('instance' => id1), d.merge('instance' => id2) ].to_set expect(JSON.parse(last_response.body).to_set).to eq(exp_data) get '/hosts/ALL/testenv' expect(last_response.status).to eq(200) exp_data = [ d.merge('instance' => id1), d.merge('instance' => id2), d.merge('instance' => id4, 'role' => 'otherrole') ].to_set expect(JSON.parse(last_response.body).to_set).to eq(exp_data) end it 'should return the server version' do get '/version' expect(last_response.status).to eq(200) expect(JSON.parse(last_response.body)).to eq('server_version' => version, 'api' => { 'v0' => '/' }) end it 'should return the list of apps' do DH.add_app('firstapp', 'testrole', 'testenv') DH.add_app('secondapp', 'testrole', 'testenv') apps_list = { 'app:firstapp' => { 'testenv' => 'testrole' }, 'app:secondapp' => { 'testenv' => 'testrole' } } get '/apps' expect(last_response.status).to eq(200) expect(JSON.parse(last_response.body)).to eq(apps_list) end it 'should fail as client did not send version' do ENV['GAPTOOL_CHECK_CLIENT_VERSION'] = '1' header 'X-GAPTOOL-VERSION', nil get '/version' expect(last_response.status).to eq(400) resp = JSON.parse(last_response.body) expect(resp['result']).to eq('error') expect(resp['message']).to match(/^Invalid version/) end it 'should not check version for clients' do ENV['GAPTOOL_CHECK_CLIENT_VERSION'] = nil header 'X_GAPTOOL_VERSION', nil get '/version' expect(last_response.status).to eq(200) end it 'should reject old clients versions' do ENV['GAPTOOL_CHECK_CLIENT_VERSION'] = '1' srv = Versionomy.parse(version) cl = Versionomy.create(major: srv.major, minor: srv.minor - 1, tiny: 0) header 'X_GAPTOOL_VERSION', cl.to_s get '/version' expect(last_response.status).to eq(400) resp = JSON.parse(last_response.body) expect(resp['result']).to eq('error') expect(resp['message']).to match(/^Invalid version/) end end