- [+] add more tests

suntong · suntong · commit 92fb160f3def · 2023-06-29T12:12:22.000-04:00
diff --git a/piece_test.go b/piece_test.go
@@ -4,16 +4,30 @@ import (
 	"testing"
 )
 
-func TestPieceAttr(t *testing.T) {
+func TestPiece(t *testing.T) {
+	testCases(t, "Piece raw html", []testCase{
+		{"piece_raw_a",
+			[]string{"-i", "opt_piece_attr.html", "-o", "-c", "ul > li", "-p", "Source=RAW:a"},
+		},
+	})
 	testCases(t, "Piece Attributes", []testCase{
 		{"id",
-			[]string{"-i", "opt_piece_attr.html", "-o", "-c", "li > a", "-p", "url=ATTR:id"},
+			[]string{"-i", "opt_piece_attr.html", "-o", "-c", "li > a", "-p", "id=ATTR:id"},
 		},
 		{"href",
 			[]string{"-i", "opt_piece_attr.html", "-o", "-c", "li > a", "-p", "url=ATTR:href"},
 		},
-		//{"both_fields",},
-		//{"noexist"},
+		{"both_fields",
+			[]string{"-i", "opt_piece_attr.html", "-o", "-c", "li > a",
+				"-p", "id=ATTR:id", "-p", "url=ATTR:href"},
+		},
+
+		{"noexist",
+			[]string{"-i", "opt_piece_attr.html", "-o", "-c", "li > a", "-p", "NoExist=ATTR:noexist"},
+		},
+		{"script_src",
+			[]string{"-i", "opt_piece_script.html", "-o", "-c", "html > head > script", "-p", "SourceJS=ATTR:src"},
+		},
 		//{},
 	})
 }
diff --git a/test/both_fields.ref b/test/both_fields.ref
@@ -0,0 +1,6 @@
+id	url	
+a1	http://www.google.com/finance	
+a2	http://finance.yahoo.com/	
+a3	https://www.google.com/news	
+a4	http://news.yahoo.com	
+
diff --git a/test/id.ref b/test/id.ref
@@ -1,4 +1,4 @@
-url	
+id	
 a1	
 a2	
 a3	
diff --git a/test/noexist.ref b/test/noexist.ref
@@ -0,0 +1,6 @@
+NoExist	
+	
+	
+	
+	
+
diff --git a/test/opt_piece_script.html b/test/opt_piece_script.html
@@ -0,0 +1,7 @@
+<html>
+<head>
+  <script src="foo.js"></script>
+  <script src="bar.js"></script>
+  <script src="baz.js"></script>
+</head>
+</html>
diff --git a/test/piece_raw_a.ref b/test/piece_raw_a.ref
@@ -0,0 +1,9 @@
+Source	
+<a id="a1" href="http://www.google.com/finance">
+                </a>	
+<a id="a2" href="http://finance.yahoo.com/">
+                </a>	
+<a id="a3" href="https://www.google.com/news"></a>	
+<a id="a4" href="http://news.yahoo.com">
+</a>	
+
diff --git a/test/script_src.ref b/test/script_src.ref
@@ -0,0 +1,5 @@
+SourceJS	
+foo.js	
+bar.js	
+baz.js	
+

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,4 @@`
`1`		`-url`
	`1`	`+id`
`2`	`2`	`a1`
`3`	`3`	`a2`
`4`	`4`	`a3`
-Original file line number
+Diff line change
@@ @@ -0,0 +1,6 @@ @@
 +NoExist
++
++
++
++
++