Added tests for CodeGeneration utility class.