wee 0.1.0 → 0.3.1

data/lib/wee/{snapshot.rb → snapshot_ext.rb}

@@ -7,7 +7,7 @@ class Object
     snap
   end
 
-  def apply_snapshot(snap)
+  def restore_snapshot(snap)
     instance_variables.each do |iv|
       instance_variable_set(iv, snap[iv])
     end
@@ -19,7 +19,7 @@ class Array
     dup
   end
 
-  def apply_snapshot(snap)
+  def restore_snapshot(snap)
     replace(snap)
   end
 end
@@ -29,7 +29,7 @@ class String
     dup
   end
 
-  def apply_snapshot(snap)
+  def restore_snapshot(snap)
     replace(snap)
   end
 end
@@ -41,7 +41,7 @@ class Struct
     snap
   end
 
-  def apply_snapshot(snap)
+  def restore_snapshot(snap)
     snap.each_pair {|k,v| send(k.to_s + "=", v)} 
   end
 end

data/test/components/calltest.rb

@@ -0,0 +1,16 @@
+class CallTest < Wee::Component
+  def msgbox(msg)
+    call MessageBox.new(msg)
+  end
+
+  def render
+    r.anchor.callback {
+      if msgbox('A')
+        msgbox('B')
+      else
+        msgbox('C')
+        msgbox('D')
+      end
+    }.with("show")
+  end
+end

data/test/components/counter.rb

@@ -0,0 +1,17 @@
+class Counter < Wee::Component
+  def initialize(cnt=0)
+    super()
+    @cnt = ValueHolder.new(cnt) 
+  end
+
+  def backtrack_state(snap)
+    super
+    snap.add(@cnt)
+  end
+
+  def render
+    r.h1(@cnt.value.to_s)
+    r.anchor.callback { @cnt.value -= 1 }.with("--")
+    r.anchor.callback { @cnt.value += 1 }.with("++")
+  end
+end

data/test/components/messagebox.rb

@@ -0,0 +1,15 @@
+class MessageBox < Wee::Component
+  def initialize(text)
+    super()
+    @text = text 
+  end
+
+  def render
+    r.bold(@text)
+    r.form do 
+      r.submit_button.value('OK').callback { answer true }
+      r.space
+      r.submit_button.value('Cancel').callback { answer false }
+    end
+  end
+end

data/test/components/page.rb

@@ -0,0 +1,14 @@
+class Page < Wee::Component
+  def initialize(*children)
+    super()
+    self.children.push(*children)
+  end
+
+  def render
+    r.page.title('').with {
+      self.children.each do |child|
+        r.render(child)
+      end
+    }
+  end
+end

data/test/components/page_decoration.rb

@@ -0,0 +1,7 @@
+class PageDecoration < Wee::Decoration
+  def do_render(rctx)
+    with_renderer_for(rctx) do
+      r.page.title('').with { super }
+    end
+  end
+end

data/test/stress.rb

@@ -0,0 +1,64 @@
+$LOAD_PATH.unshift '../lib'
+require 'wee'
+require 'wee/utils/cache'
+require 'wee/adaptors/webrick'
+require 'utils/webrick_background'
+require 'utils/memory_plotter'
+require 'utils/object_plotter'
+require 'utils/cross'
+
+require 'rubygems'
+require 'web/unit'   # require narf-lib
+
+require 'components/messagebox'
+require 'components/calltest'
+require 'components/page'
+
+NUM_SESSIONS = 5
+
+class DummyLog < WEBrick::BasicLog
+  def initialize() super(self) end
+  def <<(*args) end
+end
+
+class MySession < Wee::Session
+  def initialize
+    super do
+      self.root_component = Page.new(CallTest.new)
+      self.page_store = Wee::Utils::LRUCache.new(10) # backtrack up to 10 pages
+    end
+  end
+end
+
+app = Wee::Application.new {|app|
+  app.default_request_handler { MySession.new }
+  app.id_generator = Wee::SimpleIdGenerator.new(rand(1_000_000))
+}
+
+Wee::WEBrickAdaptor.register('/app' => app).start(:Logger => DummyLog.new, :AccessLog => [])
+
+MemoryPlotter.new(5, $$).run
+ObjectPlotter.new(5, Object, Array, String, Bignum).run
+ObjectPlotter.new(5, Thread, Continuation, Proc).run
+
+$URLBASE = 'http://localhost:2000'
+
+class StressSession
+  def initialize
+    @r = Web::Unit::Response.get('/app').redirect.redirect
+  end
+
+  def step
+    %w(OK Cancel).each {|b|
+      @r = @r.click('show').redirect.submit('OK').redirect.submit(b).redirect
+    }
+    (%w(OK Cancel) ** %w(OK Cancel)).each { |b1, b2|
+      @r = @r.click('show').redirect.submit('Cancel').redirect.submit(b1).redirect.submit(b2).redirect
+    }
+  end
+end
+
+sessions = (1..NUM_SESSIONS).map { StressSession.new }
+loop do
+  sessions.each {|s| s.step }
+end

