[utils] Added file_to_set, set_to_file + test #19

openwisp-config/files/lib/openwisp/utils.lua

@@ -134,3 +134,21 @@ function file_exists(path)
     if f ~= nil then io.close(f) return true end
     return false
 end
+
+function file_to_set(path)
+    local f = io.open(path, 'r')
+    local set = {}
+    for line in f:lines() do
+        set[line] = true
+    end
+    return set
+end
+
+function set_to_file(set, path)
+    local f = io.open(path, 'w')
+    for file, bool in pairs(set) do
+        f:write(file, '\n')
+    end
+    io.close(f)
+    return true
+end

openwisp-config/tests/test_utils.lua

@@ -247,4 +247,25 @@ function TestUtils.test_file_exists()
     luaunit.assertEquals(file_exists('./WRONG'), false)
 end
 
+function TestUtils.test_file_to_set()
+    os.execute('echo "line1" > '..write_dir..'/read.list')
+    os.execute('echo "line2" >> '..write_dir..'/read.list')
+    os.execute('echo "line3" >> '..write_dir..'/read.list')
+    set = file_to_set(write_dir..'/read.list')
+    luaunit.assertEquals(set.line1, true)
+    luaunit.assertEquals(set.line2, true)
+    luaunit.assertEquals(set.line3, true)
+    luaunit.assertEquals(set.line4, nil)
+end
+
+function TestUtils.test_set_to_file()
+    write = {line1=true, line2=true}
+    result = set_to_file(write, write_dir..'/write.list')
+    luaunit.assertEquals(result, true)
+    read = file_to_set(write_dir..'/write.list')
+    luaunit.assertEquals(read.line1, true)
+    luaunit.assertEquals(read.line2, true)
+    luaunit.assertEquals(read.line3, nil)
+end
+
 os.exit(luaunit.LuaUnit.run())