@rebasepro/studio 0.11.1-canary.gfd39654 → 0.12.0

package/src/utils/pgColumnToProperty.test.ts

@@ -1,402 +0,0 @@
-import { buildCollectionFromTableMetadata } from "./pgColumnToProperty";
-import { StringProperty, NumberProperty, DateProperty, MapProperty, ArrayProperty, BooleanProperty, TableColumnInfo, TableMetadata } from "@rebasepro/types";
-
-describe("pgColumnToProperty Inference Logic", () => {
-    describe("String Data Types", () => {
-        it("maps character varying to StringProperty with varchar columnType", () => {
-            const columns: TableColumnInfo[] = [
-                { column_name: "first_name",
-data_type: "character varying",
-udt_name: "varchar",
-is_nullable: "NO",
-column_default: null,
-character_maximum_length: 255 }
-            ];
-            const collection = buildCollectionFromTableMetadata("users", { columns } as TableMetadata);
-            const prop = collection.properties!.first_name as StringProperty;
-            expect(prop.type).toBe("string");
-            expect(prop.columnType).toBe("varchar");
-            expect(prop.validation?.required).toBe(true);
-        });
-
-        it("maps text and citext to StringProperty with text columnType", () => {
-            const columns: TableColumnInfo[] = [
-                { column_name: "bio",
-data_type: "text",
-udt_name: "text",
-is_nullable: "YES",
-column_default: null,
-character_maximum_length: null },
-                { column_name: "username",
-data_type: "citext",
-udt_name: "citext",
-is_nullable: "YES",
-column_default: null,
-character_maximum_length: null }
-            ];
-            const collection = buildCollectionFromTableMetadata("users", { columns } as TableMetadata);
-
-            const bioProp = collection.properties!.bio as StringProperty;
-            expect(bioProp.type).toBe("string");
-            expect(bioProp.columnType).toBe("text");
-            expect(bioProp.validation?.required).toBeUndefined();
-
-            const usernameProp = collection.properties!.username as StringProperty;
-            expect(usernameProp.type).toBe("string");
-            expect(usernameProp.columnType).toBe("text");
-        });
-
-        it("maps char and character to StringProperty with char columnType", () => {
-            const columns: TableColumnInfo[] = [
-                { column_name: "country_code",
-data_type: "char",
-udt_name: "char",
-is_nullable: "NO",
-column_default: null,
-character_maximum_length: 2 }
-            ];
-            const collection = buildCollectionFromTableMetadata("locations", { columns } as TableMetadata);
-
-            const prop = collection.properties!.country_code as StringProperty;
-            expect(prop.type).toBe("string");
-            expect(prop.columnType).toBe("char");
-        });
-
-        it("maps uuid without generation strategy to a standard string property", () => {
-            const columns: TableColumnInfo[] = [
-                { column_name: "token",
-data_type: "uuid",
-udt_name: "uuid",
-is_nullable: "NO",
-column_default: null,
-character_maximum_length: null }
-            ];
-            const collection = buildCollectionFromTableMetadata("sessions", { columns } as TableMetadata);
-
-            const prop = collection.properties!.token as StringProperty;
-            expect(prop.type).toBe("string");
-            expect(prop.isId).toBeUndefined();
-        });
-    });
-
-    describe("Numeric Data Types", () => {
-        it("maps integer and smallint to NumberProperty with integer columnType and validation", () => {
-            const columns: TableColumnInfo[] = [
-                { column_name: "age",
-data_type: "integer",
-udt_name: "int4",
-is_nullable: "NO",
-column_default: null,
-character_maximum_length: null },
-                { column_name: "status_code",
-data_type: "smallint",
-udt_name: "int2",
-is_nullable: "YES",
-column_default: null,
-character_maximum_length: null }
-            ];
-            const collection = buildCollectionFromTableMetadata("users", { columns } as TableMetadata);
-
-            const ageProp = collection.properties!.age as NumberProperty;
-            expect(ageProp.type).toBe("number");
-            expect(ageProp.columnType).toBe("integer");
-            expect(ageProp.validation?.integer).toBe(true);
-            expect(ageProp.validation?.required).toBe(true);
-
-            const statusProp = collection.properties!.status_code as NumberProperty;
-            expect(statusProp.type).toBe("number");
-            expect(statusProp.columnType).toBe("integer");
-            expect(statusProp.validation?.integer).toBe(true);
-            expect(statusProp.validation?.required).toBeUndefined();
-        });
-
-        it("maps bigint to NumberProperty with bigint columnType", () => {
-            const columns: TableColumnInfo[] = [
-                { column_name: "view_count",
-data_type: "bigint",
-udt_name: "int8",
-is_nullable: "YES",
-column_default: null,
-character_maximum_length: null }
-            ];
-            const collection = buildCollectionFromTableMetadata("posts", { columns } as TableMetadata);
-            const prop = collection.properties!.view_count as NumberProperty;
-            expect(prop.type).toBe("number");
-            expect(prop.columnType).toBe("bigint");
-            expect(prop.validation?.integer).toBe(true);
-        });
-
-        it("maps real, double precision, numeric, decimal correctly", () => {
-            const columns: TableColumnInfo[] = [
-                { column_name: "price",
-data_type: "numeric",
-udt_name: "numeric",
-is_nullable: "YES",
-column_default: null,
-character_maximum_length: null },
-                { column_name: "rating",
-data_type: "real",
-udt_name: "float4",
-is_nullable: "YES",
-column_default: null,
-character_maximum_length: null },
-                { column_name: "precision_val",
-data_type: "double precision",
-udt_name: "float8",
-is_nullable: "YES",
-column_default: null,
-character_maximum_length: null }
-            ];
-            const collection = buildCollectionFromTableMetadata("products", { columns } as TableMetadata);
-
-            expect((collection.properties!.price as NumberProperty).columnType).toBe("numeric");
-            expect((collection.properties!.rating as NumberProperty).columnType).toBe("real");
-            expect((collection.properties!.precision_val as NumberProperty).columnType).toBe("double precision");
-
-            // Decimal shouldn't have integer explicitly forced
-            expect((collection.properties!.price as NumberProperty).validation?.integer).toBeUndefined();
-        });
-    });
-
-    describe("Primary Keys & Auto-ID Handling", () => {
-        it("detects identity properties and marks them as incrementing IDs", () => {
-            const columns: TableColumnInfo[] = [
-                { column_name: "id",
-data_type: "integer",
-udt_name: "int4",
-is_nullable: "NO",
-column_default: "nextval('users_id_seq'::regclass)",
-character_maximum_length: null }
-            ];
-            const collection = buildCollectionFromTableMetadata("users", { columns } as TableMetadata);
-
-            const prop = collection.properties!.id as NumberProperty;
-            expect(prop.isId).toBe("increment");
-        });
-
-        it("detects serial to automatically be an incrementing ID", () => {
-            const columns: TableColumnInfo[] = [
-                { column_name: "id",
-data_type: "serial",
-udt_name: "int4",
-is_nullable: "NO",
-column_default: null,
-character_maximum_length: null },
-                { column_name: "big_id",
-data_type: "bigserial",
-udt_name: "int8",
-is_nullable: "NO",
-column_default: null,
-character_maximum_length: null }
-            ];
-            const collection = buildCollectionFromTableMetadata("users", { columns } as TableMetadata);
-
-            expect((collection.properties!.id as NumberProperty).isId).toBe("increment");
-            expect((collection.properties!.id as NumberProperty).columnType).toBe("serial");
-            expect((collection.properties!.big_id as NumberProperty).isId).toBe("increment");
-            expect((collection.properties!.big_id as NumberProperty).columnType).toBe("bigserial");
-        });
-
-        it("detects gen_random_uuid / uuid_generate as a uuid ID strategy", () => {
-            const columns: TableColumnInfo[] = [
-                { column_name: "id",
-data_type: "uuid",
-udt_name: "uuid",
-is_nullable: "NO",
-column_default: "gen_random_uuid()",
-character_maximum_length: null }
-            ];
-            const collection = buildCollectionFromTableMetadata("users", { columns } as TableMetadata);
-
-            const prop = collection.properties!.id as StringProperty;
-            expect(prop.isId).toBe("uuid");
-        });
-
-        it("detects auto-generated string IDs as manual fallback strategy if not a UUID column type", () => {
-            const columns: TableColumnInfo[] = [
-                // If it's varchar with a random identity function that Rebase doesn't purely know how to recreate natively
-                { column_name: "id",
-data_type: "varchar",
-udt_name: "varchar",
-is_nullable: "NO",
-column_default: "some_custom_id_generator()",
-character_maximum_length: null }
-            ];
-            const collection = buildCollectionFromTableMetadata("users", { columns } as TableMetadata);
-
-            const prop = collection.properties!.id as StringProperty;
-            expect(prop.isId).toBeUndefined(); // Wait, the current logic maps it as `undefined` if column_default doesn't include "nextval" / "identity" / "uuid" etc. Let's adjust test according to how we wrote pgColumnToProperty... Wait, actually our logic handles `column_default.includes('identity')` or `uuid_generate`. So custom generators fail to parse as `isAutoId=true`.
-            // The expectation checks the *current* engine behavior.
-        });
-    });
-
-    describe("Special Types (Date, JSON, Arrays, Boolean, Enums)", () => {
-        it("maps boolean fields", () => {
-            const columns: TableColumnInfo[] = [
-                { column_name: "is_active",
-data_type: "boolean",
-udt_name: "bool",
-is_nullable: "NO",
-column_default: "true",
-character_maximum_length: null }
-            ];
-            const collection = buildCollectionFromTableMetadata("users", { columns } as TableMetadata);
-            const prop = collection.properties!.is_active as BooleanProperty;
-            expect(prop.type).toBe("boolean");
-        });
-
-        it("maps timestamp and date correctly", () => {
-            const columns: TableColumnInfo[] = [
-                { column_name: "created_at",
-data_type: "timestamp with time zone",
-udt_name: "timestamptz",
-is_nullable: "NO",
-column_default: "now()",
-character_maximum_length: null },
-                { column_name: "birthday",
-data_type: "date",
-udt_name: "date",
-is_nullable: "YES",
-column_default: null,
-character_maximum_length: null },
-                { column_name: "shift_start",
-data_type: "time",
-udt_name: "time",
-is_nullable: "YES",
-column_default: null,
-character_maximum_length: null }
-            ];
-            const collection = buildCollectionFromTableMetadata("events", { columns } as TableMetadata);
-
-            expect((collection.properties!.created_at as DateProperty).columnType).toBe("timestamp");
-            expect((collection.properties!.birthday as DateProperty).columnType).toBe("date");
-            expect((collection.properties!.shift_start as DateProperty).columnType).toBe("time");
-        });
-
-        it("maps jsonb and json to MapProperty", () => {
-            const columns: TableColumnInfo[] = [
-                { column_name: "metadata",
-data_type: "jsonb",
-udt_name: "jsonb",
-is_nullable: "YES",
-column_default: null,
-character_maximum_length: null },
-                { column_name: "old_data",
-data_type: "json",
-udt_name: "json",
-is_nullable: "YES",
-column_default: null,
-character_maximum_length: null }
-            ];
-            const collection = buildCollectionFromTableMetadata("docs", { columns } as TableMetadata);
-
-            expect((collection.properties!.metadata as MapProperty).type).toBe("map");
-            expect((collection.properties!.metadata as MapProperty).columnType).toBe("jsonb");
-            expect((collection.properties!.metadata as MapProperty).keyValue).toBe(true);
-
-            expect((collection.properties!.old_data as MapProperty).type).toBe("map");
-            expect((collection.properties!.old_data as MapProperty).columnType).toBe("json");
-        });
-
-        it("maps array to ArrayProperty", () => {
-            const columns: TableColumnInfo[] = [
-                { column_name: "tags",
-data_type: "ARRAY",
-udt_name: "_varchar",
-is_nullable: "YES",
-column_default: null,
-character_maximum_length: null }
-            ];
-            const collection = buildCollectionFromTableMetadata("posts", { columns } as TableMetadata);
-
-            const prop = collection.properties!.tags as ArrayProperty;
-            expect(prop.type).toBe("array");
-            expect(prop.columnType).toBe("text[]");
-            expect(!Array.isArray(prop.of) && prop.of?.type).toBe("string");
-        });
-
-        it("maps USER-DEFINED enums properly to StringProperty with enum configs", () => {
-            const columns: TableColumnInfo[] = [
-                {
-                    column_name: "role",
-                    data_type: "USER-DEFINED",
-                    udt_name: "user_role_enum",
-                    is_nullable: "NO",
-                    column_default: null,
-                    character_maximum_length: null,
-                    enum_values: ["admin", "super_admin", "editor"]
-                }
-            ];
-            const collection = buildCollectionFromTableMetadata("users", { columns } as TableMetadata);
-
-            const prop = collection.properties!.role as StringProperty;
-            expect(prop.type).toBe("string");
-            expect(prop.validation?.required).toBe(true);
-            expect(prop.enum).toEqual([
-                { id: "admin",
-label: "Admin" },
-                { id: "super_admin",
-label: "Super Admin" },
-                { id: "editor",
-label: "Editor" }
-            ]);
-        });
-    });
-
-    describe("Property Fallbacks & Formatting", () => {
-        it("prettifies property names automatically", () => {
-            const columns: TableColumnInfo[] = [
-                { column_name: "user_first_name",
-data_type: "varchar",
-udt_name: "varchar",
-is_nullable: "YES",
-column_default: null,
-character_maximum_length: null }
-            ];
-            const collection = buildCollectionFromTableMetadata("users", { columns } as TableMetadata);
-
-            expect(collection.properties!.user_first_name.name).toBe("User First Name");
-        });
-
-        it("assigns propertiesOrder array based on iteration insertion order", () => {
-            const columns: TableColumnInfo[] = [
-                { column_name: "id",
-data_type: "integer",
-udt_name: "int4",
-is_nullable: "NO",
-column_default: null,
-character_maximum_length: null },
-                { column_name: "email",
-data_type: "varchar",
-udt_name: "varchar",
-is_nullable: "YES",
-column_default: null,
-character_maximum_length: null },
-                { column_name: "role",
-data_type: "varchar",
-udt_name: "varchar",
-is_nullable: "YES",
-column_default: null,
-character_maximum_length: null }
-            ];
-            const collection = buildCollectionFromTableMetadata("users", { columns } as TableMetadata);
-
-            expect(collection.propertiesOrder).toEqual(["id", "email", "role"]);
-        });
-
-        it("maps unrecognized data types to standard string strings", () => {
-            const columns: TableColumnInfo[] = [
-                { column_name: "weird_col",
-data_type: "macaddr",
-udt_name: "macaddr",
-is_nullable: "NO",
-column_default: null,
-character_maximum_length: null }
-            ];
-            const collection = buildCollectionFromTableMetadata("networks", { columns } as TableMetadata);
-
-            const prop = collection.properties!.weird_col as StringProperty;
-            expect(prop.type).toBe("string");
-        });
-    });
-});