1 files changed, 99 insertions, 0 deletions
diff --git a/test/mgmt_auto_adjust.rb b/test/mgmt_auto_adjust.rb
new file mode 100644
index 0000000..f2e01d6
--- /dev/null
+++ b/test/mgmt_auto_adjust.rb
@@ -0,0 +1,99 @@
+#!/usr/bin/env ruby
+# -*- encoding: binary -*-
+# Copyright (C) 2013, Eric Wong <normalperson@yhbt.net>
+# License: GPLv3 or later (see COPYING for details)
+require 'test/test_helper'
+
+class TestMgmtAutoAdjust < Test::Unit::TestCase
+  def setup
+    @tmpdir = Dir.mktmpdir('cmogstored-mgmt-test')
+    @to_close = []
+    @host = TEST_HOST
+    srv = TCPServer.new(@host, 0)
+    @port = srv.addr[1]
+    srv.close
+
+    http = TCPServer.new(@host, 0)
+    @httpport = http.addr[1]
+    http.close
+
+    @err = Tempfile.new("stderr")
+    Dir.mkdir("#@tmpdir/dev666")
+    cmd = [ "cmogstored", "--docroot=#@tmpdir", "--mgmtlisten=#@host:#@port",
+            "--httplisten=#@host:#@httpport",
+            "--maxconns=500" ]
+    vg = ENV["VALGRIND"] and cmd = vg.split(/\s+/).concat(cmd)
+    @pid = fork {
+      ENV["MOG_DISK_USAGE_INTERVAL"] = "1"
+      $stderr.reopen(@err)
+      @err.close
+      exec(*cmd)
+    }
+    @client = get_client
+  end
+
+  def teardown
+    Process.kill(:QUIT, @pid) rescue nil
+    _, status = Process.waitpid2(@pid)
+    @to_close.each { |io| io.close unless io.closed? }
+    FileUtils.rm_rf(@tmpdir)
+    @err.rewind
+    # $stderr.write(@err.read)
+    assert status.success?, status.inspect
+  end
+
+  def wait_for_seen_dev
+    @client.write "watch\r\n"
+    seen_dev = false
+    while line = @client.gets
+      line.strip!
+      case line
+      when /^666\t/
+        seen_dev = true
+      when "."
+        break if seen_dev
+      end
+    end
+  end
+
+  def test_aio_threads_auto_adjust
+    wait_for_seen_dev
+    t_yield # wait for threads to spawn
+    taskdir = "/proc/#@pid/task"
+    glob = "#{taskdir}/*"
+    prev_threads = Dir[glob].size if File.directory?(taskdir)
+    if RUBY_PLATFORM =~ /linux/
+      assert File.directory?(taskdir), "/proc not mounted on Linux?"
+    end
+
+    File.directory?(taskdir) or return
+    Dir.mkdir("#@tmpdir/dev333")
+    stop = Time.now + 10
+    while prev_threads == Dir[glob].size && Time.now < stop
+      sleep(0.1)
+    end
+    cur_threads = Dir[glob].size
+    assert_operator cur_threads, :>, prev_threads,
+           "prev_threads=#{cur_threads} > prev_threads=#{prev_threads}"
+    assert_match(%r{updating server aio_threads=20}, File.read(@err.path))
+  end
+
+  def test_aio_threads_auto_adjust_warn
+    @client.write "server aio_threads = 1\r\n"
+    assert_equal "\r\n", @client.gets
+    stop = Time.now + 60
+    expect = "server aio_threads=1"
+    while (! File.read(@err.path).include?(expect) && Time.now < stop)
+      sleep 0.1
+    end
+    assert(File.read(@err.path).include?(expect), File.read(@err.path))
+
+    Dir.mkdir("#@tmpdir/dev333")
+    warning = "fewer aio_threads(1) than devices(2)"
+    stop = Time.now + 10
+    while (! File.read(@err.path).include?(warning)) && Time.now < stop
+      sleep 0.1
+    end
+    assert(File.read(@err.path).include?(warning), File.read(@err.path))
+  end
+end