+ Penlight

1 files changed, 207 insertions, 0 deletions

diff --git a/Data/Libraries/Penlight/tests/test-class.lua b/Data/Libraries/Penlight/tests/test-class.lua
new file mode 100644
index 0000000..8621005
--- /dev/null
+++ b/Data/Libraries/Penlight/tests/test-class.lua
@@ -0,0 +1,207 @@
+local class = require 'pl.class'
+local test = require 'pl.test'
+asserteq = test.asserteq
+T = test.tuple
+
+A = class()
+
+function A:_init ()
+    self.a = 1
+end
+
+-- calling base class' ctor automatically
+A1 = class(A)
+
+asserteq(A1(),{a=1})
+
+-- explicitly calling base ctor with super
+
+B = class(A)
+
+function B:_init ()
+    self:super()
+    self.b = 2
+end
+
+function B:foo ()
+    self.eee = 1
+end
+
+function B:foo2 ()
+    self.g = 8
+end
+
+asserteq(B(),{a=1,b=2})
+
+-- can continue this chain
+
+C = class(B)
+
+function C:_init ()
+    self:super()
+    self.c = 3
+end
+
+function C:foo ()
+    -- recommended way to call inherited version of method...
+    B.foo(self)
+end
+
+c = C()
+c:foo()
+
+asserteq(c,{a=1,b=2,c=3,eee=1})
+
+-- test indirect inherit
+
+D = class(C)
+
+E = class(D)
+
+function E:_init ()
+    self:super()
+    self.e = 4
+end
+
+function E:foo ()
+    -- recommended way to call inherited version of method...
+    self.eeee = 5
+    C.foo(self)
+end
+
+F = class(E)
+
+function F:_init ()
+    self:super()
+    self.f = 6
+end
+
+f = F()
+f:foo()
+f:foo2() -- Test : invocation inherits this function from all the way up in B
+
+asserteq(f,{a=1,b=2,c=3,eee=1,e=4,eeee=5,f=6,g=8})
+
+-- Test that inappropriate calls to super() fail gracefully
+
+G = class() -- Class with no init
+
+H = class(G) -- Class with an init that wrongly calls super()
+
+function H:_init()
+    self:super() -- Notice: G has no _init
+end
+
+I = class(H) -- Inherits the init with a bad super
+J = class(I) -- Grandparent-inits the init with a bad super
+
+K = class(J) -- Has an init, which calls the init with a bad super
+
+function K:_init()
+    self:super()
+end
+
+local function createG()
+    return G()
+end
+
+local function createH() -- Wrapper function for pcall
+    return H()
+end
+
+local function createJ()
+    return J()
+end
+
+local function createK()
+    return K()
+end
+
+assert(pcall(createG)) -- Should succeed
+assert(not pcall(createH)) -- These three should fail
+assert(not pcall(createJ))
+assert(not pcall(createK))
+
+--- class methods!
+assert(c:is_a(C))
+assert(c:is_a(B))
+assert(c:is_a(A))
+assert(c:is_a() == C)
+assert(C:class_of(c))
+assert(B:class_of(c))
+assert(A:class_of(c))
+
+--- metamethods!
+
+function C:__tostring ()
+    return ("%d:%d:%d"):format(self.a,self.b,self.c)
+end
+
+function C.__eq (c1,c2)
+    return c1.a == c2.a and c1.b == c2.b and c1.c == c2.c
+end
+
+asserteq(C(),{a=1,b=2,c=3})
+
+asserteq(tostring(C()),"1:2:3")
+
+asserteq(C()==C(),true)
+
+----- properties -----
+
+local MyProps = class(class.properties)
+local setted_a, got_b
+
+function MyProps:_init ()
+    self._a = 1
+    self._b = 2
+end
+
+function MyProps:set_a (v)
+    setted_a = true
+    self._a = v
+end
+
+function MyProps:get_b ()
+    got_b = true
+    return self._b
+end
+
+function MyProps:set (a,b)
+    self._a = a
+    self._b = b
+end
+
+local mp = MyProps()
+
+mp.a = 10
+
+asserteq(mp.a,10)
+asserteq(mp.b,2)
+asserteq(setted_a and got_b, true)
+
+class.MoreProps(MyProps)
+local setted_c
+
+function MoreProps:_init()
+    self:super()
+    self._c = 3
+end
+
+function MoreProps:set_c (c)
+    setted_c = true
+    self._c = c
+end
+
+mm = MoreProps()
+
+mm:set(10,20)
+mm.c = 30
+
+asserteq(setted_c, true)
+asserteq(T(mm.a, mm.b, mm.c),T(10,20,30))
+
+
+
+
+