middleclass/spec/metamethods_spec.lua

local class = require 'middleclass'
local Object = class.Object

local function is_lua_5_2_compatible()
  return type(rawlen) == 'function'
end

local function is_lua_5_3_compatible()
  return type(string.unpack) == 'function'
end

if is_lua_5_2_compatible() then
  require 'spec/metamethods_lua_5_2'
end

if is_lua_5_3_compatible() then
  require 'spec.metamethods_lua_5_3'
end

describe('Metamethods', function()
  describe('Custom Metamethods', function()
    local Vector, a, b
    before_each(function()
      Vector= class('Vector')
      function Vector.initialize(a,x,y,z) a.x, a.y, a.z = x,y,z end
      function Vector.__tostring(a) return a.class.name .. '[' .. a.x .. ',' .. a.y .. ',' .. a.z .. ']' end
      function Vector.__eq(a,b)     return a.x==b.x and a.y==b.y and a.z==b.z end
      function Vector.__lt(a,b)     return a() < b() end
      function Vector.__le(a,b)     return a() <= b() end
      function Vector.__add(a,b)    return a.class:new(a.x+b.x, a.y+b.y ,a.z+b.z) end
      function Vector.__sub(a,b)    return a.class:new(a.x-b.x, a.y-b.y, a.z-b.z) end
      function Vector.__div(a,s)    return a.class:new(a.x/s, a.y/s, a.z/s) end
      function Vector.__unm(a)      return a.class:new(-a.x, -a.y, -a.z) end
      function Vector.__concat(a,b) return a.x*b.x+a.y*b.y+a.z*b.z end
      function Vector.__call(a)     return math.sqrt(a.x*a.x+a.y*a.y+a.z*a.z) end
      function Vector.__pow(a,b)
        return a.class:new(a.y*b.z-a.z*b.y,a.z*b.x-a.x*b.z,a.x*b.y-a.y*b.x)
      end
      function Vector.__mul(a,b)
        if type(b)=="number" then return a.class:new(a.x*b, a.y*b, a.z*b) end
        if type(a)=="number" then return b.class:new(a*b.x, a*b.y, a*b.z) end
      end

      a = Vector:new(1,2,3)
      b = Vector:new(2,4,6)
    end)

    it('implements __tostring', function()
      assert.equal(tostring(a), "Vector[1,2,3]")
    end)

    it('implements __eq', function()
      assert.equal(a, a)
    end)

    it('implements __lt', function()
      assert.is_true(a < b)
    end)

    it('implements __le', function()
      assert.is_true(a <= b)
    end)

    it('implements __add', function()
      assert.equal(a+b, Vector(3,6,9))
    end)

    it('implements __sub', function()
      assert.equal(b-a, Vector(1,2,3))
    end)

    it('implements __div', function()
      assert.equal(b/2, Vector(1,2,3))
    end)

    it('implements __concat', function()
      assert.equal(a..b, 28)
    end)

    it('implements __call', function()
      assert.equal(a(), math.sqrt(14))
    end)

    it('implements __pow', function()
      assert.equal(a^b, Vector(0,0,0))
    end)

    it('implements __mul', function()
      assert.equal(4*a, Vector(4,8,12))
    end)

    --[[
    it('implements __index', function()
      assert.equal(b[1], 3)
    end)
    --]]

    describe('Inherited Metamethods', function()
      local Vector2, c, d
      before_each(function()
        Vector2= class('Vector2', Vector)
        function Vector2:initialize(x,y,z) Vector.initialize(self,x,y,z) end

        c = Vector2:new(1,2,3)
        d = Vector2:new(2,4,6)
      end)

      it('implements __tostring', function()
        assert.equal(tostring(c), "Vector2[1,2,3]")
      end)

      it('implements __eq', function()
        assert.equal(c, c)
      end)

      it('implements __lt', function()
        assert.is_true(c < d)
      end)

      it('implements __le', function()
        assert.is_true(c <= d)
      end)

      it('implements __add', function()
        assert.equal(c+d, Vector2(3,6,9))
      end)

      it('implements __sub', function()
        assert.equal(d-c, Vector2(1,2,3))
      end)

      it('implements __div', function()
        assert.equal(d/2, Vector2(1,2,3))
      end)

      it('implements __concat', function()
        assert.equal(c..d, 28)
      end)

      it('implements __call', function()
        assert.equal(c(), math.sqrt(14))
      end)

      it('implements __pow', function()
        assert.equal(c^d, Vector2(0,0,0))
      end)

      it('implements __mul', function()
        assert.equal(4*c, Vector2(4,8,12))
      end)

      describe('Updates', function()
        it('overrides __add', function()
          Vector2.__add = function(a, b) return Vector.__add(a, b)/2 end
          assert.equal(c+d, Vector2(1.5,3,4.5))
        end)

        it('updates __add', function()
          Vector.__add = Vector.__sub
          assert.equal(c+d, Vector2(-1,-2,-3))
        end)

        it('does not update __add after overriding', function()
          Vector2.__add = function(a, b) return Vector.__add(a, b)/2 end
          Vector.__add = Vector.__sub
          assert.equal(c+d, Vector2(-0.5,-1,-1.5))
        end)

        it('reverts __add override', function()
          Vector2.__add = function(a, b) return Vector.__add(a, b)/2 end
          Vector2.__add = nil
          assert.equal(c+d, Vector2(3,6,9))
        end)
      end)
    end)
  end)

  describe('Default Metamethods', function()

    local Peter, peter

    before_each(function()
      Peter = class('Peter')
      peter = Peter()
    end)

    describe('A Class', function()
      it('has a call metamethod properly set', function()
        assert.is_true(peter:isInstanceOf(Peter))
      end)
      it('has a tostring metamethod properly set', function()
        assert.equal(tostring(Peter), 'class Peter')
      end)
    end)

    describe('An instance', function()
      it('has a tostring metamethod, returning a different result from Object.__tostring', function()
        assert.equal(tostring(peter), 'instance of class Peter')
      end)
    end)
  end)

end)