Add test on "rewrite" and if statements with regexps

Author: odiszapc

src/test/java/com/github/odiszapc/nginxparser/IfTest.java

@@ -0,0 +1,54 @@
+package com.github.odiszapc.nginxparser;
+
+import org.junit.Assert;
+import org.junit.Test;
+import java.util.Iterator;
+
+import static com.github.odiszapc.nginxparser.TestUtils.assertIfBlock;
+
+public class IfTest extends ParseTestBase {
+
+    @Test
+    public void testC1() throws Exception {
+        Iterator<NgxEntry> it = parse("if/c1.conf").getEntries().iterator();
+        NgxBlock block = (NgxBlock) it.next();
+        Iterator<NgxEntry> it2 = block.iterator();
+        assertIfBlock(it2.next(), "if", "$request_method", "=", "POST");
+        Assert.assertFalse(it.hasNext());
+    }
+
+    @Test
+    public void testC2() throws Exception {
+        assertIfBlock(extractIF("if/c2.conf"), "if", "$query_string", "!=", "\"\"");
+    }
+
+    @Test
+    public void testC3() throws Exception {
+        assertIfBlock(extractIF("if/c3.conf"), "if", "$request_uri", "!~", "\\/$");
+    }
+
+    @Test
+    public void testC4() throws Exception {
+        assertIfBlock(extractIF("if/c4.conf"), "if", "$request_uri", "~*",
+                "\"(sitemap(_index)?\\.xml(\\.gz)?|[a-z0-9_\\-]+-sitemap([0-9]+)?\\.xml(\\.gz)?)\"");
+    }
+
+    @Test
+    public void testC5() throws Exception {
+        assertIfBlock(extractIF("if/c5.conf"), "if", "$request_uri", "~*",
+                "(sitemap(_index)?\\.xml(\\.gz)?|[a-z0-9_\\-]+-sitemap([0-9]+)?\\.xml(\\.gz)?)");
+    }
+
+    @Test
+    public void testC6() throws Exception {
+        assertIfBlock(extractIF("if/c6.conf"), "if", "-f",
+                "\"$document_root/wp-content/cache/page_enhanced/$host/$request_uri/_index$w3tc_ua$w3tc_ref$w3tc_ssl.html$w3tc_enc\"");
+    }
+
+    private NgxEntry extractIF(String path) throws Exception {
+        Iterator<NgxEntry> it = parse(path).getEntries().iterator();
+        NgxBlock block = (NgxBlock) it.next();
+        Iterator<NgxEntry> it2 = block.iterator();
+        return it2.next();
+    }
+}

src/test/java/com/github/odiszapc/nginxparser/ParseTestBase.java

@@ -1,7 +1,14 @@
 package com.github.odiszapc.nginxparser;
 
+import java.util.Iterator;
+
 public class ParseTestBase {
     protected NgxConfig parse(String path) throws Exception {
         return TestUtils.parseAntlr(path);
     }
+
+    protected NgxEntry getFirstParam(String path) throws Exception {
+        Iterator<NgxEntry> it = parse(path).getEntries().iterator();
+        return it.next();
+    }
 }

src/test/java/com/github/odiszapc/nginxparser/RewriteTest.java

@@ -0,0 +1,36 @@
+package com.github.odiszapc.nginxparser;
+
+import org.junit.Test;
+
+public class RewriteTest extends ParseTestBase {
+
+    @Test
+    public void testC1() throws Exception {
+        TestUtils.assertParam(getFirstParam("rewrite/c1.conf"), "rewrite", "^/dashboard$", "/admin.php?controller=user&action=login", "last");
+    }
+
+    @Test
+    public void testС2() throws Exception {
+        TestUtils.assertParam(getFirstParam("rewrite/c2.conf"), "rewrite", "^/feed", "/feed.php", "last");
+    }
+
+    @Test
+    public void testС3() throws Exception {
+        TestUtils.assertParam(getFirstParam("rewrite/c3.conf"), "rewrite", "^/category/([^/]+)/page-([0-9]+)$", "/index.php?controller=blog&action=view&category=$1&page=$2", "last");
+    }
+
+    @Test
+    public void testС4() throws Exception {
+        TestUtils.assertParam(getFirstParam("rewrite/c4.conf"), "rewrite", "^/category/([^/]+)/$", "/index.php?controller=blog&action=view&category=$1&page=0", "last");
+    }
+
+    @Test
+    public void testС5() throws Exception {
+        TestUtils.assertParam(getFirstParam("rewrite/c5.conf"), "rewrite", "^/page-([0-9]+)$", "/index.php?controller=blog&action=view&page=$1", "last");
+    }
+
+    @Test
+    public void testС6() throws Exception {
+        TestUtils.assertParam(getFirstParam("rewrite/c6.conf"), "rewrite", "^", "/index.php?controller=post&action=view&id_post=$1", "last");
+    }
+}

