From 05b3b6ca2e85f612dd017d8ad2a94ab229364788 Mon Sep 17 00:00:00 2001
From: Shreyas Goenka <shreyas.goenka@databricks.com>
Date: Wed, 17 May 2023 14:02:41 +0200
Subject: [PATCH] added tests for castfloat

---
 libs/template/schema_test.go | 68 ++++++++++++++++++++++++++++++++++++
 1 file changed, 68 insertions(+)
 create mode 100644 libs/template/schema_test.go

diff --git a/libs/template/schema_test.go b/libs/template/schema_test.go
new file mode 100644
index 00000000..962262b1
--- /dev/null
+++ b/libs/template/schema_test.go
@@ -0,0 +1,68 @@
+package template
+
+import (
+	"encoding/json"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+)
+
+func TestTemplateSchematIsInterger(t *testing.T) {
+	assert.False(t, isIntegerValue(1.1))
+	assert.False(t, isIntegerValue(0.1))
+	assert.False(t, isIntegerValue(-0.1))
+
+	assert.True(t, isIntegerValue(-1.0))
+	assert.True(t, isIntegerValue(0.0))
+	assert.True(t, isIntegerValue(2.0))
+}
+
+func TestTemplateSchemaCastFloatToInt(t *testing.T) {
+	// define schema for config
+	schemaJson := `{
+		"int_val": {
+			"type": "integer"
+		},
+		"float_val": {
+			"type": "float"
+		},
+		"bool_val": {
+			"type": "boolean"
+		},
+		"string_val": {
+			"type": "string"
+		}
+	}`
+	var schema Schema
+	err := json.Unmarshal([]byte(schemaJson), &schema)
+	require.NoError(t, err)
+
+	// define the config
+	configJson := `{
+		"int_val":    1,
+		"float_val":  2,
+		"bool_val":   true,
+		"string_val": "main hoon na"
+	}`
+	var config map[string]any
+	err = json.Unmarshal([]byte(configJson), &config)
+	require.NoError(t, err)
+
+	// assert types before casting, checking that the integer was indeed loaded
+	// as a floating point
+	assert.IsType(t, float64(0), config["int_val"])
+	assert.IsType(t, float64(0), config["float_val"])
+	assert.IsType(t, true, config["bool_val"])
+	assert.IsType(t, "abc", config["string_val"])
+
+	err = schema.CastFloatToInt(config)
+	require.NoError(t, err)
+
+	// assert type after casting, that the float value was converted to an integer
+	// for int_val.
+	assert.IsType(t, int(0), config["int_val"])
+	assert.IsType(t, float64(0), config["float_val"])
+	assert.IsType(t, true, config["bool_val"])
+	assert.IsType(t, "abc", config["string_val"])
+}