Fix Cache Warm Bug

Author: its-hammer-time

validator.go

@@ -468,15 +468,17 @@ func warmMediaTypeSchema(mediaType *v3.MediaType, schemaCache cache.SchemaCache,
 			schema := mediaType.Schema.Schema()
 			if schema != nil {
 				renderedInline, _ := schema.RenderInline()
+				referenceSchema := string(renderedInline)
 				renderedJSON, _ := utils.ConvertYAMLtoJSON(renderedInline)
 				if len(renderedInline) > 0 {
 					compiledSchema, _ := helpers.NewCompiledSchema(fmt.Sprintf("%x", hash), renderedJSON, options)
 
 					schemaCache.Store(hash, &cache.SchemaCacheEntry{
-						Schema:         schema,
-						RenderedInline: renderedInline,
-						RenderedJSON:   renderedJSON,
-						CompiledSchema: compiledSchema,
+						Schema:          schema,
+						RenderedInline:  renderedInline,
+						ReferenceSchema: referenceSchema,
+						RenderedJSON:    renderedJSON,
+						CompiledSchema:  compiledSchema,
 					})
 				}
 			}
@@ -513,16 +515,18 @@ func warmParameterSchema(param *v3.Parameter, schemaCache cache.SchemaCache, opt
 		if schema != nil {
 			if _, exists := schemaCache.Load(hash); !exists {
 				renderedInline, _ := schema.RenderInline()
+				referenceSchema := string(renderedInline)
 				renderedJSON, _ := utils.ConvertYAMLtoJSON(renderedInline)
 				if len(renderedInline) > 0 {
 					compiledSchema, _ := helpers.NewCompiledSchema(fmt.Sprintf("%x", hash), renderedJSON, options)
 
 					// Store in cache using the shared SchemaCache type
 					schemaCache.Store(hash, &cache.SchemaCacheEntry{
-						Schema:         schema,
-						RenderedInline: renderedInline,
-						RenderedJSON:   renderedJSON,
-						CompiledSchema: compiledSchema,
+						Schema:          schema,
+						RenderedInline:  renderedInline,
+						ReferenceSchema: referenceSchema,
+						RenderedJSON:    renderedJSON,
+						CompiledSchema:  compiledSchema,
 					})
 				}
 			}

validator_test.go

@@ -2013,6 +2013,8 @@ func TestCacheWarming_PopulatesCache(t *testing.T) {
 	validator.options.SchemaCache.Range(func(key [32]byte, value *cache.SchemaCacheEntry) bool {
 		count++
 		assert.NotNil(t, value.CompiledSchema, "Cache entry should have compiled schema")
+		assert.NotEmpty(t, value.ReferenceSchema, "Cache entry should have pre-converted ReferenceSchema string")
+		assert.Equal(t, string(value.RenderedInline), value.ReferenceSchema, "ReferenceSchema should match string conversion of RenderedInline")
 		return true
 	})
 	assert.Greater(t, count, 0, "Schema cache should have entries from request and response bodies")