src/test/java/com/github/odiszapc/nginxparser/TestUtils.java

@@ -65,6 +65,19 @@ public static NgxBlock assertBlock(NgxEntry entry, String name, String... values
         return (NgxBlock) entry;
     }
 
+    public static NgxIfBlock assertIfBlock(NgxEntry entry, String name, String... values) {
+        Assert.assertTrue(entry instanceof NgxBlock);
+        Assert.assertEquals(((NgxIfBlock) entry).getName(), name);
+
+        Assert.assertEquals(values.length, ((NgxBlock) entry).getValues().size());
+        Iterator<String> it = ((NgxBlock) entry).getValues().iterator();
+        for (String value : values) {
+            Assert.assertEquals(it.next(), value);
+        }
+
+        return (NgxIfBlock) entry;
+    }
+
     @SuppressWarnings("UnusedReturnValue")
     public static NgxComment assertComment(NgxEntry entry, @SuppressWarnings("SameParameterValue") String value) {
         Assert.assertTrue(entry instanceof NgxComment);

src/test/resources/if/c1.conf

@@ -0,0 +1,3 @@
+block {
+  if ($request_method = POST) { param 0;}
+}

src/test/resources/if/c2.conf

@@ -0,0 +1,3 @@
+block {
+  if ($query_string != "") { param 0;}
+}

src/test/resources/if/c3.conf

@@ -0,0 +1,3 @@
+block {
+  if ($request_uri !~ \/$) { param 0;}
+}

src/test/resources/if/c4.conf

@@ -0,0 +1,3 @@
+block {
+  if ($request_uri ~* "(sitemap(_index)?\.xml(\.gz)?|[a-z0-9_\-]+-sitemap([0-9]+)?\.xml(\.gz)?)") { param 0;}
+}

src/test/resources/if/c5.conf

@@ -0,0 +1,3 @@
+block {
+  if ($request_uri ~* (sitemap(_index)?\.xml(\.gz)?|[a-z0-9_\-]+-sitemap([0-9]+)?\.xml(\.gz)?)) { param 0;}
+}

src/test/resources/if/c6.conf

@@ -0,0 +1,3 @@
+block {
+  if (-f "$document_root/wp-content/cache/page_enhanced/$host/$request_uri/_index$w3tc_ua$w3tc_ref$w3tc_ssl.html$w3tc_enc") { param 0;}
+}

src/test/resources/rewrite/c1.conf

@@ -0,0 +1 @@
+rewrite ^/dashboard$ /admin.php?controller=user&action=login last;

src/test/resources/rewrite/c2.conf

@@ -0,0 +1 @@
+rewrite ^/feed /feed.php last;

src/test/resources/rewrite/c3.conf

@@ -0,0 +1 @@
+rewrite ^/category/([^/]+)/page-([0-9]+)$ /index.php?controller=blog&action=view&category=$1&page=$2 last;

src/test/resources/rewrite/c4.conf

@@ -0,0 +1 @@
+rewrite ^/category/([^/]+)/$ /index.php?controller=blog&action=view&category=$1&page=0 last;

src/test/resources/rewrite/c5.conf

@@ -0,0 +1 @@
+rewrite ^/page-([0-9]+)$ /index.php?controller=blog&action=view&page=$1 last;

src/test/resources/rewrite/c6.conf

@@ -0,0 +1 @@
+rewrite ^ /index.php?controller=post&action=view&id_post=$1 last;