Add variadic tests

tembo-io · Apr 12, 2024 · 78eec5b · 78eec5b
1 parent 8c950c6
commit 78eec5b
Show file tree

Hide file tree

Showing 4 changed files with 48 additions and 0 deletions.
diff --git a/test/expected/is_valid.out b/test/expected/is_valid.out
@@ -72,6 +72,19 @@ SELECT jsonschema_is_valid(:'user_schema_id', :'addr_schema'::jsonb, :'user_sche
  t
 (1 row)
 
+-- Variadic array
+SELECT jsonschema_is_valid(:'user_schema_id', VARIADIC ARRAY[:'addr_schema', :'user_schema']::json[]);
+ jsonschema_is_valid 
+---------------------
+ t
+(1 row)
+
+SELECT jsonschema_is_valid(:'user_schema_id', VARIADIC ARRAY[:'addr_schema', :'user_schema']::jsonb[]);
+ jsonschema_is_valid 
+---------------------
+ t
+(1 row)
+
 -- Invalid
 SELECT jsonschema_is_valid('foo', :'addr_schema'::json);
 INFO:  error loading file:///Users/david/.pgenv/pgsql-16.2/data/foo: unsupported scheme in file:///Users/david/.pgenv/pgsql-16.2/data/foo

diff --git a/test/expected/validates.out b/test/expected/validates.out
@@ -141,6 +141,31 @@ SELECT jsonschema_validates(:'user'::jsonb, :'user_schema_id', :'addr_schema'::j
  t
 (1 row)
 
+-- Variadic
+SELECT jsonschema_validates(:'user'::json, :'user_schema_id', VARIADIC ARRAY[:'addr_schema', :'user_schema']::json[] );
+ jsonschema_validates 
+----------------------
+ t
+(1 row)
+
+SELECT jsonschema_validates(:'user'::json, :'user_schema_id', VARIADIC ARRAY[:'addr_schema', :'user_schema']::jsonb[] );
+ jsonschema_validates 
+----------------------
+ t
+(1 row)
+
+SELECT jsonschema_validates(:'user'::jsonb, :'user_schema_id', VARIADIC ARRAY[:'addr_schema', :'user_schema']::jsonb[] );
+ jsonschema_validates 
+----------------------
+ t
+(1 row)
+
+SELECT jsonschema_validates(:'user'::jsonb, :'user_schema_id', VARIADIC ARRAY[:'addr_schema', :'user_schema']::json[] );
+ jsonschema_validates 
+----------------------
+ t
+(1 row)
+
 -- Invalid User
 \set invalid_user '{"username": "hello"}'
 SELECT jsonschema_validates(:'invalid_user'::json, :'user_schema_id', :'addr_schema'::json, :'user_schema'::json );

diff --git a/test/sql/is_valid.sql b/test/sql/is_valid.sql
@@ -24,6 +24,10 @@ SELECT jsonschema_is_valid(:'addr_schema_id', :'addr_schema'::jsonb);
 SELECT jsonschema_is_valid(:'user_schema_id', :'addr_schema'::json, :'user_schema'::json);
 SELECT jsonschema_is_valid(:'user_schema_id', :'addr_schema'::jsonb, :'user_schema'::jsonb);
 
+-- Variadic array
+SELECT jsonschema_is_valid(:'user_schema_id', VARIADIC ARRAY[:'addr_schema', :'user_schema']::json[]);
+SELECT jsonschema_is_valid(:'user_schema_id', VARIADIC ARRAY[:'addr_schema', :'user_schema']::jsonb[]);
+
 -- Invalid
 SELECT jsonschema_is_valid('foo', :'addr_schema'::json);
 SELECT jsonschema_is_valid('bar', :'addr_schema'::jsonb);

diff --git a/test/sql/validates.sql b/test/sql/validates.sql
@@ -59,6 +59,12 @@ SELECT jsonschema_validates(:'user'::json, :'user_schema_id', :'addr_schema'::js
 SELECT jsonschema_validates(:'user'::jsonb, :'user_schema_id', :'addr_schema'::jsonb, :'user_schema'::jsonb );
 SELECT jsonschema_validates(:'user'::jsonb, :'user_schema_id', :'addr_schema'::json, :'user_schema'::json );
 
+-- Variadic
+SELECT jsonschema_validates(:'user'::json, :'user_schema_id', VARIADIC ARRAY[:'addr_schema', :'user_schema']::json[] );
+SELECT jsonschema_validates(:'user'::json, :'user_schema_id', VARIADIC ARRAY[:'addr_schema', :'user_schema']::jsonb[] );
+SELECT jsonschema_validates(:'user'::jsonb, :'user_schema_id', VARIADIC ARRAY[:'addr_schema', :'user_schema']::jsonb[] );
+SELECT jsonschema_validates(:'user'::jsonb, :'user_schema_id', VARIADIC ARRAY[:'addr_schema', :'user_schema']::json[] );
+
 -- Invalid User
 \set invalid_user '{"username": "hello"}'
 SELECT jsonschema_validates(:'invalid_user'::json, :'user_schema_id', :'addr_schema'::json, :'user_schema'::json );