I don't remember

1 files changed, 104 insertions, 0 deletions

diff --git a/test/peers.c b/test/peers.c
new file mode 100644
index 0000000..1e9a454
--- /dev/null
+++ b/test/peers.c
@@ -0,0 +1,104 @@
+#include "unity.h"
+#include "peers.h"
+#include "log.h"
+
+#include <string.h>
+#include <arpa/inet.h>
+
+#define IP(a, b, c, d) htonl(a << 24 | b << 16 | c << 8 | d)
+
+void test_add_single_peer() {
+	allocate_hashtable();
+
+	struct infohash sometorrent = {.inner={0x0034048f, 0x08000020, 0x00888880, 0x02008460, 0x0ab00521}};
+	struct addr addr = (struct addr){.ip = IP(128,0,0,1), .port = 0};
+
+	int rc = add_peer(&sometorrent, &addr);
+	TEST_ASSERT_EQUAL(0, rc);
+}
+
+void test_add_9_peers_same_torrent() {
+	allocate_hashtable();
+
+	struct infohash sometorrent = {.inner={0x0034048f, 0x08000020, 0x00888880, 0x02008460, 0x0ab00521}};
+	struct addr addr = (struct addr){.ip = IP(128,0,0,1), .port = 0};
+
+	int rc = add_peer(&sometorrent, &addr);
+	TEST_ASSERT_EQUAL(0, rc);
+	rc = add_peer(&sometorrent, &addr);
+	TEST_ASSERT_EQUAL(0, rc);
+	rc = add_peer(&sometorrent, &addr);
+	TEST_ASSERT_EQUAL(0, rc);
+	rc = add_peer(&sometorrent, &addr);
+	TEST_ASSERT_EQUAL(0, rc);
+	rc = add_peer(&sometorrent, &addr);
+	TEST_ASSERT_EQUAL(0, rc);
+	rc = add_peer(&sometorrent, &addr);
+	TEST_ASSERT_EQUAL(0, rc);
+	rc = add_peer(&sometorrent, &addr);
+	TEST_ASSERT_EQUAL(0, rc);
+	rc = add_peer(&sometorrent, &addr);
+	TEST_ASSERT_EQUAL(0, rc);
+	// There's room for 8 peers per infohash
+	rc = add_peer(&sometorrent, &addr);
+	TEST_ASSERT_EQUAL(PEER_EFULL, rc);
+}
+
+void test_peers_for_only_torrent() {
+	allocate_hashtable();
+
+	struct infohash sometorrent  = {.inner={0x0034048f, 0x08000020, 0x00888880, 0x02008460, 0x0ab00521}};
+	struct infohash othertorrent = {.inner={0x0034048f, 0x08000020, 0x00888880, 0x02008460, 0x0ab00522}};
+	struct addr someaddr = (struct addr){.ip = IP(128,0,0,1), .port = 0};
+	struct addr otheraddr = (struct addr){.ip = IP(128,0,0,1), .port = 1};
+
+	int rc = add_peer(&sometorrent, &someaddr);
+	TEST_ASSERT_EQUAL(0, rc);
+	rc = add_peer(&othertorrent, &otheraddr);
+	TEST_ASSERT_EQUAL(0, rc);
+
+	struct addr* found;
+	size_t found_len;
+
+	get_peers(&sometorrent, &found, &found_len);
+	TEST_ASSERT_EQUAL(1, found_len);
+	TEST_ASSERT_EQUAL_MEMORY(&someaddr, &found[0], sizeof(struct addr));
+
+	get_peers(&othertorrent, &found, &found_len);
+	TEST_ASSERT_EQUAL(1, found_len);
+	TEST_ASSERT_EQUAL_MEMORY(&otheraddr, &found[0], sizeof(struct addr));
+}
+
+void test_have_no_peers() {
+	allocate_hashtable();
+
+	struct infohash sometorrent = {.inner={0x0034048f, 0x08000020, 0x00888880, 0x02008460, 0x0ab00521}};
+
+	struct addr* found;
+	size_t found_len;
+	get_peers(&sometorrent, &found, &found_len);
+	TEST_ASSERT_NULL(found);
+	TEST_ASSERT_EQUAL(0, found_len);
+}
+
+void test_grows() {
+	allocate_hashtable();
+
+	struct addr addr = (struct addr){.ip = IP(128,0,0,1), .port = 0};
+
+	for(int i = 0; i < 17; i++) { // Peer table size + 1 to make sure it HAS to grow
+		struct infohash sometorrent = {.inner={0x0034048f, 0x08000020, 0x00888880, 0x02008460, i}};
+		int rc = add_peer(&sometorrent, &addr);
+		TEST_ASSERT_EQUAL(0, rc);
+	}
+
+	{
+		// We can find a peer after the resize again
+		struct infohash sometorrent = {.inner={0x0034048f, 0x08000020, 0x00888880, 0x02008460, 0x00000001}};
+		struct addr* found;
+		size_t found_len;
+		get_peers(&sometorrent, &found, &found_len);
+		TEST_ASSERT_NOT_NULL(found);
+		TEST_ASSERT_EQUAL(1, found_len);
+	}
+}