1 files changed, 75 insertions, 0 deletions
diff --git a/src/test/test_streams.cpp b/src/test/test_streams.cpp
new file mode 100644
index 0000000..cbcbcac
--- /dev/null
+++ b/src/test/test_streams.cpp
@@ -0,0 +1,75 @@
+#include <stdio.h>
+#include <assert.h>
+#include <stdlib.h>
+#include <string.h>
+#include <time.h>
+
+#include "../shared/lua_api/stream.hpp"
+
+int main(){
+	//Simple
+	struct stream* s = stream_create();
+	stream_writeInt(s,5);
+	assert(stream_readInt(s) == 5);
+	stream_free(s);
+
+	//Little more complex
+	s = stream_create();
+	stream_writeInt(s,1);
+	stream_writeInt(s,2);
+	stream_writeInt(s,3);
+	stream_writeInt(s,4);
+	assert(stream_readInt(s) == 1);
+	assert(stream_readInt(s) == 2);
+	assert(stream_readInt(s) == 3);
+	assert(stream_readInt(s) == 4);
+	stream_free(s);
+
+	//Make sure we're not leaking memory
+	//Uncomment this section, and check htop or windows task manager
+	/*
+	time_t t;
+	time(&t);
+	srand((unsigned) t);
+	int i,j;
+	for(i = 0; i < 100000; i++){
+		s = stream_create();
+		for(j = 0; j < 10000; j++){
+			stream_writeInt(s,rand());
+		}
+		stream_free(s);
+	}
+	*/
+
+	//Read/write data
+	s = stream_create();
+	char* str = (char*)"One two three four!\n";
+	int slen = strlen(str);
+	stream_writeData(s,str,slen);
+	char out[slen+1];
+	stream_readData(s,slen,out);
+	out[slen] = '\0';
+	assert(strcmp(out,str) == 0);
+	stream_free(s);
+
+	//Read/write string
+	s = stream_create();
+	stream_writeString(s,str,slen);
+	char* rstr = stream_readString(s);
+	assert(strcmp(rstr,str) == 0);
+	stream_free(s);
+
+	//Make sure the string reads it's null character
+	s = stream_create();
+	stream_writeString(s,str,slen);
+	stream_writeInt(s,5);
+	stream_print(s);
+	stream_readString(s);
+	stream_print(s);
+	int i = stream_readInt(s);
+	assert(i == 5);
+	stream_free(s);
+
+	printf("OK!\n");
+	return 0;
+}