ruby-dtrace 0.0.2 → 0.0.3

data/test/test_dtrace.rb

@@ -6,6 +6,8 @@
 require 'dtrace'
 require 'test/unit'
 
+# Tests for the Dtrace handle class
+
 class TestDtrace < Test::Unit::TestCase
   def test_dtrace
     t = Dtrace.new

data/test/test_dtrace_aggregates.rb

@@ -0,0 +1,56 @@
+#
+# Ruby-Dtrace
+# (c) 2007 Chris Andrews <chris@nodnol.org>
+#
+
+require 'dtrace'
+require 'test/unit'
+
+# Tests using the DTrace profile provider.
+
+class TestDtraceAggregates < Test::Unit::TestCase
+  def test_aggregate_group
+    t = Dtrace.new 
+    t.setopt("bufsize", "4m")
+    t.setopt("aggsize", "4m")
+
+    progtext =<<EOD
+profile-1000
+{ 
+  @a[execname] = count();
+  @b[execname] = count();
+}
+
+profile-10
+{
+  printa(@a);
+  printa(@b);
+}
+EOD
+    
+    prog = t.compile progtext
+    prog.execute
+    t.go
+
+    sleep 1
+
+    c = DtraceConsumer.new(t)
+    assert c
+
+    data = []
+    c.consume_once do |d|
+      data << d
+    end
+    
+    assert data.length > 0
+    data.each do |d|
+      assert d
+      assert_equal DtraceData, d.class
+      d.data.each do |agg|
+        assert_equal DtraceAggregateSet, agg.class
+      end
+    end
+
+  end
+
+end

data/test/test_dtrace_processes.rb

@@ -0,0 +1,83 @@
+#
+# Ruby-Dtrace
+# (c) 2007 Chris Andrews <chris@nodnol.org>
+#
+
+require 'dtrace'
+require 'test/unit'
+
+# Tests for creating and grabbing processes.
+
+class TestDtraceProcesses < Test::Unit::TestCase
+
+  def test_createprocess
+
+    t = Dtrace.new 
+    t.setopt("bufsize", "8m")
+    t.setopt("aggsize", "4m")
+    t.setopt("stackframes", "5")
+    t.setopt("strsize", "131072")
+
+    progtext = <<EOD
+pid$target:*::entry,
+pid$target:*::return
+{
+  trace(pid);
+}
+EOD
+
+    p = t.createprocess([ '/usr/bin/true' ])
+    prog = t.compile(progtext)
+    prog.execute
+    t.go
+    p.continue
+
+    i = 0
+    c = DtraceConsumer.new(t)
+    c.consume do |d|
+      assert d
+      assert_equal "pid#{d.data[0].value}", d.probe.provider
+      i = i + 1
+      if i > 10
+        c.finish
+      end
+    end
+    assert i > 0
+  end
+
+  def test_grabprocess
+
+    t = Dtrace.new 
+    t.setopt("bufsize", "8m")
+    t.setopt("aggsize", "4m")
+    t.setopt("stackframes", "5")
+    t.setopt("strsize", "131072")
+
+    progtext = <<EOD
+pid$target:*::entry,
+pid$target:*::return
+{
+
+}
+EOD
+
+    pid = Kernel.fork { (0..9).each do sleep 1 end }
+    p = t.grabprocess(pid)
+    prog = t.compile(progtext)
+    prog.execute
+
+    t.go
+    p.continue
+
+    sleep 3
+
+    i = 0
+    c = DtraceConsumer.new(t)
+    c.consume_once do |d|
+      assert d
+      assert_equal "pid#{pid}", d.probe.provider
+      i = i + 1
+    end
+    assert i > 0
+  end
+end

data/test/test_dtrace_profile.rb

