From 7e844cf618d67675928fb4c6fadc6cc16db19d19 Mon Sep 17 00:00:00 2001
From: Garrett D'Amore <garrett@damore.org>
Date: Wed, 18 Jan 2017 01:16:40 -0800
Subject: Fixes for valgrind issues.

---
 tests/sock.c | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

(limited to 'tests/sock.c')

diff --git a/tests/sock.c b/tests/sock.c
index 7d1d226b..6a122cdf 100644
--- a/tests/sock.c
+++ b/tests/sock.c
@@ -10,10 +10,15 @@
 #include "convey.h"
 #include "stubs.h"
 #include "nng.h"
+#include "core/nng_impl.h"
 
 #include <string.h>
 
-TestMain("Socket Operations", {
+Main({
+	nni_init();
+
+	Test("Socket Operations", {
+
 	Convey("We are able to open a PAIR socket", {
 		int rv;
 		nng_socket *sock = NULL;
@@ -194,4 +199,7 @@ TestMain("Socket Operations", {
 			nng_close(sock2);
 		})
 	})
+	})
+
+	nni_fini();
 })
-- 
cgit v1.2.3-70-g09d2