test/test_xmlcanonicalizer.rb in canonix-0.1.3 vs test/test_xmlcanonicalizer.rb in canonix-0.1.4

@@ -1,80 +1,106 @@
 require File.dirname(File.expand_path(__FILE__))+'/helper'
 
-class TestXmlcanonicalizer < Test::Unit::TestCase
+class TestXmlCanonicalizer < Test::Unit::TestCase
   
   should "canonicalize a simple xml file" do
     xml_canonicalizer = XML::Util::XmlCanonicalizer.new(true,true)
-    xml = "<foo bar='test'/>";
-    rexml = REXML::Document.new(xml);
-    xml_canonicalized = xml_canonicalizer.canonicalize(rexml);
-    xml_expect = "<foo bar=\"test\"></foo>";
+    
+    xml = "<foo bar='test'/>"
+    rexml = REXML::Document.new(xml)
+    xml_canonicalized = xml_canonicalizer.canonicalize(rexml)
+    
+    xml_expect = "<foo bar=\"test\"></foo>"
     assert_equal xml_expect, xml_canonicalized
   end
   
   should "canonicalize a complex xml file" do
-    fp = File.new(File.dirname(File.expand_path(__FILE__))+'/complex.xml','r')
-    xml = ''
-    while (l = fp.gets)
-      xml += l
-    end
-    fp.close
-    
     xml_canonicalizer = XML::Util::XmlCanonicalizer.new(true,true)
-    rexml = REXML::Document.new(xml);
-    xml_canonicalized = xml_canonicalizer.canonicalize(rexml);
     
-    fp = File.new(File.dirname(File.expand_path(__FILE__))+'/expected.xml','r')
-    xml_expect = ''
-    while (l = fp.gets)
-      xml_expect += l
-    end
-    fp.close
+    rexml = rexml_fixture("complex.xml")
+    xml_canonicalized = xml_canonicalizer.canonicalize(rexml)
     
+    xml_expect = fixture("expected.xml")
     assert_equal xml_expect, xml_canonicalized
   end
-
-  should "canonicalize a saml xml file correctly" do
-    fp = File.new(File.dirname(File.expand_path(__FILE__))+'/saml_assertion.xml','r')
-    xml = ''
-    while (l = fp.gets)
-      xml += l
-    end
-    fp.close
+  
+  should "canonicalize multiple documents" do
+    xml_canonicalizer = XML::Util::XmlCanonicalizer.new(true,true)
     
+    rexml_1 = rexml_fixture("complex.xml")
+    xml_canonicalizer.canonicalize(rexml_1)
+    
+    rexml_2 = rexml_fixture("complex.xml")
+    xml_canonicalized_2 = xml_canonicalizer.canonicalize(rexml_2)
+    
+    xml_expect = fixture("expected.xml")
+    assert_equal xml_expect, xml_canonicalized_2
+  end
+  
+  should "canonicalize the same document multiple times" do
+    xml_canonicalizer = XML::Util::XmlCanonicalizer.new(true,true)
+    
+    rexml = rexml_fixture("complex.xml")
+    xml_canonicalized_1 = xml_canonicalizer.canonicalize(rexml)
+    xml_canonicalized_2 = xml_canonicalizer.canonicalize(rexml)
+    
+    assert_equal xml_canonicalized_1, xml_canonicalized_2
+  end
+  
+  should "canonicalize an xml element correctly" do
+    xml_canonicalizer = XML::Util::XmlCanonicalizer.new(true,true)
+    
+    rexml = rexml_fixture("complex.xml")
+    element = REXML::XPath.first(rexml, "//AttributeValue[@FriendlyName='type']")
+    element_canonicalized = xml_canonicalizer.canonicalize(element)
+    
+    element_expected = '<AttributeValue FriendlyName="type" type="example:profile:attribute">Person</AttributeValue>'
+    assert_equal element_expected, element_canonicalized
+  end
+  
+  should "canonicalize the same element multiple times" do
+    xml_canonicalizer = XML::Util::XmlCanonicalizer.new(true,true)
+    
+    rexml = rexml_fixture("complex.xml")
+    element = REXML::XPath.first(rexml, "//AttributeValue[@FriendlyName='type']")
+    element_canonicalized_1 = xml_canonicalizer.canonicalize(element)
+    element_canonicalized_2 = xml_canonicalizer.canonicalize(element)
+    
+    assert_equal element_canonicalized_1, element_canonicalized_2
+  end
+  
+  should "canonicalize multiple xml elements correctly" do
+    xml_canonicalizer = XML::Util::XmlCanonicalizer.new(true,true)
+    
+    rexml = rexml_fixture("complex.xml")
+    
+    element_1 = REXML::XPath.first(rexml, "//AttributeValue[@FriendlyName='type']")
+    xml_canonicalizer.canonicalize(element_1)
+    
+    element_2 = REXML::XPath.first(rexml, "//AuthnStatement")
+    element_2_canonicalized = xml_canonicalizer.canonicalize(element_2)
+    element_2_expected = '<AuthnStatement AuthnInstant="2010-09-10T00:00:50-05:00"><AuthnContext><AuthnContextClassRef>urn:oasis:names:tc:SAML:2.0:ac:classes:PreviousSession</AuthnContextClassRef></AuthnContext></AuthnStatement>'
+    assert_equal element_2_expected, element_2_canonicalized
+  end
+  
+  should "canonicalize a saml xml file correctly" do
     xml_canonicalizer = XML::Util::XmlCanonicalizer.new(false,true)
-    rexml = REXML::Document.new(xml);
-    xml_canonicalized = xml_canonicalizer.canonicalize(rexml);
     
-    fp = File.new(File.dirname(File.expand_path(__FILE__))+'/saml_expected_canonical_form.xml','r')
-    xml_expect = ''
-    while (l = fp.gets)
-      xml_expect += l
-    end
-    fp.close
+    rexml = rexml_fixture("saml_assertion.xml")
+    xml_canonicalized = xml_canonicalizer.canonicalize(rexml)
     
+    xml_expect = fixture("saml_expected_canonical_form.xml")
     assert_equal xml_expect, xml_canonicalized
   end
   
   should "canonicalize a saml file with inclusive namespaces" do
-    fp = File.new(File.dirname(File.expand_path(__FILE__))+'/saml_with_inclusive_ns_assertion.xml','r')
-    xml = ''
-    while (l = fp.gets)
-      xml += l
-    end
-    fp.close
-    
     xml_canonicalizer = XML::Util::XmlCanonicalizer.new(false,true)
-    rexml = REXML::Document.new(xml);
+    
+    rexml = rexml_fixture("saml_with_inclusive_ns_assertion.xml")
     xml_canonicalizer.inclusive_namespaces = %w(ds saml samlp xs)
-    xml_canonicalized = xml_canonicalizer.canonicalize(rexml);
+    xml_canonicalized = xml_canonicalizer.canonicalize(rexml)
     
-    fp = File.new(File.dirname(File.expand_path(__FILE__))+'/saml_with_inclusive_ns_expected_canonical_form.xml','r')
-    xml_expect = ''
-    while (l = fp.gets)
-      xml_expect += l
-    end
-    fp.close
+    xml_expect = fixture("saml_with_inclusive_ns_expected_canonical_form.xml")
     assert_equal xml_expect, xml_canonicalized #, (xml_canonicalized.to_s + "\n\n" + xml_expect)
   end
 
 end