require 'tempfile'

assert('no files') do
  o = `bin/mruby-strip 2>&1`
  assert_equal 1, $?.exitstatus
  assert_equal "no files to strip\n", o
end

assert('file not found') do
  o = `bin/mruby-strip not_found.mrb 2>&1`
  assert_equal 1, $?.exitstatus
  assert_equal "can't open file not_found.mrb\n", o
end

assert('not irep file') do
  t = Tempfile.new('script.rb')
  t.write 'p test\n'
  t.flush
  o = `bin/mruby-strip #{t.path} 2>&1`
  assert_equal 1, $?.exitstatus
  assert_equal "can't read irep file #{t.path}\n", o
end

assert('success') do
  script_file, compiled1, compiled2 =
    Tempfile.new('script.rb'), Tempfile.new('c1.mrb'), Tempfile.new('c2.mrb')
  script_file.write "p 'test'\n"
  script_file.flush
  `bin/mrbc -g -o #{compiled1.path} #{script_file.path}`
  `bin/mrbc -g -o #{compiled2.path} #{script_file.path}`

  o = `bin/mruby-strip #{compiled1.path}`
  assert_equal 0, $?.exitstatus
  assert_equal "", o
  assert_equal `bin/mruby #{script_file.path}`, `bin/mruby -b #{compiled1.path}`

  o = `bin/mruby-strip #{compiled1.path} #{compiled2.path}`
  assert_equal 0, $?.exitstatus
  assert_equal "", o
end

assert('check debug section') do
  script_file, with_debug, without_debug =
    Tempfile.new('script.rb'), Tempfile.new('c1.mrb'), Tempfile.new('c2.mrb')
  script_file.write "p 'test'\n"
  script_file.flush
  `bin/mrbc -o #{without_debug.path} #{script_file.path}`
  `bin/mrbc -g -o #{with_debug.path} #{script_file.path}`

  assert_true with_debug.size >= without_debug.size

  `bin/mruby-strip #{with_debug.path}`
  assert_equal without_debug.size, with_debug.size
end