@@ -0,0 +1,232 @@
+#
+# Ruby-Dtrace
+# (c) 2007 Chris Andrews <chris@nodnol.org>
+#
+
+require 'dtrace'
+require 'test/unit'
+
+# Tests using the DTrace profile provider.
+
+class TestDtraceProfile < Test::Unit::TestCase
+  
+  def test_dprogram_run
+    t = Dtrace.new 
+    t.setopt("bufsize", "4m")
+    t.setopt("aggsize", "4m")
+
+    progtext = 'profile:::profile-1 { trace("foo"); }'
+    
+    prog = t.compile progtext
+    prog.execute
+    t.go
+    sleep 2
+
+    c = DtraceConsumer.new(t)
+    assert c
+
+    i = 0
+    c.consume do |d|
+      assert d
+      assert_equal "profile:::profile-1", d.probe.to_s
+      assert_not_nil d.cpu
+
+      d.data.each do |r|
+        assert_equal r.value, "foo"
+      end
+
+      i = i + 1
+      if i > 10
+        c.finish
+      end
+    end
+    assert i > 0
+  end
+    
+  def test_dprogram_aggregate
+    t = Dtrace.new 
+    t.setopt("bufsize", "4m")
+    t.setopt("aggsize", "4m")
+
+    progtext = <<EOD
+profile-1000
+{ 
+  @a[execname] = count(); 
+}
+
+profile-10
+{
+  printa(@a)
+}
+EOD
+
+    prog = t.compile progtext
+    prog.execute
+    t.go
+    sleep 2
+
+    c = DtraceConsumer.new(t)
+
+    i = 0
+    c.consume do |d|
+      assert d
+      assert_not_nil d.cpu
+      assert_equal "profile:::profile-10", d.probe.to_s
+
+      d.data.each do |r|
+        assert_equal DtraceAggregateSet, r.class
+        r.data.each do |a|
+          assert_not_nil a.value
+          assert_not_nil a.tuple
+          assert_equal 1, a.tuple.length
+        end
+      end
+
+      i = i + 1
+      if i >= 10
+        c.finish
+      end
+    end
+    assert i > 0
+  end
+
+  def test_dprogram_printf
+    t = Dtrace.new 
+    t.setopt("bufsize", "4m")
+    t.setopt("aggsize", "4m")
+
+    progtext = <<EOD
+profile-1
+{ 
+  printf("execname: %s %s", execname, "foo")
+}
+EOD
+
+    prog = t.compile progtext
+    prog.execute
+    t.go
+    sleep 2
+    
+    c = DtraceConsumer.new(t)
+
+    i = 0
+    c.consume do |d|
+      assert d
+      assert_not_nil d.cpu
+      assert_equal "profile:::profile-1", d.probe.to_s
+
+      i = i + 1
+      if i >= 10
+        c.finish
+      end
+    end
+    assert i > 0
+  end
+
+  def test_dprogram_aggregate_once
+    t = Dtrace.new 
+    t.setopt("bufsize", "4m")
+    t.setopt("aggsize", "4m")
+
+    progtext = <<EOD
+profile-1000hz
+{ 
+  @a[execname] = count(); 
+}
+
+END
+{
+  printa(@a)
+}
+EOD
+
+    prog = t.compile progtext
+    prog.execute
+    t.go
+    sleep 2
+    
+    i = 0
+    c = DtraceConsumer.new(t)
+    c.consume_once do |d|
+      i = i + 1
+      assert d
+      assert_not_nil d.cpu
+      assert_equal "dtrace:::END", d.probe.to_s
+      
+      d.data.each do |r|
+        assert_equal DtraceAggregateSet, r.class
+        r.data.each do |a|
+          assert_not_nil a.value
+          assert_not_nil a.tuple
+          assert_equal 1, a.tuple.length
+        end
+      end
+    end
+    assert i > 0    
+  end
+  
+  def test_stack
+    t = Dtrace.new 
+    t.setopt("bufsize", "8m")
+    t.setopt("aggsize", "4m")
+    t.setopt("stackframes", "5")
+    t.setopt("strsize", "131072")
+
+    progtext = "profile-1 { trace(execname); stack(); }"
+    prog = t.compile progtext
+    prog.execute
+    t.go
+    sleep 2
+
+    c = DtraceConsumer.new(t)
+    i = 0
+    c.consume do |d|
+      assert d
+      assert_not_nil d.cpu
+      assert_equal "profile:::profile-1", d.probe.to_s
+
+      assert_equal 2, d.data.length
+      assert_equal DtraceRecord, d.data[0].class
+      assert_equal DtraceStackRecord, d.data[1].class
+      
+      i = i + 1
+      if i > 10
+        c.finish
+      end
+    end
+    assert i > 0
+  end
+
+  def test_ustack
+    t = Dtrace.new 
+    t.setopt("bufsize", "8m")
+    t.setopt("aggsize", "4m")
+    t.setopt("stackframes", "5")
+    t.setopt("strsize", "131072")
+
+    progtext = "profile-1 { trace(execname); ustack(); }"
+    prog = t.compile progtext
+    prog.execute
+    t.go
+    sleep 2
+
+    c = DtraceConsumer.new(t)
+    i = 0
+    c.consume do |d|
+      assert d
+      assert_not_nil d.cpu
+      assert_equal "profile:::profile-1", d.probe.to_s
+
+      assert_equal 2, d.data.length
+      assert_equal DtraceRecord, d.data[0].class
+      assert_equal DtraceStackRecord, d.data[1].class
+
+      i = i + 1
+      if i > 10
+        c.finish
+      end
+    end
+    assert i > 0
+  end
+
+end  

