From 08e6402a1bf093bfc27003ffcc98a26b2fd7607f Mon Sep 17 00:00:00 2001
From: Reuven <rh@tufin.com>
Date: Sun, 7 Apr 2024 13:35:36 +0300
Subject: [PATCH] del GetJsonOrigValue and GetJsonNewValue

---
 ...rameter-x-extensible-enum-value-removed.go |  4 +--
 ...roperty-x-extensible-enum-value-removed.go |  4 +--
 diff/json_diff.go                             | 26 -------------------
 3 files changed, 4 insertions(+), 30 deletions(-)

diff --git a/checker/check-request-parameter-x-extensible-enum-value-removed.go b/checker/check-request-parameter-x-extensible-enum-value-removed.go
index 29f09046..06e99556 100644
--- a/checker/check-request-parameter-x-extensible-enum-value-removed.go
+++ b/checker/check-request-parameter-x-extensible-enum-value-removed.go
@@ -45,11 +45,11 @@ func RequestParameterXExtensibleEnumValueRemovedCheck(diffReport *diff.Diff, ope
 					if paramItem.SchemaDiff.ExtensionsDiff.Modified[diff.XExtensibleEnumExtension] == nil {
 						continue
 					}
-					from, ok := diff.GetJsonOrigValue(paramItem.SchemaDiff.ExtensionsDiff.Modified[diff.XExtensibleEnumExtension])
+					from, ok := paramItem.SchemaDiff.Base.Extensions[diff.XExtensibleEnumExtension].(json.RawMessage)
 					if !ok {
 						continue
 					}
-					to, ok := diff.GetJsonNewValue(paramItem.SchemaDiff.ExtensionsDiff.Modified[diff.XExtensibleEnumExtension])
+					to, ok := paramItem.SchemaDiff.Revision.Extensions[diff.XExtensibleEnumExtension].(json.RawMessage)
 					if !ok {
 						continue
 					}
diff --git a/checker/check-request-property-x-extensible-enum-value-removed.go b/checker/check-request-property-x-extensible-enum-value-removed.go
index 1bd6d28a..d8e2f73a 100644
--- a/checker/check-request-property-x-extensible-enum-value-removed.go
+++ b/checker/check-request-property-x-extensible-enum-value-removed.go
@@ -45,11 +45,11 @@ func RequestPropertyXExtensibleEnumValueRemovedCheck(diffReport *diff.Diff, oper
 						if propertyDiff.ExtensionsDiff.Modified[diff.XExtensibleEnumExtension] == nil {
 							return
 						}
-						from, ok := diff.GetJsonOrigValue(propertyDiff.ExtensionsDiff.Modified[diff.XExtensibleEnumExtension])
+						from, ok := propertyDiff.Base.Extensions[diff.XExtensibleEnumExtension].(json.RawMessage)
 						if !ok {
 							return
 						}
-						to, ok := diff.GetJsonOrigValue(propertyDiff.ExtensionsDiff.Modified[diff.XExtensibleEnumExtension])
+						to, ok := propertyDiff.Base.Extensions[diff.XExtensibleEnumExtension].(json.RawMessage)
 						if !ok {
 							return
 						}
diff --git a/diff/json_diff.go b/diff/json_diff.go
index ddadc4fd..63d1f7d7 100644
--- a/diff/json_diff.go
+++ b/diff/json_diff.go
@@ -1,8 +1,6 @@
 package diff
 
 import (
-	"encoding/json"
-
 	"github.com/wI2L/jsondiff"
 )
 
@@ -43,27 +41,3 @@ func compareJson(source, target interface{}, opts ...jsondiff.Option) (jsonPatch
 	}
 	return toJsonPatch(patch), nil
 }
-
-// GetJsonOrigValue returns the original value of the diff, only if there is exactly one diff
-func GetJsonOrigValue(patch jsonPatch) (json.RawMessage, bool) {
-	if len(patch) != 1 {
-		return nil, false
-	}
-	result, ok := patch[0].OldValue.(json.RawMessage)
-	if !ok {
-		return nil, false
-	}
-	return result, true
-}
-
-// GetJsonOrigValue returns the new value of the diff, only if there is exactly one diff
-func GetJsonNewValue(patch jsonPatch) (json.RawMessage, bool) {
-	if len(patch) != 1 {
-		return nil, false
-	}
-	result, ok := patch[0].Value.(json.RawMessage)
-	if !ok {
-		return nil, false
-	}
-	return result, true
-}