//
// Copyright 2017 Garrett D'Amore <garrett@damore.org>
//
// This software is supplied under the terms of the MIT License, a
// copy of which should be located in the distribution where this
// file was obtained (LICENSE.txt).  A copy of the license may also be
// found online at https://opensource.org/licenses/MIT.
//

#include "convey.h"
#include "nng.h"
#include "core/nng_impl.h"

#include <string.h>

Main({
	Test("Socket Operations", {

	Convey("We are able to open a PAIR socket", {
		int rv;
		nng_socket sock;

		So(nng_open(&sock, NNG_PROTO_PAIR) == 0);

		Reset({
			nng_close(sock);
		})

		Convey("And we can shut it down", {
			rv = nng_shutdown(sock);
			So(rv == 0);
			rv = nng_shutdown(sock);
			So(rv == NNG_ECLOSED);
		})

		Convey("It's type is still proto", {
			So(nng_protocol(sock) == NNG_PROTO_PAIR);
		})

		Convey("Recv with no pipes times out correctly", {
			nng_msg *msg = NULL;
			int64_t when = 100000;
			uint64_t now;

			now = nni_clock();

			rv = nng_setopt(sock, NNG_OPT_RCVTIMEO, &when,
				sizeof (when));
			So(rv == 0);
			rv = nng_recvmsg(sock, &msg, 0);
			So(rv == NNG_ETIMEDOUT);
			So(msg == NULL);
			So(nni_clock() >= (now + when));
			So(nni_clock() < (now + (when * 2)));
		})

		Convey("Recv nonblock with no pipes gives EAGAIN", {
			nng_msg *msg = NULL;
			rv = nng_recvmsg(sock, &msg, NNG_FLAG_NONBLOCK);
			So(rv == NNG_EAGAIN);
			So(msg == NULL);
		})

		Convey("Send with no pipes times out correctly", {
			nng_msg *msg = NULL;
			int64_t when = 100000;
			uint64_t now;

			// We cheat to get access to the core's clock.
			So(nng_msg_alloc(&msg, 0) == 0);
			So(msg != NULL);
			now = nni_clock();

			rv = nng_setopt(sock, NNG_OPT_SNDTIMEO, &when,
				sizeof (when));
			So(rv == 0);
			rv = nng_sendmsg(sock, msg, 0);
			So(rv == NNG_ETIMEDOUT);
			So(nni_clock() >= (now + when));
			So(nni_clock() < (now + (when * 2)));
			nng_msg_free(msg);
		})

		Convey("We can set and get options", {
			int64_t when = 1234;
			int64_t check = 0;
			size_t sz;
			rv = nng_setopt(sock, NNG_OPT_SNDTIMEO, &when,
				sizeof (when));
			So(rv == 0);
			sz = sizeof (check);
			Convey("Short size is not copied", {
				sz = 0;
				rv = nng_getopt(sock, NNG_OPT_SNDTIMEO,
					&check, &sz);
				So(rv == 0);
				So(sz == sizeof (check));
				So(check == 0);
			})
			Convey("Correct size is copied", {
				sz = sizeof (check);
				rv = nng_getopt(sock, NNG_OPT_SNDTIMEO, &check,
					&sz);
				So(rv == 0);
				So(sz == sizeof (check));
				So(check == 1234);
			})
		})

		Convey("Bogus URLs not supported", {
			Convey("Dialing fails properly", {
				rv = nng_dial(sock, "bogus://somewhere", NULL, 0);
				So(rv == NNG_ENOTSUP);
			})
			Convey("Listening fails properly", {
				rv = nng_listen(sock, "bogus://elsewhere", NULL, 0);
				So(rv == NNG_ENOTSUP);
			})
		})

		Convey("Dialing synch can get refused", {
			rv = nng_dial(sock, "inproc://notthere", NULL, NNG_FLAG_SYNCH);
			So(rv == NNG_ECONNREFUSED);
		})

		Convey("Listening works", {
			rv = nng_listen(sock, "inproc://here", NULL, NNG_FLAG_SYNCH);
			So(rv == 0);

			Convey("Second listen fails ADDRINUSE", {
				rv = nng_listen(sock, "inproc://here", NULL, NNG_FLAG_SYNCH);
				So(rv == NNG_EADDRINUSE);
			})

			Convey("We can connect to it", {
				nng_socket sock2;
				So(nng_open(&sock2, NNG_PROTO_PAIR) == 0);
				Reset({
					nng_close(sock2);
				})
				rv = nng_dial(sock2, "inproc://here", NULL, NNG_FLAG_SYNCH);
				So(rv == 0);
				nng_close(sock2);
			})
		})

		Convey("We can send and receive messages", {
			nng_socket sock2;
			int len = 1;
			size_t sz;
			uint64_t second = 3000000;
			char *buf;

			So(nng_open(&sock2, NNG_PROTO_PAIR) == 0);
			Reset({
				nng_close(sock2);
			})

			So(nng_setopt(sock, NNG_OPT_RCVBUF, &len, sizeof (len)) == 0);
			So(nng_setopt(sock, NNG_OPT_SNDBUF, &len, sizeof (len)) == 0);

			So(nng_setopt(sock2, NNG_OPT_RCVBUF, &len, sizeof (len)) == 0);
			So(nng_setopt(sock2, NNG_OPT_SNDBUF, &len, sizeof (len)) == 0);

			So(nng_setopt(sock, NNG_OPT_SNDTIMEO, &second, sizeof (second)) == 0);
			So(nng_setopt(sock, NNG_OPT_RCVTIMEO, &second, sizeof (second)) == 0);
			So(nng_setopt(sock2, NNG_OPT_SNDTIMEO, &second, sizeof (second)) == 0);
			So(nng_setopt(sock2, NNG_OPT_RCVTIMEO, &second, sizeof (second)) == 0);

			So(nng_listen(sock, "inproc://test1", NULL, NNG_FLAG_SYNCH) == 0);
			So(nng_dial(sock2, "inproc://test1", NULL, NNG_FLAG_SYNCH) == 0);

			So(nng_send(sock, "abc", 4, 0) == 0);
			So(nng_recv(sock2 , &buf, &sz, NNG_FLAG_ALLOC) == 0);
			So(buf != NULL);
			So(sz == 4);
			So(memcmp(buf, "abc", 4) == 0);
			nng_free(buf, sz);
		})
	})
	})
})