data/test/test_component.rb

@@ -0,0 +1,106 @@
+require 'test/unit'
+module Wee; end
+require 'wee/core'
+
+class Test_Component < Test::Unit::TestCase
+  def test_add_remove_one_decoration
+    c = Wee::Component.new
+    d = Wee::Decoration.new
+
+    assert_same c, c.decoration
+    assert_nil d.owner
+
+    c.add_decoration(d)
+
+    assert_same d, c.decoration
+    assert_same c, d.owner
+    assert_same d, d.owner.decoration
+
+    assert_same d, c.remove_decoration(d)
+
+    assert_same c, c.decoration  
+    assert_nil d.owner
+  end
+
+  def test_add_remove_multiple_decorations
+    c = Wee::Component.new
+    d1 = Wee::Decoration.new
+    d2 = Wee::Decoration.new
+    d3 = Wee::Decoration.new
+
+    c.add_decoration(d1)
+    c.add_decoration(d2)
+    c.add_decoration(d3)
+
+    assert_same d3, c.decoration 
+    assert_same d2, d3.owner 
+    assert_same d1, d2.owner
+    assert_same c, d1.owner
+
+    assert_same d2, c.remove_decoration(d2)
+
+    assert_same d3, c.decoration 
+    assert_same d1, d3.owner 
+    assert_nil  d2.owner  
+    assert_same c, d1.owner
+
+    assert_same d1, c.remove_decoration(d1)
+    assert_same d3, c.decoration
+    assert_same c, d3.owner   
+    assert_nil  d1.owner
+
+    # try to remove an already removed decoration 
+    assert_nil c.remove_decoration(d2)
+    assert_nil c.remove_decoration(d1)  
+
+    assert_same d3, c.remove_decoration(d3)
+    assert_same c, c.decoration
+    assert_nil  d3.owner
+  end
+
+  def test_each_decoration
+    require 'enumerator'
+    c = Wee::Component.new
+    d1 = Wee::Decoration.new
+    d2 = Wee::Decoration.new
+    d3 = Wee::Decoration.new
+
+    c.add_decoration(d1)
+    c.add_decoration(d2)
+    c.add_decoration(d3)
+
+    assert_equal [d3, d2, d1], c.to_enum(:each_decoration).to_a
+  end
+
+  def test_remove_decoration_if
+    c = Wee::Component.new
+    d1 = Wee::Decoration.new
+    d2 = Wee::Decoration.new
+    d3 = Wee::Decoration.new
+    d4 = Wee::Decoration.new
+
+    c.add_decoration(d1)
+    c.add_decoration(d2)
+    c.add_decoration(d3)
+    c.add_decoration(d4)
+
+    def d1.a() end
+    def d3.a() end
+    c.remove_decoration_if {|d| d.respond_to?(:a)}
+    assert_equal [d4, d2], c.to_enum(:each_decoration).to_a
+
+    def d4.a() end
+    c.remove_decoration_if {|d| d.respond_to?(:a)}
+    assert_equal [d2], c.to_enum(:each_decoration).to_a
+
+    c.remove_decoration_if {|d| d.respond_to?(:a)}
+    assert_equal [d2], c.to_enum(:each_decoration).to_a
+
+    def d2.a() end
+    c.remove_decoration_if {|d| d.respond_to?(:a)}
+    assert_equal [], c.to_enum(:each_decoration).to_a
+
+    assert_equal c, c.decoration
+  end
+
+end

data/test/test_html_canvas.rb

