Add a function token test and fix add_token

2 files changed, 9 insertions, 0 deletions

diff --git a/lib/tongue/langpack.lua b/lib/tongue/langpack.lua
index a2c0e24..b9a22d1 100644
--- a/lib/tongue/langpack.lua
+++ b/lib/tongue/langpack.lua
@@ -97,6 +97,7 @@ function langpack:add_token(token, expansion, strict)
       end
    end
    self.entries[token:upper()] = expansion
+   return true
 end
 
 --- Expand a token and arguments into a message.
diff --git a/test/test-tongue.langpack.lua b/test/test-tongue.langpack.lua
index 102dd6d..04f7050 100644
--- a/test/test-tongue.langpack.lua
+++ b/test/test-tongue.langpack.lua
@@ -52,6 +52,14 @@ function suite.tryparentfallback()
    assert(string.find(expn, 'foo="bar"'), expn)
 end
 
+function suite.add_function_expn()
+   local lang = tongue.langpack.create("en")
+   assert(lang, "Could not create language pack")
+   assert(lang:add_token("TEST", function(args) return args.foo end),
+	  "Unable to add a function token")
+   assert(lang:expand("TEST",{foo="bar"}) == "bar", "Function tokens don't work")
+end
+
 local count_ok = 0
 for _, testname in ipairs(testnames) do
 --   print("Run: " .. testname)