require_relative 'integration_spec_helper'

include GivenFilesystemSpecHelpers
include CliTester

def run_helper(working_dir, sprint_number, extra_args = [])
  helper_dir = File.expand_path('../../../scripts', __FILE__)
  args = ['run']
  args += ['-v', "#{helper_dir}:/trollolo/helper"]
  args += ['-v', "#{working_dir}:/trollolo/data"]
  args += ['-w', '/trollolo/data']
  args += ['trollolo-matplotlib']
  args += ['/trollolo/helper/create_burndown.py', sprint_number]
  args += extra_args
  run_command(cmd: 'docker', args: args)
end

def compare_images_for_sprint(sprint_number, extra_args = [])
  @working_dir = given_directory do
    given_file("burndown-data-#{sprint_number}.yaml", from: "create_burndown_helper/burndown-data-#{sprint_number}.yaml")
  end

  result = run_helper(@working_dir, sprint_number, extra_args)
  expect(result).to exit_with_success('')
  expect(File.join(@working_dir, "burndown-#{sprint_number}.png")).
    to be_same_image_as("create_burndown_helper/burndown-#{sprint_number}.png")
end

describe 'create_burndown.py' do
  use_given_filesystem(keep_files: true)

  before(:all) do
    if `docker images -q trollolo-matplotlib`.empty?
      raise "Required docker image 'trollolo-matplotlib' not found. Build it with 'docker build -t trollolo-matplotlib spec/containers/matplotlib'"
    end
  end

  it 'creates burndown chart with varying number of total story points and tasks' do
    compare_images_for_sprint('23')
  end

  it 'creates burndown chart with done tasks at the beginning' do
    compare_images_for_sprint('31')
  end

  it 'creates burndown chart of unfinished sprint' do
    compare_images_for_sprint('35')
  end

  it 'creates burndown chart with fast lane and no tasks' do
    compare_images_for_sprint('08', ['--no-tasks', '--with-fast-lane'])
  end

  it 'creates perfect burndown chart' do
    compare_images_for_sprint('42')
  end

  it 'creates burndown chart with unplanned cards' do
    compare_images_for_sprint('56')
  end

  it 'creates burndown chart with unplanned tasks on day one' do
    compare_images_for_sprint('01')
  end
end