require "test_helper"

class GetProcessMemTest < Test::Unit::TestCase
  def setup
    @mem = GetProcessMem.new
  end

  def test_different_pid_returns_different_memory
    pid = Process.spawn("tail -f Gemfile")

    other_mem = GetProcessMem.new(pid)
    assert @mem.kb > other_mem.kb
  ensure
    Process.kill("TERM", pid) if pid
    Process.wait(pid) if pid
  end

  def test_invalid_pid
    raised_exception = false
    GetProcessMem.new("ls")
  rescue ArgumentError
    raised_exception = true
  ensure
    assert raised_exception
  end

  def test_seems_to_work
    assert @mem.kb > 0
    assert @mem.mb > 0
    assert @mem.gb > 0
    assert @mem.bytes > 0
  end

  def test_linux_smap
    delta = 1
    bytes = @mem.linux_memory(fixture_path("heroku-bash-smap"))
    assert_in_delta BigDecimal("2122240.0"), bytes, delta
  end

  def test_linux_status
    bytes = @mem.linux_status_memory(fixture_path("heroku-bash-status"))
    assert_equal bytes, 2080768
  end

  def test_conversions
    bytes = 0
    delta = BigDecimal("0.0000001")
    assert_in_delta 0.0, @mem.kb(bytes), delta
    assert_in_delta 0.0, @mem.mb(bytes), delta
    assert_in_delta 0.0, @mem.gb(bytes), delta

    # kb
    bytes = 1024
    assert_in_delta 1.0, @mem.kb(bytes), delta
    assert_in_delta 0.0009765625, @mem.mb(bytes), delta
    assert_in_delta 9.5367431640625e-07, @mem.gb(bytes), delta

    # mb
    bytes = 1_048_576
    assert_in_delta 1024.0, @mem.kb(bytes), delta
    assert_in_delta 1.0, @mem.mb(bytes), delta
    assert_in_delta 0.0009765625, @mem.gb(bytes), delta

    # gb
    bytes = 1_073_741_824
    assert_in_delta 1048576.0, @mem.kb(bytes), delta
    assert_in_delta 1024.0, @mem.mb(bytes), delta
    assert_in_delta 1.0, @mem.gb(bytes), delta
  end
end