$:.unshift(File.dirname(__FILE__) + '/../lib/')

RMAGICK_BYPASS_VERSION_TEST = true

require 'test/unit'
require 'gruff'
require 'fileutils'

TEST_OUTPUT_DIR = File.dirname(__FILE__) + "/output#{'_java' if RUBY_PLATFORM == 'java'}"
FileUtils.mkdir_p(TEST_OUTPUT_DIR)
FileUtils.rm_f Dir[TEST_OUTPUT_DIR + '/*']

if ENV['RM_INFO'] && RUBY_VERSION =~ /^(1\.9|2\.0)\./
  require 'minitest/reporters'
  MiniTest::Reporters.use!
end

class Gruff::Base
  alias :write_org :write

  def write(filename='graph.png')
    basefilename = File.basename(filename).split('.')[0..-2].join('.')
    extension = filename.slice(/\.[^\.]*$/)
    testfilename = File.join(TEST_OUTPUT_DIR, basefilename) + extension
    counter = 0
    while File.exists?(testfilename)
      counter += 1
      testfilename = File.join(TEST_OUTPUT_DIR, basefilename) + "-#{counter}#{extension}"
    end
    write_org(testfilename)
  end
end

class GruffTestCase < Test::Unit::TestCase
  def setup
    srand 42
    @datasets = [
        [:Jimmy, [25, 36, 86, 39, 25, 31, 79, 88]],
        [:Charles, [80, 54, 67, 54, 68, 70, 90, 95]],
        [:Julie, [22, 29, 35, 38, 36, 40, 46, 57]],
        [:Jane, [95, 95, 95, 90, 85, 80, 88, 100]],
        [:Philip, [90, 34, 23, 12, 78, 89, 98, 88]],
        [:Arthur, [5, 10, 13, 11, 6, 16, 22, 32]],
    ]

    @labels = {
        0 => '5/6',
        1 => '5/15',
        2 => '5/24',
        3 => '5/30',
        4 => '6/4',
        5 => '6/12',
        6 => '6/21',
        7 => '6/28',
    }
  end

  def setup_single_dataset
    @datasets = [
        [:Jimmy, [25, 36, 86]]
    ]

    @labels = {
        0 => 'You',
        1 => 'Average',
        2 => 'Lifetime'
    }
  end

  def setup_wide_dataset
    @datasets = [
        ['Auto', 25],
        ['Food', 5],
        ['Entertainment', 15]
    ]

    @labels = {0 => 'This Month'}
  end

  def test_dummy
    assert true
  end

  protected

  # Generate graphs at several sizes.
  #
  # Also writes the graph to disk.
  #
  #   graph_sized 'bar_basic' do |g|
  #     g.data('students', [1, 2, 3, 4])
  #   end
  #
  def graph_sized(filename, sizes=['', 400])
    class_name = self.class.name.gsub(/^TestGruff/, '')
    Array(sizes).each do |size|
      g = instance_eval("Gruff::#{class_name}.new #{size}")
      g.title = "#{class_name} Graph"
      yield g
      write_test_file g, "#{filename}_#{size}.png"
    end
  end

  def write_test_file(graph, filename)
    testfilename = [TEST_OUTPUT_DIR, filename].join('/')
    basefilename = filename.split('.')[0..-2].join('.')
    extension = filename.slice(/\..*$/)
    counter = 0
    while File.exists? testfilename
      counter += 1
      testfilename = [TEST_OUTPUT_DIR, basefilename].join('/') + "-#{counter}#{extension}"
    end
    graph.write(testfilename)
  end

  ##
  # Example:
  #
  #   setup_basic_graph Gruff::Pie, 400
  #
  def setup_basic_graph(*args)
    klass, size = Gruff::Bar, 400
    # Allow args to be klass, size or just klass or just size.
    #
    # TODO Refactor
    case args.length
    when 1
      case args[0]
      when Fixnum
        size = args[0]
        klass = eval("Gruff::#{self.class.name.gsub(/^TestGruff/, '')}")
      when String
        size = args[0]
        klass = eval("Gruff::#{self.class.name.gsub(/^TestGruff/, '')}")
      else
        klass = args[0]
      end
    when 2
      klass, size = args[0], args[1]
    end

    g = klass.new(size)
    g.title = 'My Bar Graph'
    g.labels = @labels
    g.font = '/Library/Fonts/Verdana.ttf'


    @datasets.each do |data|
      g.data(data[0], data[1])
    end
    g
  end

end
