Test all Lua module functions

Author: tarleb

test/test-doclayout.lua

@@ -24,8 +24,39 @@ return {
       assert.are_equal(type(doclayout.space), 'userdata')
     end),
 
+    test('chomp', function ()
+      local doc = 'a' .. doclayout.blanklines(2)
+      assert.are_equal(doclayout.chomp(doc), doclayout.literal 'a')
+    end),
+    test('nestle', function ()
+      local doc = doclayout.blanklines(3) .. 'a'
+      assert.are_equal(doclayout.nestle(doc), doclayout.literal 'a')
+    end),
+    group 'enclosing' {
+      test('quotes', function ()
+        local doc = doclayout.literal 'single'
+        assert.are_equal(
+          doclayout.quotes(doc),
+          "'" .. doclayout.literal 'single' .. "'"
+        )
+      end),
+      test('double quotes', function ()
+        local doc = doclayout.literal 'double'
+        assert.are_equal(
+          doclayout.double_quotes(doc),
+          '"' .. doclayout.literal 'double' .. '"'
+        )
+      end),
+      test('inside', function ()
+        local doc = doclayout.literal 'Hello,' + 'World!'
+        assert.are_equal(
+          doclayout.inside('Yo! ', ' Wassup?', doc),
+          'Yo! ' .. doc .. ' Wassup?'
+        )
+      end),
+    },
     group 'concat' {
-      test('without sep', function ()
+      test('with sep', function ()
         local list = {
           doclayout.literal 'one',
           doclayout.literal 'two',
@@ -44,13 +75,122 @@ return {
         }
         assert.are_equal(doclayout.concat(list), list[1] .. list[2] .. list[3])
       end),
-    }
+    },
   },
 
   group 'render' {
     test('empty doc', function ()
       assert.are_equal(doclayout.render(doclayout.empty), '')
-    end)
+    end),
+    test('reflow', function ()
+      local greeting = doclayout.literal 'Hi!' .. doclayout.space
+        .. doclayout.literal 'How' .. doclayout.space
+        .. doclayout.literal 'are' .. doclayout.space
+        .. doclayout.literal 'you?'
+      assert.are_equal(doclayout.render(greeting, 7), 'Hi! How\nare\nyou?')
+    end),
+    test('after_break', function ()
+      local doc = doclayout.literal 'hi'
+        + doclayout.after_break '!'
+        .. doclayout.after_break '?'
+        .. doclayout.literal 'x'
+        .. doclayout.after_break '?'
+      assert.are_equal(doclayout.render(doc, 2), 'hi\n!x')
+    end),
+    test('before_non_blank', function ()
+      local doc = doclayout.before_non_blank '!!' .. ' ab'
+        / doclayout.before_non_blank '!!' .. 'a b'
+      assert.are_equal(doclayout.render(doc), ' ab\n!!a b')
+    end),
+    test('blanks at beginning', function ()
+      local doc = doclayout.blanklines(2) .. 'aa'
+      -- only one newline, as the top of doc is treated as implicit blank
+      assert.are_equal(doclayout.render(doc), '\naa')
+    end),
+    test('blanklines', function ()
+      local doc = 'aa' .. doclayout.blanklines(2) .. 'bb'
+      assert.are_equal(doclayout.render(doc), 'aa\n\n\nbb')
+    end),
+    test('braces', function ()
+      local doc = doclayout.braces 'maybe'
+      assert.are_equal(doclayout.render(doc), '{maybe}')
+    end),
+    test('brackets', function ()
+      local doc = doclayout.brackets '1'
+      assert.are_equal(doclayout.render(doc), '[1]')
+    end),
+    test('flush', function ()
+      local doc = doclayout.flush(doclayout.nest(2, 'hi'))
+      assert.are_equal(doclayout.render(doc), 'hi')
+    end),
+    test('hang', function ()
+      local doc = doclayout.hang(4, '  - ', 'aa\nbb\ncc')
+      assert.are_equal(
+        doclayout.render(doc),
+        table.concat{
+          '  - aa\n',
+          '    bb\n',
+          '    cc',
+        }
+      )
+    end),
+    test('nest', function ()
+      local doc = doclayout.nest(2, 'aa\n\nbb\ncc')
+      assert.are_equal(
+        doclayout.render(doc),
+        table.concat{
+          '  aa\n',
+          '\n',
+          '  bb\n',
+          '  cc'
+        }
+      )
+    end),
+    test('nowrap', function()
+      local doc = doclayout.nowrap(doclayout.literal 'first' + 'second')
+      assert.are_equal(doclayout.render(doc, 8), 'first second')
+    end),
+    test('parens', function ()
+      local doc = doclayout.parens 'lisp'
+      assert.are_equal(doclayout.render(doc), '(lisp)')
+    end),
+    test('prefixed', function ()
+      local doc = doclayout.prefixed('# ', doclayout.literal 'aa' // 'bb')
+      assert.are_equal(doclayout.render(doc), '# aa\n#\n# bb')
+    end),
+    group 'table helpers' {
+      test('cblock', function ()
+        local doc = doclayout.cblock(2, '| ')
+          .. doclayout.cblock(4, 'aa')
+          .. doclayout.cblock(2, ' |')
+        assert.are_equal(doclayout.render(doc), '|  aa  |')
+      end),
+      test('lblock', function ()
+        local doc = doclayout.lblock(2, '| ')
+          .. doclayout.lblock(4, 'aa')
+          .. doclayout.lblock(2, ' |')
+        assert.are_equal(doclayout.render(doc), '| aa   |')
+      end),
+      test('rblock', function ()
+        local doc = doclayout.rblock(2, '| ')
+          .. doclayout.rblock(4, 'aa')
+          .. doclayout.rblock(2, ' |')
+        assert.are_equal(doclayout.render(doc), '|   aa |')
+      end),
+      test('vfill', function ()
+        local doc = doclayout.vfill '| '
+          .. doclayout.lblock(4, doclayout.literal 'aa' // 'bbb')
+          .. doclayout.vfill(' |')
+        assert.are_equal(
+          doclayout.render(doc),
+          table.concat{
+            '| aa   |\n',
+            '|      |\n',
+            '| bbb  |'
+          }
+        )
+      end)
+    }
   },
 
   group 'document querying' {
@@ -86,6 +226,13 @@ return {
       assert.are_equal(doclayout.real_length('❄'), 1)
       assert.are_equal(doclayout.real_length('シ'), 2)
     end),
+
+    test('update_column', function ()
+      local doc = 'long' .. doclayout.cr .. 'longer'
+      assert.are_equal(doclayout.update_column(doc, 0), 6)
+      assert.are_equal(doclayout.update_column(doclayout.empty, 42), 0)
+      assert.are_equal(doclayout.update_column('four', 4), 8)
+    end)
   },
 
   group 'Doc type' {