From 249d8db4092846357fc952aac974581ee9238ba4 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Vin=C3=ADcius=20Garcia?= <vingarcia00@gmail.com>
Date: Tue, 25 Jan 2022 11:32:03 -0300
Subject: [PATCH] Improve user_permissions table used in tests for allowing
 more interesting tests

---
 ksql_test.go | 16 ++++++++++------
 1 file changed, 10 insertions(+), 6 deletions(-)

diff --git a/ksql_test.go b/ksql_test.go
index ee05662..f69ff6a 100644
--- a/ksql_test.go
+++ b/ksql_test.go
@@ -2060,7 +2060,7 @@ func createTables(driver string) error {
 		)`)
 	}
 	if err != nil {
-		return fmt.Errorf("failed to create new users table: %s", err.Error())
+		return fmt.Errorf("failed to create new posts table: %s", err.Error())
 	}
 
 	db.Exec(`DROP TABLE user_permissions`)
@@ -2068,31 +2068,35 @@ func createTables(driver string) error {
 	switch driver {
 	case "sqlite3":
 		_, err = db.Exec(`CREATE TABLE user_permissions (
+		  id INTEGER PRIMARY KEY,
 		  user_id INTEGER,
 		  post_id INTEGER,
-			PRIMARY KEY (user_id, post_id)
+			UNIQUE (user_id, post_id)
 		)`)
 	case "postgres":
 		_, err = db.Exec(`CREATE TABLE user_permissions (
+		  id serial PRIMARY KEY,
 			user_id INT,
 		  post_id INT,
-			PRIMARY KEY (user_id, post_id)
+			UNIQUE (user_id, post_id)
 		)`)
 	case "mysql":
 		_, err = db.Exec(`CREATE TABLE user_permissions (
+			id INT AUTO_INCREMENT PRIMARY KEY,
 			user_id INT,
 			post_id INT,
-			PRIMARY KEY (user_id, post_id)
+			UNIQUE KEY (user_id, post_id)
 		)`)
 	case "sqlserver":
 		_, err = db.Exec(`CREATE TABLE user_permissions (
+			id INT IDENTITY(1,1) PRIMARY KEY,
 			user_id INT,
 			post_id INT,
-			PRIMARY KEY (user_id, post_id)
+			CONSTRAINT unique_1 UNIQUE (user_id, post_id)
 		)`)
 	}
 	if err != nil {
-		return fmt.Errorf("failed to create new users table: %s", err.Error())
+		return fmt.Errorf("failed to create new user_permissions table: %s", err.Error())
 	}
 
 	return nil