require_relative 'test_helper'
require 'json'

redis = Gaptool.redis

describe 'data helpers' do
  before(:each) do
    redis.flushall
    DH.set_config('chef_repo', 'myrepo')
    DH.set_config('chef_branch', 'master')
  end

  after(:all) do
    redis.flushall
  end

  def data
    {
      'role' => 'role',
      'environment' => 'env'
    }
  end

  def instid
    'i-1234567'
  end

  it 'should add a server' do
    expect { DH.addserver('i-1234567', {}, 'secret') }.to raise_error(ArgumentError)
    expect { DH.addserver('', data, 'secret') }.to raise_error(ArgumentError)

    DH.addserver(instid, data, 'secret')
    server = DH.get_server_data(instid)
    expect(server).to eq(data.merge('instance' => instid,
                                    'chef_repo' => 'myrepo',
                                    'chef_branch' => 'master',
                                    'registered' => 'false',
                                    'secret' => 'secret',
                                    'apps' => []))
  end

  it 'should add a registered server' do
    DH.addserver(instid, data, nil)
    server = DH.get_server_data(instid)
    expect(server).to eq(data.merge('instance' => instid,
                                    'chef_repo' => 'myrepo',
                                    'chef_branch' => 'master',
                                    'apps' => []))
  end

  it 'should register a server' do
    DH.addserver(instid, data, 'secret')
    res = DH.register_server(data['role'], data['environment'], '')
    expect(res).to be(nil)

    res = DH.register_server(data['role'], data['environment'], 'secret')
    expect(res).to eq(instid)
    server = DH.get_server_data(instid)
    expect(server).to eq(data.merge('instance' => instid,
                                    'chef_repo' => 'myrepo',
                                    'chef_branch' => 'master',
                                    'apps' => []))
  end

  it 'should remove a server' do
    DH.addserver(instid, data, 'secret')
    DH.register_server(data['role'], data['environment'], 'secret')
    res = DH.rmserver(instid)
    expect(res).to eq(instid)
    server = DH.get_server_data(instid)
    expect(server).to be(nil)

    res = DH.rmserver(instid)
    expect(res).to be(nil)
  end

  it 'should set a config' do
    c = DH.get_config('fakekey')
    expect(c).to be(nil)

    DH.set_config('fakekey', 'value')
    c = DH.get_config('fakekey')
    expect(c).to eq('value')
  end

  it 'should get the runlist for a node from the role' do
    DH.save_role_data('role', chef_runlist: ['recipe[myrecipe]'].to_json)
    DH.addserver(instid, data, nil)
    role = DH.get_role_data('role', 'testenv')
    expect(role).to eq('chef_runlist' => ['recipe[myrecipe]'],
                       'apps' => [],
                       'amis' => {},
                       'sg' => {})
    server = DH.get_server_data(instid)
    expect(server).to eq(data.merge('instance' => instid,
                                    'chef_runlist' => ['recipe[myrecipe]'],
                                    'chef_repo' => 'myrepo',
                                    'chef_branch' => 'master',
                                    'apps' => []))
  end

  it 'shoud get the ami for a node from the role' do
    DH.save_role_data('role', ami: 'ami-1234567')
    DH.addserver(instid, data, nil)
    role = DH.get_role_data('role', 'testenv')
    expect(role).to eq('ami' => 'ami-1234567',
                       'apps' => [],
                       'amis' => {},
                       'sg' => {})
  end

  it 'shoud get the sg for a node from the role' do
    DH.save_role_data('role', ami: 'ami-1234567', 'sg' => { 'test' => 'my_role' })
    role = DH.get_role_data('role', 'testenv')
    expect(role).to eq('ami' => 'ami-1234567',
                       'apps' => [],
                       'amis' => {},
                       'sg' => { 'test' => 'my_role' })
  end
end