local upb = require "upb"
local pb = require "upb.pb"
local lunit = require "lunit"

if _VERSION >= 'Lua 5.2' then
  _ENV = lunit.module("testupb_pb", "seeall")
else
  module("testupb_pb", lunit.testcase, package.seeall)
end

local primitive_types_msg = upb.build_defs{
  upb.MessageDef{full_name = "TestMessage", fields = {
    upb.FieldDef{name = "i32", number = 1, type = upb.TYPE_INT32},
    upb.FieldDef{name = "u32", number = 2, type = upb.TYPE_UINT32},
    upb.FieldDef{name = "i64", number = 3, type = upb.TYPE_INT64},
    upb.FieldDef{name = "u64", number = 4, type = upb.TYPE_UINT64},
    upb.FieldDef{name = "dbl", number = 5, type = upb.TYPE_DOUBLE},
    upb.FieldDef{name = "flt", number = 6, type = upb.TYPE_FLOAT},
    upb.FieldDef{name = "bool", number = 7, type = upb.TYPE_BOOL},
    }
  }
}

function test_decodermethod()
  local dm = pb.DecoderMethod(primitive_types_msg)

  assert_error(
    function()
      -- Needs at least one argument to construct.
      pb.DecoderMethod()
    end)
end

function test_parse_primitive()
  local binary_pb =
         "\008\128\128\128\128\002\016\128\128\128\128\004\024\128\128"
      .. "\128\128\128\128\128\002\032\128\128\128\128\128\128\128\001\041\000"
      .. "\000\000\000\000\000\248\063\053\000\000\096\064\056\001"
  local dm = pb.DecoderMethod(primitive_types_msg)
  msg = dm:parse(binary_pb)
  assert_equal(536870912, msg.i32)
  assert_equal(1073741824, msg.u32)
  assert_equal(1125899906842624, msg.i64)
  assert_equal(562949953421312, msg.u64)
  assert_equal(1.5, msg.dbl)
  assert_equal(3.5, msg.flt)
  assert_equal(true, msg.bool)
end

function test_parse_string()
  local msgdef = upb.build_defs{
    upb.MessageDef{full_name = "TestMessage", fields = {
      upb.FieldDef{name = "str", number = 1, type = upb.TYPE_STRING},
      }
    }
  }

  local binary_pb = "\010\005Hello"
  local dm = pb.DecoderMethod(msgdef)
  msg = dm:parse(binary_pb)
  assert_equal("Hello", msg.str)
end


local stats = lunit.main()

if stats.failed > 0 or stats.errors > 0 then
  error("One or more errors in test suite")
end