@@ -0,0 +1,25 @@
+require 'test/unit'
+module Wee; end
+require 'wee/renderer/html/writer'
+require 'wee/renderer/html/brushes'
+require 'wee/renderer/html/canvas'
+require 'wee/context'
+
+class Test_HtmlCanvas < Test::Unit::TestCase
+  def test_simple
+    rctx = Wee::RenderingContext.new
+    rctx.document = Wee::HtmlWriter.new(doc='')
+
+    c = Wee::HtmlCanvas.new(rctx)
+    c.form.action("foo").with {
+      c.table {
+        c.table_row.id("myrow").with {
+          c.table_data.align_top.with("Hello world")
+        }
+      }
+      c.space
+    }
+
+    assert_equal %[<form action="foo" method="POST"><table><tr id="myrow"><td align="top">Hello world</td></tr></table>&nbsp;</form>], doc
+  end
+end

data/test/test_html_writer.rb

@@ -0,0 +1,27 @@
+require 'test/unit'
+module Wee; end
+require 'wee/renderer/html/writer'
+
+class Test_HtmlWriter < Test::Unit::TestCase
+  def test_document
+    w = Wee::HtmlWriter.new(doc='')
+    w.start_tag('html')
+    w.start_tag('body')
+    w.start_tag('a', 'href' => 'http://...')
+    w.text('link')
+    w.end_tag('a')
+    w.end_tag('body')
+    w.end_tag('html')
+
+    assert_equal true, w.valid?
+    assert_equal '<html><body><a href="http://...">link</a></body></html>', doc
+  end
+
+  def test_merge_start_end_tag
+    w = Wee::HtmlWriter.new(doc='')
+    w.start_tag('a', 'href' => '')
+    w.end_tag('a')
+    assert_equal true, w.valid?
+    assert_equal '<a href=""/>', doc
+  end
+end

data/test/test_request.rb

@@ -0,0 +1,13 @@
+require 'test/unit'
+module Wee; end
+require 'wee/request'
+
+class Test_Request < Test::Unit::TestCase
+  def test_parse
+    req = Wee::Request.new('/app', '/app/...@req_handler_id/page_id', nil, nil)
+    assert_equal 'req_handler_id', req.request_handler_id
+    assert_equal 'page_id', req.page_id
+    assert_equal '/app/@a/b?c', req.build_url('a', 'b', 'c')
+    assert_equal '/app/@req_handler_id/b', req.build_url('req_handler_id', 'b')
+  end
+end

data/test/utils/cross.rb

@@ -0,0 +1,65 @@
+#
+# Cross product for Enumerable's
+#
+
+# TODO: use ** for [1,2,3].cross([4,5,6])
+
+require 'generator'
+
+module Enumerable
+  def **(otherEnumerable)
+    res = Array.new
+    self.each do |i|
+      otherEnumerable.each do |j|
+        res << [i,j]
+      end
+    end
+    return res
+  end
+
+  def cross(*enums, &block)
+    Enumerable.cross(self, *enums, &block)
+  end
+
+  def self.cross(*enums, &block)
+    raise if enums.empty?
+    gens = enums.map{|e| Generator.new(e)}
+    return [] if gens.any? {|g| g.end?}
+    sz = gens.size
+    res = []
+    tuple = Array.new(sz)
+
+    loop do
+      # fill tuple
+      (0 ... sz).each { |i| 
+        tuple[i] = gens[i].current 
+      }
+      if block.nil?
+        res << tuple.dup
+      else
+        block.call(tuple.dup)
+      end
+
+      # step forward
+      gens[-1].next
+      (sz-1).downto(0) { |i|
+        if gens[i].end?
+          if i > 0
+            gens[i].rewind
+            gens[i-1].next
+          else
+            return res
+          end
+        end
+      }
+    end
+  end
+end
+
+if __FILE__ == $0
+  p Enumerable.cross([1,2,3], [4], ["apple", "banana"])
+
+  [1,2,3].cross([4,5,6]) {|elem| p elem }
+
+  p Enumerable.cross([1,2], [3,4], [5,6], [7,8])
+end

data/test/utils/generic_plotter.rb

@@ -0,0 +1,28 @@
+require 'utils/gnuplot'
+
+class GenericPlotter
+  def initialize(interval, dataset_configs)
+    @interval = interval
+    @datasets = dataset_configs
+    @datasets.each_with_index {|cfg, i|
+      cfg[:params] ||= 'with lines'
+      cfg[:title] ||= i.to_s
+      cfg[:data] ||= []
+    }
+    @gnuplot = GnuPlot.spawn
+  end
+
+  def run
+    Thread.start {
+      @time = 0
+      loop do
+        @datasets.each do |cfg|
+          cfg[:proc].call(cfg[:data], @time) 
+        end
+        @gnuplot.plot(@datasets)
+        sleep @interval
+        @time += @interval
+      end
+    }
+  end
+end