initial commit

1 files changed, 114 insertions, 0 deletions

diff --git a/qtest.lua b/qtest.lua
new file mode 100755
index 0000000..c1c9c0e
--- /dev/null
+++ b/qtest.lua
@@ -0,0 +1,114 @@
+#!/usr/bin/env lem
+--
+-- This file is part of lem-postgres.
+-- Copyright 2011 Emil Renner Berthing
+--
+-- lem-postgres is free software: you can redistribute it and/or
+-- modify it under the terms of the GNU General Public License as
+-- published by the Free Software Foundation, either version 3 of
+-- the License, or (at your option) any later version.
+--
+-- lem-postgres is distributed in the hope that it will be useful,
+-- but WITHOUT ANY WARRANTY; without even the implied warranty of
+-- MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+-- GNU General Public License for more details.
+--
+-- You should have received a copy of the GNU General Public License
+-- along with lem-postgres. If not, see <http://www.gnu.org/licenses/>.
+--
+
+print("Entering " .. arg[0])
+
+local prettyprint
+do
+	local write, format, tostring = io.write, string.format, tostring
+
+	function prettyprint(t)
+		local widths = {}
+		for i = 1, #t[0] do
+			widths[i] = 0
+		end
+
+		for i = 0, #t do
+			local row = t[i]
+			for j = 1, #row do
+				local value = row[j]
+				if value and #value > widths[j] then
+					widths[j] = #value
+				end
+			end
+		end
+
+		for i = 1, #widths do
+			widths[i] = '%-' .. tostring(widths[i] + 1) .. 's';
+		end
+
+		for i = 0, #t do
+			local row = t[i]
+			for j = 1, #row do
+				write(format(widths[j], row[j] or 'NULL'))
+			end
+			write('\n')
+		end
+	end
+end
+
+local utils    = require 'lem.utils'
+local postgres = require 'lem.postgres.queued'
+
+local db = assert(postgres.connect("host=localhost dbname=mydb"))
+
+assert(db:exec(
+'CREATE TABLE mytable (id integer PRIMARY KEY, name CHARACTER VARYING)'))
+
+do
+	local raw = db:lock()
+	assert(raw:exec("COPY mytable FROM STDIN (delimiter ',')"))
+	assert(raw:put('1,alpha\n'))
+	assert(raw:put('2,beta\n'))
+	assert(raw:put('3,gamma\n'))
+	assert(raw:put('4,delta\n'))
+	assert(raw:done())
+	db:unlock()
+end
+
+utils.spawn(function()
+	assert(db:prepare('mystmt', 'SELECT * FROM mytable WHERE id = $1'))
+	local res = assert(db:run('mystmt', '3'))
+	prettyprint(res)
+end)
+
+utils.spawn(function()
+	local res = assert(db:exec('SELECT * FROM mytable WHERE id = $1', '1'))
+	prettyprint(res)
+end)
+
+utils.spawn(function()
+	local res1, res2 = assert(db:exec([[
+SELECT count(id) FROM mytable;
+SELECT * FROM mytable WHERE id = 2]]));
+	prettyprint(res2)
+end)
+
+do
+	local raw = db:lock()
+	assert(raw:exec("COPY mytable TO STDIN (format csv)"))
+
+	print("\nDumping CSV:")
+	while true do
+		local row = assert(raw:get())
+		if type(row) ~= 'string' then break end
+
+		io.write(row)
+	end
+
+	db:unlock()
+end
+
+utils.timer(1.0, function()
+	assert(db:exec('DROP TABLE mytable'))
+end)
+
+print("Exiting " .. arg[0])
+
+-- vim: syntax=lua ts=2 sw=2 noet: