| 1 | require "test/unit/testcase" |
|---|
| 2 | |
|---|
| 3 | require "rexml/document" |
|---|
| 4 | |
|---|
| 5 | class FunctionsTester < Test::Unit::TestCase |
|---|
| 6 | include REXML |
|---|
| 7 | def test_functions |
|---|
| 8 | # trivial text() test |
|---|
| 9 | # confuse-a-function |
|---|
| 10 | source = "<a>more <b id='1'/><b id='2'>dumb</b><b id='3'/><c/> text</a>" |
|---|
| 11 | doc = Document.new source |
|---|
| 12 | res = "" |
|---|
| 13 | XPath::each(doc.root, "text()") {|val| res << val.to_s} |
|---|
| 14 | assert_equal "more text", res |
|---|
| 15 | |
|---|
| 16 | res = XPath::first(doc.root, "b[last()]") |
|---|
| 17 | assert_equal '3', res.attributes['id'] |
|---|
| 18 | res = XPath::first(doc.root, "b[position()=2]") |
|---|
| 19 | assert_equal '2', res.attributes['id'] |
|---|
| 20 | res = XPath::first(doc.root, "*[name()='c']") |
|---|
| 21 | assert_equal "c", res.name |
|---|
| 22 | end |
|---|
| 23 | |
|---|
| 24 | # Contributed by Mike Stok |
|---|
| 25 | def test_starts_with |
|---|
| 26 | source = <<-EOF |
|---|
| 27 | <foo> |
|---|
| 28 | <a href="mailto:a@b.c">a@b.c</a> |
|---|
| 29 | <a href="http://www.foo.com">http://www.foo.com</a> |
|---|
| 30 | </foo> |
|---|
| 31 | EOF |
|---|
| 32 | doc = Document.new source |
|---|
| 33 | mailtos = doc.elements.to_a("//a[starts-with(@href, 'mailto:')]") |
|---|
| 34 | assert_equal 1, mailtos.size |
|---|
| 35 | assert_equal "mailto:a@b.c", mailtos[0].attributes['href'] |
|---|
| 36 | |
|---|
| 37 | ailtos = doc.elements.to_a("//a[starts-with(@href, 'ailto:')]") |
|---|
| 38 | assert_equal 0, ailtos.size |
|---|
| 39 | end |
|---|
| 40 | |
|---|
| 41 | def test_string_length |
|---|
| 42 | doc = Document.new <<-EOF |
|---|
| 43 | <AAA> |
|---|
| 44 | <Q/> |
|---|
| 45 | <SSSS/> |
|---|
| 46 | <BB/> |
|---|
| 47 | <CCC/> |
|---|
| 48 | <DDDDDDDD/> |
|---|
| 49 | <EEEE/> |
|---|
| 50 | </AAA> |
|---|
| 51 | EOF |
|---|
| 52 | assert doc, "create doc" |
|---|
| 53 | |
|---|
| 54 | set = doc.elements.to_a("//*[string-length(name()) = 3]") |
|---|
| 55 | assert_equal 2, set.size, "nodes with names length = 3" |
|---|
| 56 | |
|---|
| 57 | set = doc.elements.to_a("//*[string-length(name()) < 3]") |
|---|
| 58 | assert_equal 2, set.size, "nodes with names length < 3" |
|---|
| 59 | |
|---|
| 60 | set = doc.elements.to_a("//*[string-length(name()) > 3]") |
|---|
| 61 | assert_equal 3, set.size, "nodes with names length > 3" |
|---|
| 62 | end |
|---|
| 63 | |
|---|
| 64 | # Test provided by Mike Stok |
|---|
| 65 | def test_contains |
|---|
| 66 | source = <<-EOF |
|---|
| 67 | <foo> |
|---|
| 68 | <a href="mailto:a@b.c">a@b.c</a> |
|---|
| 69 | <a href="http://www.foo.com">http://www.foo.com</a> |
|---|
| 70 | </foo> |
|---|
| 71 | EOF |
|---|
| 72 | doc = Document.new source |
|---|
| 73 | |
|---|
| 74 | [['o', 2], ['foo', 1], ['bar', 0]].each { |test| |
|---|
| 75 | search, expected = test |
|---|
| 76 | set = doc.elements.to_a("//a[contains(@href, '#{search}')]") |
|---|
| 77 | assert_equal expected, set.size |
|---|
| 78 | } |
|---|
| 79 | end |
|---|
| 80 | |
|---|
| 81 | # Mike Stok and Sean Russell |
|---|
| 82 | def test_substring |
|---|
| 83 | # examples from http://www.w3.org/TR/xpath#function-substring |
|---|
| 84 | doc = Document.new('<test string="12345" />') |
|---|
| 85 | |
|---|
| 86 | d = Document.new("<a b='1'/>") |
|---|
| 87 | #puts XPath.first(d, 'node()[0 + 1]') |
|---|
| 88 | #d = Document.new("<a b='1'/>") |
|---|
| 89 | #puts XPath.first(d, 'a[0 mod 0]') |
|---|
| 90 | [ [1.5, 2.6, '234'], |
|---|
| 91 | [0, 3, '12'], |
|---|
| 92 | [0, '0 div 0', ''], |
|---|
| 93 | [1, '0 div 0', ''], |
|---|
| 94 | ['-42', '1 div 0', '12345'], |
|---|
| 95 | ['-1 div 0', '1 div 0', ''] |
|---|
| 96 | ].each { |start, length, expected| |
|---|
| 97 | set = doc.elements.to_a("//test[substring(@string, #{start}, #{length}) = '#{expected}']") |
|---|
| 98 | assert_equal 1, set.size, "#{start}, #{length}, '#{expected}'" |
|---|
| 99 | } |
|---|
| 100 | end |
|---|
| 101 | |
|---|
| 102 | def test_substring_angrez |
|---|
| 103 | testString = REXML::Functions::substring_after("helloworld","hello") |
|---|
| 104 | assert_equal( 'world', testString ) |
|---|
| 105 | end |
|---|
| 106 | |
|---|
| 107 | def test_translate |
|---|
| 108 | source = <<-EOF |
|---|
| 109 | <doc> |
|---|
| 110 | <case name='w3c one' result='BAr' /> <!-- w3c --> |
|---|
| 111 | <case name='w3c two' result='AAA' /> <!-- w3c --> |
|---|
| 112 | <case name='alchemy' result="gold" /> <!-- mike --> |
|---|
| 113 | <case name='vbxml one' result='A Space Odyssey' /> |
|---|
| 114 | <case name='vbxml two' result='AbCdEf' /> |
|---|
| 115 | </doc> |
|---|
| 116 | EOF |
|---|
| 117 | |
|---|
| 118 | doc = Document.new(source) |
|---|
| 119 | |
|---|
| 120 | [ ['bar', 'abc', 'ABC', 'w3c one'], |
|---|
| 121 | ['--aaa--','abc-','ABC', 'w3c two'], |
|---|
| 122 | ['lead', 'dear language', 'doll groover', 'alchemy'], |
|---|
| 123 | ['A Space Odissei', 'i', 'y', 'vbxml one'], |
|---|
| 124 | ['abcdefg', 'aceg', 'ACE', 'vbxml two'], |
|---|
| 125 | ].each { |arg1, arg2, arg3, name| |
|---|
| 126 | translate = "translate('#{arg1}', '#{arg2}', '#{arg3}')" |
|---|
| 127 | set = doc.elements.to_a("//case[@result = #{translate}]") |
|---|
| 128 | assert_equal 1, set.size, translate |
|---|
| 129 | assert_equal name, set[0].attributes['name'] |
|---|
| 130 | } |
|---|
| 131 | end |
|---|
| 132 | |
|---|
| 133 | def test_name |
|---|
| 134 | d = REXML::Document.new("<a xmlns:x='foo'><b/><x:b/></a>") |
|---|
| 135 | assert_equal 1, d.root.elements.to_a('*[name() = "b"]').size |
|---|
| 136 | assert_equal 1, d.elements.to_a('//*[name() = "x:b"]').size |
|---|
| 137 | end |
|---|
| 138 | |
|---|
| 139 | def test_local_name |
|---|
| 140 | d = REXML::Document.new("<a xmlns:x='foo'><b/><x:b/></a>") |
|---|
| 141 | assert_equal 2, d.root.elements.to_a('*[local_name() = "b"]').size |
|---|
| 142 | assert_equal 2, d.elements.to_a('//*[local_name() = "b"]').size |
|---|
| 143 | end |
|---|
| 144 | |
|---|
| 145 | def test_substring2 |
|---|
| 146 | doc = Document.new('<test string="12345" />') |
|---|
| 147 | assert_equal(1,doc.elements.to_a("//test[substring(@string,2)='2345']").size) |
|---|
| 148 | end |
|---|
| 149 | |
|---|
| 150 | # Submitted by Kouhei |
|---|
| 151 | def test_floor_ceiling_round |
|---|
| 152 | source = "<a><b id='1'/><b id='2'/><b id='3'/></a>" |
|---|
| 153 | doc = REXML::Document.new(source) |
|---|
| 154 | |
|---|
| 155 | id_1 = doc.elements["/a/b[@id='1']"] |
|---|
| 156 | id_2 = doc.elements["/a/b[@id='2']"] |
|---|
| 157 | id_3 = doc.elements["/a/b[@id='3']"] |
|---|
| 158 | |
|---|
| 159 | good = { |
|---|
| 160 | "floor" => [[], [id_1], [id_2], [id_3]], |
|---|
| 161 | "ceiling" => [[id_1], [id_2], [id_3], []], |
|---|
| 162 | "round" => [[id_1], [id_2], [id_3], []] |
|---|
| 163 | } |
|---|
| 164 | good.each do |key, value| |
|---|
| 165 | (0..3).each do |i| |
|---|
| 166 | xpath = "//b[number(@id) = #{key}(#{i+0.5})]" |
|---|
| 167 | assert_equal(value[i], REXML::XPath.match(doc, xpath)) |
|---|
| 168 | end |
|---|
| 169 | end |
|---|
| 170 | |
|---|
| 171 | good["round"] = [[], [id_1], [id_2], [id_3]] |
|---|
| 172 | good.each do |key, value| |
|---|
| 173 | (0..3).each do |i| |
|---|
| 174 | xpath = "//b[number(@id) = #{key}(#{i+0.4})]" |
|---|
| 175 | assert_equal(value[i], REXML::XPath.match(doc, xpath)) |
|---|
| 176 | end |
|---|
| 177 | end |
|---|
| 178 | end |
|---|
| 179 | |
|---|
| 180 | # Submitted by Kou |
|---|
| 181 | def test_lang |
|---|
| 182 | d = Document.new(<<-XML) |
|---|
| 183 | <a xml:lang="en"> |
|---|
| 184 | <b xml:lang="ja"> |
|---|
| 185 | <c xml:lang="fr"/> |
|---|
| 186 | <d/> |
|---|
| 187 | <e xml:lang="ja-JP"/> |
|---|
| 188 | <f xml:lang="en-US"/> |
|---|
| 189 | </b> |
|---|
| 190 | </a> |
|---|
| 191 | XML |
|---|
| 192 | |
|---|
| 193 | assert_equal(1, d.elements.to_a("//*[lang('fr')]").size) |
|---|
| 194 | assert_equal(3, d.elements.to_a("//*[lang('ja')]").size) |
|---|
| 195 | assert_equal(2, d.elements.to_a("//*[lang('en')]").size) |
|---|
| 196 | assert_equal(1, d.elements.to_a("//*[lang('en-us')]").size) |
|---|
| 197 | |
|---|
| 198 | d = Document.new(<<-XML) |
|---|
| 199 | <root> |
|---|
| 200 | <para xml:lang="en"/> |
|---|
| 201 | <div xml:lang="en"><para/></div> |
|---|
| 202 | <para xml:lang="EN"/> |
|---|
| 203 | <para xml:lang="en-us"/> |
|---|
| 204 | </root> |
|---|
| 205 | XML |
|---|
| 206 | |
|---|
| 207 | assert_equal(5, d.elements.to_a("//*[lang('en')]").size) |
|---|
| 208 | end |
|---|
| 209 | |
|---|
| 210 | def test_ticket_60 |
|---|
| 211 | document = REXML::Document.new("<a><b>A</b><b>1</b></a>") |
|---|
| 212 | assert_equal( "A", REXML::XPath.first(document, '//b[.="A"]').text ) |
|---|
| 213 | assert_equal( "1", REXML::XPath.first(document, '//b[.="1"]').text ) |
|---|
| 214 | end |
|---|
| 215 | |
|---|
| 216 | def test_normalize_space |
|---|
| 217 | source = "<a><!--COMMENT A--><b><!-- COMMENT A --></b></a>" |
|---|
| 218 | doc = REXML::Document.new(source) |
|---|
| 219 | predicate = "string(.)=normalize_space('\nCOMMENT \n A \n\n ')" |
|---|
| 220 | m = REXML::XPath.match(doc, "//comment()[#{predicate}]") |
|---|
| 221 | assert_equal( [REXML::Comment.new("COMMENT A")], m ) |
|---|
| 222 | end |
|---|
| 223 | end |
|---|