From 5cc93d31f8e84e815eea25633826115689191bcb Mon Sep 17 00:00:00 2001
From: ahgamut <41098605+ahgamut@users.noreply.github.com>
Date: Tue, 6 Jul 2021 21:28:33 +0530
Subject: [PATCH] Added tests for LookupProto functions

Similar to tests for LookupService tests.

Also, the man page for getprotobyname does not specify case-insensitive,
but left it there anyway.
---
 test/libc/dns/prototxt_test.c | 73 +++++++++++++++++++++++++++++++++++
 1 file changed, 73 insertions(+)
 create mode 100644 test/libc/dns/prototxt_test.c

diff --git a/test/libc/dns/prototxt_test.c b/test/libc/dns/prototxt_test.c
new file mode 100644
index 000000000..548460933
--- /dev/null
+++ b/test/libc/dns/prototxt_test.c
@@ -0,0 +1,73 @@
+/*-*- mode:c;indent-tabs-mode:nil;c-basic-offset:2;tab-width:8;coding:utf-8 -*-│
+│vi: set net ft=c ts=2 sts=2 sw=2 fenc=utf-8                                :vi│
+╞══════════════════════════════════════════════════════════════════════════════╡
+│ This is free and unencumbered software released into the public domain.      │
+│                                                                              │
+│ Anyone is free to copy, modify, publish, use, compile, sell, or              │
+│ distribute this software, either in source code form or as a compiled        │
+│ binary, for any purpose, commercial or non-commercial, and by any            │
+│ means.                                                                       │
+│                                                                              │
+│ In jurisdictions that recognize copyright laws, the author or authors        │
+│ of this software dedicate any and all copyright interest in the              │
+│ software to the public domain. We make this dedication for the benefit       │
+│ of the public at large and to the detriment of our heirs and                 │
+│ successors. We intend this dedication to be an overt act of                  │
+│ relinquishment in perpetuity of all present and future rights to this        │
+│ software under copyright law.                                                │
+│                                                                              │
+│ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,              │
+│ EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF           │
+│ MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.       │
+│ IN NO EVENT SHALL THE AUTHORS BE LIABLE FOR ANY CLAIM, DAMAGES OR            │
+│ OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,        │
+│ ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR        │
+│ OTHER DEALINGS IN THE SOFTWARE.                                              │
+│                                                                              │
+╚─────────────────────────────────────────────────────────────────────────────*/
+#include "libc/dns/prototxt.h"
+
+#include "libc/calls/calls.h"
+#include "libc/dns/dns.h"
+#include "libc/dns/ent.h"
+#include "libc/testlib/testlib.h"
+
+char testlib_enable_tmp_setup_teardown;
+
+void SetUp() {
+  int fd;
+  const char* sample = "\
+# skip comment string\n\
+rspf	73	RSPF CPHB	\n\
+ggp	    3	GGP		    ";
+
+  ASSERT_NE(-1, (fd = creat("protocols", 0755)));
+  ASSERT_NE(-1, write(fd, sample, strlen(sample)));
+  ASSERT_NE(-1, close(fd));
+}
+
+TEST(LookupProtoByNumber, GetNameWhenNumberCorrect) {
+  char name[16]; /* sample has only names of length 3-4 */
+
+  ASSERT_EQ(-1, /*non-existent number */
+            LookupProtoByNumber(24, name, sizeof(name), "protocols"));
+
+  ASSERT_EQ(0, /* works with valid number */
+            LookupProtoByNumber(73, name, sizeof(name), "protocols"));
+  ASSERT_STREQ(name, "rspf"); /* official name written */
+}
+
+TEST(LookupProtoByName, GetNumberWhenNameOrAlias) {
+  char name[16]; /* sample has only names of length 3-4 */
+
+  ASSERT_EQ(-1, /* non-existent name or alias */
+            LookupProtoByName("tcp", name, sizeof(name), "protocols"));
+
+  ASSERT_EQ(3, /* works with valid name */
+            LookupProtoByName("ggp", name, sizeof(name), "protocols"));
+  ASSERT_STREQ(name, "ggp");
+
+  ASSERT_EQ(73, /* works with valid alias */
+            LookupProtoByName("CPHB", name, sizeof(name), "protocols"));
+  ASSERT_STREQ(name, "rspf"); /* official name written */
+}