more test cases for pretty print

Author: pyramation

__fixtures__/generated/generated.json

@@ -110,6 +110,9 @@
   "pretty/constraints-12.sql": "ALTER TABLE products ADD CONSTRAINT fk_category \n  FOREIGN KEY (category_id) \n  REFERENCES categories(id) \n  ON UPDATE CASCADE \n  ON DELETE SET NULL \n  DEFERRABLE INITIALLY DEFERRED",
   "pretty/constraints-13.sql": "ALTER TABLE products ADD CONSTRAINT check_price CHECK (price > 0)",
   "pretty/constraints-14.sql": "ALTER TABLE users ADD CONSTRAINT unique_email UNIQUE (email)",
+  "pretty/constraints-15.sql": "ALTER TABLE school.enrollments\n  ADD CONSTRAINT fk_student_course\n  FOREIGN KEY (student_id, course_id)\n  REFERENCES school.courses_students (student_id, course_id)",
+  "pretty/constraints-16.sql": "ALTER TABLE school.enrollments\n  ADD CONSTRAINT chk_enrollment_date\n  CHECK (\n    enrollment_date <= CURRENT_DATE\n    AND status IN ('active', 'completed', 'withdrawn')\n  )",
+  "pretty/constraints-17.sql": "CREATE TABLE school.enrollments (\n  student_id INT NOT NULL,\n  course_id INT NOT NULL,\n  enrollment_date DATE NOT NULL,\n  status TEXT CHECK (\n    status IN ('active', 'completed', 'withdrawn')\n  ),\n  CHECK (\n    enrollment_date <= CURRENT_DATE\n  )\n)",
   "pretty/casing-1.sql": "INSERT INTO users (name) VALUES ('John Doe')",
   "pretty/casing-2.sql": "INSERT INTO users (name) VALUES ('ADMINISTRATOR')",
   "pretty/casing-3.sql": "INSERT INTO users (name) VALUES ('lowercase')",

__fixtures__/kitchen-sink/pretty/constraints.sql

@@ -71,3 +71,32 @@ ALTER TABLE products ADD CONSTRAINT check_price CHECK (price > 0);
 ALTER TABLE users ADD CONSTRAINT unique_email UNIQUE (email);
 
 
+-- 15 
+
+ALTER TABLE school.enrollments
+  ADD CONSTRAINT fk_student_course
+  FOREIGN KEY (student_id, course_id)
+  REFERENCES school.courses_students (student_id, course_id);
+
+-- 16 
+
+ALTER TABLE school.enrollments
+  ADD CONSTRAINT chk_enrollment_date
+  CHECK (
+    enrollment_date <= CURRENT_DATE
+    AND status IN ('active', 'completed', 'withdrawn')
+  );
+
+-- 17
+
+CREATE TABLE school.enrollments (
+  student_id INT NOT NULL,
+  course_id INT NOT NULL,
+  enrollment_date DATE NOT NULL,
+  status TEXT CHECK (
+    status IN ('active', 'completed', 'withdrawn')
+  ),
+  CHECK (
+    enrollment_date <= CURRENT_DATE
+  )
+);

packages/deparser/__tests__/kitchen-sink/pretty-constraints.test.ts

@@ -17,6 +17,9 @@ it('pretty-constraints', async () => {
   "pretty/constraints-11.sql",
   "pretty/constraints-12.sql",
   "pretty/constraints-13.sql",
-  "pretty/constraints-14.sql"
+  "pretty/constraints-14.sql",
+  "pretty/constraints-15.sql",
+  "pretty/constraints-16.sql",
+  "pretty/constraints-17.sql"
 ]);
 });

packages/deparser/__tests__/pretty/__snapshots__/constraints-pretty.test.ts.snap

@@ -28,6 +28,12 @@ exports[`non-pretty: pretty/constraints-13.sql 1`] = `"ALTER TABLE products ADD
 
 exports[`non-pretty: pretty/constraints-14.sql 1`] = `"ALTER TABLE users ADD CONSTRAINT unique_email UNIQUE (email)"`;
 
+exports[`non-pretty: pretty/constraints-15.sql 1`] = `"ALTER TABLE school.enrollments ADD CONSTRAINT fk_student_course FOREIGN KEY (student_id, course_id) REFERENCES school.courses_students (student_id, course_id)"`;
+
+exports[`non-pretty: pretty/constraints-16.sql 1`] = `"ALTER TABLE school.enrollments ADD CONSTRAINT chk_enrollment_date CHECK (enrollment_date <= CURRENT_DATE AND status IN ('active', 'completed', 'withdrawn'))"`;
+
+exports[`non-pretty: pretty/constraints-17.sql 1`] = `"CREATE TABLE school.enrollments (student_id int NOT NULL, course_id int NOT NULL, enrollment_date date NOT NULL, status text CHECK (status IN ('active', 'completed', 'withdrawn')), CHECK (enrollment_date <= CURRENT_DATE))"`;
+
 exports[`pretty: pretty/constraints-1.sql 1`] = `
 "ALTER TABLE public.users 
   ADD CONSTRAINT users_pkey PRIMARY KEY (id)"
@@ -111,3 +117,24 @@ exports[`pretty: pretty/constraints-14.sql 1`] = `
 "ALTER TABLE users 
   ADD CONSTRAINT unique_email UNIQUE (email)"
 `;
+
+exports[`pretty: pretty/constraints-15.sql 1`] = `
+"ALTER TABLE school.enrollments 
+  ADD CONSTRAINT fk_student_course FOREIGN KEY (student_id, course_id) REFERENCES school.courses_students (student_id, course_id)"
+`;
+
+exports[`pretty: pretty/constraints-16.sql 1`] = `
+"ALTER TABLE school.enrollments 
+  ADD CONSTRAINT chk_enrollment_date CHECK (enrollment_date <= CURRENT_DATE
+    AND status IN ('active', 'completed', 'withdrawn'))"
+`;
+
+exports[`pretty: pretty/constraints-17.sql 1`] = `
+"CREATE TABLE school.enrollments (
+  student_id int NOT NULL,
+  course_id int NOT NULL,
+  enrollment_date date NOT NULL,
+  status text CHECK (status IN ('active', 'completed', 'withdrawn')),
+  CHECK (enrollment_date <= CURRENT_DATE)
+)"
+`;

packages/deparser/__tests__/pretty/constraints-pretty.test.ts

@@ -14,6 +14,9 @@ const prettyTest = new PrettyTest([
   'pretty/constraints-12.sql',
   'pretty/constraints-13.sql',
   'pretty/constraints-14.sql',
+  'pretty/constraints-15.sql',
+  'pretty/constraints-16.sql',
+  'pretty/constraints-17.sql',
 ]);
 
 prettyTest.generateTests();