Implemented tests for Toy_Bytecode and Toy_Routine

tests/cases/test_bytecode.c

@@ -1,6 +1,10 @@
 #include "toy_bytecode.h"
 #include "toy_console_colors.h"
 
+#include "toy_opcodes.h"
+#include "toy_lexer.h"
+#include "toy_parser.h"
+
 #include <stdio.h>
 #include <string.h>
 
@@ -30,6 +34,135 @@ int test_bytecode_header(Toy_Bucket* bucket) {
 			return -1;
 		}
 
+		if (bc.count % 4 != 0) {
+			fprintf(stderr, TOY_CC_ERROR "ERROR: bytecode size is not a multiple of 4, size is: %d\n" TOY_CC_RESET, bc.count);
+
+			//cleanup and return
+			Toy_freeBytecode(bc);
+			return -1;
+		}
+
+		//cleanup
+		Toy_freeBytecode(bc);
+	}
+
+	return 0;
+}
+
+int test_bytecode_from_source(Toy_Bucket* bucket) {
+	{
+		//setup
+		const char* source = "(1 + 2) * (3 + 4);";
+		Toy_Lexer lexer;
+		Toy_Parser parser;
+
+		Toy_bindLexer(&lexer, source);
+		Toy_bindParser(&parser, &lexer);
+		Toy_Ast* ast = Toy_scanParser(&bucket, &parser);
+
+		//run
+		Toy_Bytecode bc = Toy_compileBytecode(ast);
+
+		//check bytecode alignment
+		if (bc.count % 4 != 0) {
+			fprintf(stderr, TOY_CC_ERROR "ERROR: bytecode alignment is not a multiple of 4 (size is %d), source: %s\n" TOY_CC_RESET, bc.count, source);
+
+			//cleanup and return
+			Toy_freeBytecode(bc);
+			return -1;
+		}
+
+		//check bytecode header
+		if (bc.ptr[0] != TOY_VERSION_MAJOR ||
+			bc.ptr[1] != TOY_VERSION_MINOR ||
+			bc.ptr[2] != TOY_VERSION_PATCH ||
+			strcmp((char*)(bc.ptr + 3), TOY_VERSION_BUILD) != 0)
+		{
+			fprintf(stderr, TOY_CC_ERROR "ERROR: failed to write the bytecode header, source: %s\n" TOY_CC_RESET, source);
+
+			//cleanup and return
+			Toy_freeBytecode(bc);
+			return -1;
+		}
+
+		//check contents of the routine (this is copy/pasted from test_routine.c, and tweaked with the offset)
+		int offset = strlen(TOY_VERSION_BUILD) + 3;
+		if (offset % 4 != 0) {
+			offset += 4 - (offset % 4); //ceil
+		}
+
+		int* ptr = (int*)(bc.ptr + offset);
+
+		if ((ptr++)[0] != 72 || //total size
+			(ptr++)[0] != 0 || //param count
+			(ptr++)[0] != 0 || //jump count
+			(ptr++)[0] != 0 || //data count
+			(ptr++)[0] != 0) //subs count
+		{
+			fprintf(stderr, TOY_CC_ERROR "ERROR: failed to produce the expected routine header within bytecode, source: %s\n" TOY_CC_RESET, source);
+
+			//cleanup and return
+			Toy_freeBytecode(bc);
+			return -1;
+		}
+
+		//check code
+		if (
+			//left hand side
+			*((unsigned char*)(offset + bc.ptr + 24)) != TOY_OPCODE_READ ||
+			*((unsigned char*)(offset + bc.ptr + 25)) != TOY_VALUE_INTEGER ||
+			*((unsigned char*)(offset + bc.ptr + 26)) != 0 ||
+			*((unsigned char*)(offset + bc.ptr + 27)) != 0 ||
+			*(int*)(offset + bc.ptr + 28) != 1 ||
+
+			*((unsigned char*)(offset + bc.ptr + 32)) != TOY_OPCODE_READ ||
+			*((unsigned char*)(offset + bc.ptr + 33)) != TOY_VALUE_INTEGER ||
+			*((unsigned char*)(offset + bc.ptr + 34)) != 0 ||
+			*((unsigned char*)(offset + bc.ptr + 35)) != 0 ||
+			*(int*)(offset + bc.ptr + 36) != 2 ||
+
+			*((unsigned char*)(offset + bc.ptr + 40)) != TOY_OPCODE_ADD ||
+			*((unsigned char*)(offset + bc.ptr + 41)) != 0 ||
+			*((unsigned char*)(offset + bc.ptr + 42)) != 0 ||
+			*((unsigned char*)(offset + bc.ptr + 43)) != 0 ||
+
+			//right hand side
+			*((unsigned char*)(offset + bc.ptr + 44)) != TOY_OPCODE_READ ||
+			*((unsigned char*)(offset + bc.ptr + 45)) != TOY_VALUE_INTEGER ||
+			*((unsigned char*)(offset + bc.ptr + 46)) != 0 ||
+			*((unsigned char*)(offset + bc.ptr + 47)) != 0 ||
+			*(int*)(offset + bc.ptr + 48) != 3 ||
+
+			*((unsigned char*)(offset + bc.ptr + 52)) != TOY_OPCODE_READ ||
+			*((unsigned char*)(offset + bc.ptr + 53)) != TOY_VALUE_INTEGER ||
+			*((unsigned char*)(offset + bc.ptr + 54)) != 0 ||
+			*((unsigned char*)(offset + bc.ptr + 55)) != 0 ||
+			*(int*)(offset + bc.ptr + 56) != 4 ||
+
+			*((unsigned char*)(offset + bc.ptr + 60)) != TOY_OPCODE_ADD ||
+			*((unsigned char*)(offset + bc.ptr + 61)) != 0 ||
+			*((unsigned char*)(offset + bc.ptr + 62)) != 0 ||
+			*((unsigned char*)(offset + bc.ptr + 63)) != 0 ||
+
+			//multiply the two values
+			*((unsigned char*)(offset + bc.ptr + 64)) != TOY_OPCODE_MULTIPLY ||
+			*((unsigned char*)(offset + bc.ptr + 65)) != 0 ||
+			*((unsigned char*)(offset + bc.ptr + 66)) != 0 ||
+			*((unsigned char*)(offset + bc.ptr + 67)) != 0 ||
+
+			*((unsigned char*)(offset + bc.ptr + 68)) != TOY_OPCODE_RETURN ||
+			*((unsigned char*)(offset + bc.ptr + 69)) != 0 ||
+			*((unsigned char*)(offset + bc.ptr + 70)) != 0 ||
+			*((unsigned char*)(offset + bc.ptr + 71)) != 0
+		)
+		{
+			fprintf(stderr, TOY_CC_ERROR "ERROR: failed to produce the expected routine code within bytecode, source: %s\n" TOY_CC_RESET, source);
+
+			//cleanup and return
+			Toy_freeBytecode(bc);
+			return -1;
+		}
+
 		//cleanup
 		Toy_freeBytecode(bc);
 	}
@@ -52,5 +185,16 @@ int main() {
 		total += res;
 	}
 
+	{
+		Toy_Bucket* bucket = NULL;
+		TOY_BUCKET_INIT(Toy_Ast, bucket, 32);
+		res = test_bytecode_from_source(bucket);
+		TOY_BUCKET_FREE(bucket);
+		if (res == 0) {
+			printf(TOY_CC_NOTICE "All good\n" TOY_CC_RESET);
+		}
+		total += res;
+	}
+
 	return total;
 }