Test decimals

BemiHQ · Nov 5, 2024 · 4fc1fc2 · 4fc1fc2
1 parent ae9723d
commit 4fc1fc2
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 6 deletions.
diff --git a/src/init_test.go b/src/init_test.go
@@ -19,19 +19,27 @@ var TEST_PG_SCHEMA_COLUMNS = []PgSchemaColumn{
 		OrdinalPosition: "2",
 	},
 	{
-		ColumnName:      "value",
+		ColumnName:      "int_value",
 		UdtName:         "int4",
 		IsNullable:      "NO",
 		OrdinalPosition: "3",
 	},
+	{
+		ColumnName:       "decimal_value",
+		UdtName:          "numeric",
+		IsNullable:       "NO",
+		OrdinalPosition:  "4",
+		NumericPrecision: "10",
+		NumericScale:     "2",
+	},
 }
 
 var TEST_LOADED_ROWS = [][]string{
-	{"1", "metric_1", "5"},
-	{"2", "metric_2", "10"},
-	{"3", "metric_1", "5"},
-	{"4", "metric_2", "10"},
-	{"5", "metric_1", "5"},
+	{"1", "metric_1", "5", "5.0"},
+	{"2", "metric_2", "10", "10.0"},
+	{"3", "metric_1", "5", "5.0"},
+	{"4", "metric_2", "10", "10.0"},
+	{"5", "metric_1", "5", "5.0"},
 }
 
 func init() {

diff --git a/src/proxy_test.go b/src/proxy_test.go
@@ -63,6 +63,10 @@ func TestHandleQuery(t *testing.T) {
 			"description": {"count"},
 			"values":      {"5"},
 		},
+		"SELECT AVG(decimal_value) AS average FROM public.test_table": {
+			"description": {"average"},
+			"values":      {"7"},
+		},
 	}
 
 	for query, responses := range responsesByQuery {