require 'spec_helper'
require 'rack/test'

module Vmpooler
  class API
    module Helpers
      def authenticate(auth, username_str, password_str)
        username_str == 'admin' and password_str == 's3cr3t'
      end
    end
  end
end

def has_set_tag?(vm, tag, value)
  value == redis.hget("vmpooler__vm__#{vm}", "tag:#{tag}")
end

describe Vmpooler::API::V1 do
  include Rack::Test::Methods

  def app()
    Vmpooler::API
  end

  describe '/status' do
    let(:prefix) { '/api/v1' }

    let(:config) {
      {
        config: {
          'site_name' => 'test pooler',
          'vm_lifetime_auth' => 2,
        },
        pools: [
          {'name' => 'pool1', 'size' => 5},
          {'name' => 'pool2', 'size' => 10}
        ],
        alias: { 'poolone' => 'pool1' },
      }
    }

    let(:current_time) { Time.now }

    before(:each) do
      redis.flushdb

      app.settings.set :config, config
      app.settings.set :redis, redis
      app.settings.set :config, auth: false
      create_token('abcdefghijklmnopqrstuvwxyz012345', 'jdoe', current_time)
    end

    describe 'GET /status' do
      it 'returns the configured maximum size for each pool' do
        get "#{prefix}/status/"

        # of course /status doesn't conform to the weird standard everything else uses...
        expect(last_response.header['Content-Type']).to eq('application/json')
        result = JSON.parse(last_response.body)
        expect(result["pools"]["pool1"]["max"]).to be(5)
        expect(result["pools"]["pool2"]["max"]).to be(10)
      end

      it 'returns the number of ready vms for each pool' do
        3.times {|i| create_ready_vm("pool1", "vm-#{i}") }
        get "#{prefix}/status/"

        # of course /status doesn't conform to the weird standard everything else uses...
        expect(last_response.header['Content-Type']).to eq('application/json')
        result = JSON.parse(last_response.body)
        expect(result["pools"]["pool1"]["ready"]).to be(3)
        expect(result["pools"]["pool2"]["ready"]).to be(0)
      end

      it 'returns the number of running vms for each pool' do
        3.times {|i| create_running_vm("pool1", "vm-#{i}") }
        4.times {|i| create_running_vm("pool2", "vm-#{i}") }

        get "#{prefix}/status/"

        # of course /status doesn't conform to the weird standard everything else uses...
        expect(last_response.header['Content-Type']).to eq('application/json')
        result = JSON.parse(last_response.body)
        expect(result["pools"]["pool1"]["running"]).to be(3)
        expect(result["pools"]["pool2"]["running"]).to be(4)
      end

      it 'returns the number of pending vms for each pool' do
        3.times {|i| create_pending_vm("pool1", "vm-#{i}") }
        4.times {|i| create_pending_vm("pool2", "vm-#{i}") }

        get "#{prefix}/status/"

        # of course /status doesn't conform to the weird standard everything else uses...
        expect(last_response.header['Content-Type']).to eq('application/json')
        result = JSON.parse(last_response.body)
        expect(result["pools"]["pool1"]["pending"]).to be(3)
        expect(result["pools"]["pool2"]["pending"]).to be(4)
      end

      it '(for v1 backwards compatibility) lists any empty pools in the status section' do
        get "#{prefix}/status/"

        # of course /status doesn't conform to the weird standard everything else uses...
        expect(last_response.header['Content-Type']).to eq('application/json')
        result = JSON.parse(last_response.body)
        expect(result["status"]["empty"].sort).to eq(["pool1", "pool2"])
      end
    end
  end
end