From d7bf645504afb334e52c247e5ca44eb0dcb9a777 Mon Sep 17 00:00:00 2001
From: Matthew Bucci <buccim2@gene.com>
Date: Sun, 26 Nov 2023 15:24:45 -0800
Subject: [PATCH] handle regular key expressions

---
 tests/rewriteTest.php | 42 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 42 insertions(+)

diff --git a/tests/rewriteTest.php b/tests/rewriteTest.php
index 9d417df..e2d2ad7 100644
--- a/tests/rewriteTest.php
+++ b/tests/rewriteTest.php
@@ -123,5 +123,47 @@ final class rewriteTest extends TestCase
         $this->assertSame(trim($postgresql), trim($expected));
     }
 
+    public function test_it_handles_keys_without_unique() 
+    {
+        $sql = <<<SQL
+            CREATE TABLE wp_itsec_vulnerabilities (
+                id varchar(128) NOT NULL,
+                software_type varchar(20) NOT NULL,
+                software_slug varchar(255) NOT NULL,
+                first_seen timestamp NOT NULL,
+                last_seen timestamp NOT NULL,
+                resolved_at timestamp default NULL,
+                resolved_by bigint NOT NULL default 0,
+                resolution varchar(20) NOT NULL default '',
+                details text NOT NULL,
+                PRIMARY KEY (id),
+                KEY resolution (resolution),
+                KEY software_type (software_type),
+                KEY last_seen (last_seen)
+            )
+        SQL;
+        
+        $expected = <<<SQL
+            CREATE TABLE wp_itsec_vulnerabilities (
+                id varchar(128) NOT NULL,
+                software_type varchar(20) NOT NULL,
+                software_slug varchar(255) NOT NULL,
+                first_seen timestamp NOT NULL,
+                last_seen timestamp NOT NULL,
+                resolved_at timestamp default NULL,
+                resolved_by bigint NOT NULL default 0,
+                resolution varchar(20) NOT NULL default '',
+                details text NOT NULL,
+                PRIMARY KEY (id)
+            );
+        CREATE INDEX wp_itsec_vulnerabilities_resolution ON wp_itsec_vulnerabilities (resolution);
+        CREATE INDEX wp_itsec_vulnerabilities_software_type ON wp_itsec_vulnerabilities (software_type);
+        CREATE INDEX wp_itsec_vulnerabilities_last_seen ON wp_itsec_vulnerabilities (last_seen);
+        SQL;
+
+        $postgresql = pg4wp_rewrite($sql);
+        $this->assertSame(trim($postgresql), trim($expected));
+    }
+
   
 }
\ No newline at end of file