import { assertEquals, assertRejects,} from "https://deno.land/std@0.203.0/assert/mod.ts";import { StringReader } from "https://deno.land/std@0.203.0/io/string_reader.ts";import { readDelim } from "https://deno.land/std@0.203.0/io/read_delim.ts";import { type Command, readReply, RedisClient, type Reply } from "./mod.ts";
const encoder = new TextEncoder();
const CRLF = "\r\n";
async function readReplyTest(output: string, expected: Reply) { assertEquals( await readReply(readDelim(new StringReader(output), encoder.encode(CRLF))), expected, );}
async function readReplyRejectTest(output: string, expected: string) { await assertRejects( async () => await readReply( readDelim(new StringReader(output), encoder.encode(CRLF)), ), expected, );}
Deno.test("readReply() - mixed array", async () => await readReplyTest("*3\r\n$5\r\nstring\r\n:123\r\n$-1", [ "string", 123, null, ]));
Deno.test("readReply() - empty array", async () => await readReplyTest("*0\r\n", []));
Deno.test("readReply() - null array", async () => await readReplyTest("*-1\r\n", null));
Deno.test("readReply() - nested array", async () => await readReplyTest("*2\r\n*3\r\n:1\r\n$5\r\nhello\r\n:2\r\n#f\r\n", [[ 1, "hello", 2, ], false]));
Deno.test("readReply() - attribute", async () => { await readReplyTest( "|1\r\n+key-popularity\r\n%2\r\n$1\r\na\r\n,0.1923\r\n$1\r\nb\r\n,0.0012\r\n*2\r\n:2039123\r\n:9543892\r\n", [2039123, 9543892], ); await readReplyTest("*3\r\n:1\r\n:2\r\n|1\r\n+ttl\r\n:3600\r\n:3\r\n", [ 1, 2, 3, ]);});
Deno.test("readReply() - positive big number", async () => await readReplyTest( "(3492890328409238509324850943850943825024385\r\n", 3492890328409238509324850943850943825024385n, ));
Deno.test("readReply() - negative big number", async () => await readReplyTest( "(-3492890328409238509324850943850943825024385\r\n", -3492890328409238509324850943850943825024385n, ));
Deno.test("readReply() - true boolean", async () => await readReplyTest("#t\r\n", true));
Deno.test("readReply() - false boolean", async () => await readReplyTest("#f\r\n", false));
Deno.test("readReply() - integer", async () => await readReplyTest(":42\r\n", 42));
Deno.test("readReply() - bulk string", async () => await readReplyTest("$5\r\nhello\r\n", "hello"));
Deno.test("readReply() - emtpy bulk string", async () => await readReplyTest("$0\r\n\r\n", ""));
Deno.test("readReply() - null bulk string", async () => await readReplyTest("$-1\r\n", null));
Deno.test("readReply() - raw bulk string", async () => {});
Deno.test("readReply() - blob error", async () => { await readReplyRejectTest( "!21\r\nSYNTAX invalid syntax\r\n", "SYNTAX invalid syntax", );});
Deno.test("readReply() - error", async () => { await readReplyRejectTest( "-ERR this is the error description\r\n", "ERR this is the error description", );});
Deno.test("readReply() - double", async () => await readReplyTest(",1.23\r\n", 1.23));
Deno.test("readReply() - positive infinity double", async () => await readReplyTest(",inf\r\n", Infinity));
Deno.test("readReply() - negative infinity double", async () => await readReplyTest(",-inf\r\n", -Infinity));
Deno.test("readReply() - map", async () => await readReplyTest("%2\r\n+first\r\n:1\r\n+second\r\n:2\r\n", { first: 1, second: 2, }));
Deno.test("readReply() - null", async () => await readReplyTest("_\r\n", null));
Deno.test("readReply() - push", async () => { await readReplyTest( ">4\r\n+pubsub\r\n+message\r\n+somechannel\r\n+this is the message\r\n", ["pubsub", "message", "somechannel", "this is the message"], );});
Deno.test("readReply() - set", async () => { await readReplyTest( "~5\r\n+orange\r\n+apple\r\n#t\r\n:100\r\n:999\r\n", new Set(["orange", "apple", true, 100, 999]), );});
Deno.test("readReply() - simple string", async () => await readReplyTest("+OK\r\n", "OK"));
Deno.test("readReply() - streamed string", async () => { await readReplyTest( "$?\r\n;4\r\nHell\r\n;5\r\no wor\r\n;1\r\nd\r\n;0\r\n", "Hello word", );});
Deno.test("readReply() - streamed array", async () => { await readReplyTest("*?\r\n:1\r\n:2\r\n:3\r\n.\r\n", [1, 2, 3]);});
Deno.test("readReply() - streamed set", async () => { await readReplyTest( "~?\r\n+a\r\n:1\r\n+b\r\n:2\r\n.\r\n", new Set(["a", 1, "b", 2]), );});
Deno.test("readReply() - streamed map", async () => { await readReplyTest("%?\r\n+a\r\n:1\r\n+b\r\n:2\r\n.\r\n", { a: 1, b: 2 });});
Deno.test("readReply() - verbatim string", async () => { await readReplyTest("=15\r\ntxt:Some string\r\n", "txt:Some string");});
Deno.test("readReply() - large reply", async () => { const reply = "a".repeat(4096 * 2); await readReplyTest(`$${reply.length}\r\n${reply}\r\n`, reply);});
const PORT = 6379;const redisConn = await Deno.connect({ port: PORT });const redisClient = new RedisClient(redisConn);
await redisClient.sendCommand(["FLUSHALL"]);
async function sendCommandTest( command: Command, expected: Reply,): Promise<void> { assertEquals(await redisClient.sendCommand(command), expected);}
Deno.test("redisClient.sendCommand() - transactions", async () => { await sendCommandTest(["MULTI"], "OK"); await sendCommandTest(["INCR", "FOO"], "QUEUED"); await sendCommandTest(["INCR", "BAR"], "QUEUED"); await sendCommandTest(["EXEC"], [1, 1]);});
Deno.test("redisClient.sendCommand() - raw data", async () => { const data = new Uint8Array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9]); assertEquals(await redisClient.sendCommand(["SET", "binary", data]), "OK"); assertEquals(await redisClient.sendCommand(["GET", "binary"], true), data);});
Deno.test("redisClient.sendCommand() - eval script", async () => { await sendCommandTest(["EVAL", "return ARGV[1]", 0, "hello"], "hello");});
Deno.test("redisClient.sendCommand() - Lua script", async () => { await sendCommandTest([ "FUNCTION", "LOAD", "#!lua name=mylib\nredis.register_function('knockknock', function() return 'Who\\'s there?' end)", ], "mylib"); await sendCommandTest(["FCALL", "knockknock", 0], "Who's there?");});
Deno.test("redisClient.sendCommand() - RESP3", async () => { await redisClient.sendCommand(["HELLO", 3]); await sendCommandTest(["HSET", "hash3", "foo", 1, "bar", 2], 2); await sendCommandTest(["HGETALL", "hash3"], { foo: "1", bar: "2", });});
Deno.test("redisClient.sendCommand() - race condition", async () => { async function fn() { const key = crypto.randomUUID(); const value = crypto.randomUUID(); await redisClient.sendCommand(["SET", key, value]); const result = await redisClient.sendCommand(["GET", key]); assertEquals(result, value); }
await Promise.all([ fn(), fn(), fn(), fn(), fn(), fn(), fn(), fn(), fn(), fn(), fn(), fn(), fn(), fn(), fn(), fn(), fn(), fn(), fn(), ]);});
Deno.test("redisClient.pipelineCommands()", async () => { assertEquals( await redisClient.pipelineCommands([ ["INCR", "X"], ["INCR", "X"], ["INCR", "X"], ["INCR", "X"], ]), [1, 2, 3, 4], );});
Deno.test("redisClient.writeCommand() + redisClient.readReplies()", async () => { await redisClient.writeCommand(["SUBSCRIBE", "mychannel"]); const iterator = redisClient.readReplies(); assertEquals(await iterator.next(), { value: ["subscribe", "mychannel", 1], done: false, }); await redisClient.writeCommand(["UNSUBSCRIBE"]); assertEquals(await iterator.next(), { value: ["unsubscribe", "mychannel", 0], done: false, });});
Deno.test("redisClient.sendCommand() - no reply", async () => { await assertRejects( async () => await redisClient.sendCommand(["SHUTDOWN"]), TypeError, "No reply received", );});
addEventListener("unload", () => redisConn.close());