data/test/test_dtrace_repeat.rb

@@ -0,0 +1,51 @@
+#
+# Ruby-Dtrace
+# (c) 2007 Chris Andrews <chris@nodnol.org>
+#
+
+require 'dtrace'
+require 'test/unit'
+
+# Test repeatedly using DTrace in the same process.  Show that we can
+# reopen the DTrace handle multiple times, without explictly closing
+# it (that happens in GC, so in this script that's probably right at
+# the end).
+
+class TestDtraceRepeat < Test::Unit::TestCase
+  
+  def test_repeats
+    (0..9).each do |i|
+      t = Dtrace.new 
+      t.setopt("bufsize", "4m")
+      t.setopt("aggsize", "4m")
+
+      progtext = 'syscall:::entry { trace("foo"); }'
+      
+      prog = t.compile progtext
+      prog.execute
+      t.go
+      
+      # Let some activity happen.
+      sleep 1
+
+      c = DtraceConsumer.new(t)
+      assert c
+
+      i = 0
+      c.consume do |d|
+        assert d
+        assert_not_nil d.cpu
+        assert_equal 'syscall', d.probe.provider
+        assert_not_nil d.probe.func
+        assert_equal 'entry', d.probe.name
+
+        d.data.each do |r|
+          assert_equal 'foo', r.value
+        end
+        c.finish
+        i = i + 1
+      end
+      assert i > 0
+    end
+  end
+end

data/test/test_dtrace_rubyprobe.rb

@@ -0,0 +1,52 @@
+#
+# Ruby-Dtrace
+# (c) 2007 Chris Andrews <chris@nodnol.org>
+#
+
+require 'dtrace'
+require 'test/unit'
+
+class TestDtrace < Test::Unit::TestCase
+  def test_rubyprobe
+    t = Dtrace.new 
+    t.setopt("bufsize", "4m")
+
+    progtext = <<EOD
+ruby*:::ruby-probe
+{
+  trace(copyinstr(arg0));
+  trace(copyinstr(arg1));
+}
+EOD
+    
+    prog = t.compile progtext
+    prog.execute
+    t.go
+
+    c = DtraceConsumer.new(t)
+
+    # Leopard's ruby-probe is DTracer, Solaris's is Tracer.
+    begin
+      trace_module = DTracer
+    rescue NameError
+      trace_module = Tracer
+    end
+
+    (0..9).each do |i|
+      trace_module.fire("foo", i.to_s)
+    end
+
+    data = []
+    c.consume_once do |d|
+      data << d
+    end
+    
+    (0..9).each do |i|
+      d = data.shift
+      assert_equal("foo", d.data[0].value)
+      assert_equal(i.to_s, d.data[1].value)
+    end
+
+  end
+    
+end