#!/usr/bin/env ruby

$LOAD_PATH.unshift File.expand_path('..', __FILE__)
require 'helper'

class TestAggregation < TestHBaseJRubyBase
  def test_aggregation
    omit 'AggregationClient is not found' unless @aggregation

    range = 1..100
    @table.put range.map { |idx|
      { idx => { 'cf1:a' => idx, 'cf1:b' => idx * 2 } }
    }.reduce(&:merge)

    assert_nil @table.enable_aggregation!
    assert_nil @table.enable_aggregation! # no prob!

    lci = org.apache.hadoop.hbase.client.coprocessor.LongColumnInterpreter.new
    scope = @table.range(range)
    [nil, :fixnum, :long, lci].each do |ci|
      assert_equal 100,  scope.project('cf1:a').aggregate(:row_count, *[*ci].compact)
      assert_equal 5050, scope.project('cf1:a').aggregate(:sum, *[*ci].compact)
      assert_equal 1,    scope.project('cf1:a').aggregate(:min, *[*ci].compact)
      assert_equal 100,  scope.project('cf1:a').aggregate(:max, *[*ci].compact)
      assert_equal 50.5, scope.project('cf1:a').aggregate(:avg, *[*ci].compact)
      assert_equal 28,   scope.project('cf1:a').aggregate(:std, *[*ci].compact).to_i # FIXME: 28 or 29?
    end

    [%w[cf1:a cf1:b], %w[cf1]].each do |prj|
      assert_equal 5050 * 3, scope.project(*prj).aggregate(:sum)
      assert_equal 1,        scope.project(*prj).aggregate(:min)
      assert_equal 200,      scope.project(*prj).aggregate(:max)
    end

    # No projection
    assert_raises(ArgumentError) { @table.aggregate(:sum) }
    assert_raises(ArgumentError) { scope.aggregate(:sum) }

    # Invalid type
    assert_raises(ArgumentError) { scope.project('cf1:a').aggregate(:sum, :double) }
  end
end