Module: Beaker::LoggerJunit

Defined in:
lib/beaker/logger_junit.rb

Overview

The Beaker JUnit Logger class This module handles message reporting from Beaker to the JUnit format

There is a specific pattern for using this class. Here’s a list of example usages:

Class Method Summary collapse

Class Method Details

.copy_stylesheet_into_xml_dir(stylesheet, xml_file) ⇒ Object

copies given stylesheet into the directory of the xml file given



69
70
71
72
73
 
# File 'lib/beaker/logger_junit.rb', line 69

def self.copy_stylesheet_into_xml_dir(stylesheet, xml_file)
  return if File.file?(File.join(File.dirname(xml_file), File.basename(stylesheet)))

  FileUtils.copy(stylesheet, File.join(File.dirname(xml_file), File.basename(stylesheet)))
end

.escape_invalid_xml_chars(string) ⇒ String

Escape invalid XML UTF-8 codes from provided string, see www.w3.org/TR/xml/#charsets for valid character specification



127
128
129
130
131
132
133
134
135
136
137
138
 
# File 'lib/beaker/logger_junit.rb', line 127

def self.escape_invalid_xml_chars string
  escaped_string = ""
  string.chars.each do |i|
    char_as_codestring = i.unpack("U*").join
    escaped_string << if self.is_valid_xml(char_as_codestring.to_i)
                        i
                      else
                        "\\#{char_as_codestring}"
                      end
  end
  escaped_string
end

.finish(doc, xml_file) ⇒ Object

writes out xml content for a doc



41
42
43
44
45
 
# File 'lib/beaker/logger_junit.rb', line 41

def self.finish(doc, xml_file)
  # junit/name.xml will be created in a directory relative to the CWD

  File.open(xml_file, 'w') { |f| doc.write(f, 2) }
end

.format_cdata(string) ⇒ String

Remove color codes and invalid XML characters from provided string



119
120
121
 
# File 'lib/beaker/logger_junit.rb', line 119

def self.format_cdata string
  self.escape_invalid_xml_chars(Logger.strip_color_codes(string))
end

.get_doc_for_filename(filename, stylesheet, already_exists) ⇒ REXML::Document

gives the document object for a particular file



103
104
105
106
107
108
109
110
111
112
113
114
 
# File 'lib/beaker/logger_junit.rb', line 103

def self.get_doc_for_filename(filename, stylesheet, already_exists)
  if already_exists
    doc = REXML::Document.new File.open(filename)
  else
    # no existing file, create a new one
    doc = REXML::Document.new
    doc << REXML::XMLDecl.new("1.0", "UTF-8")
    instruction_content = "type='text/xsl' href='#{File.basename(stylesheet)}'"
    doc << REXML::Instruction.new("xml-stylesheet", instruction_content)
  end
  return doc
end

.get_testsuites_from_doc(doc, name, already_existed) ⇒ Rexml::Element

sets up doc & gives us the suites for the testsuite named



82
83
84
85
86
87
88
89
90
91
92
93
94
 
# File 'lib/beaker/logger_junit.rb', line 82

def self.get_testsuites_from_doc(doc, name, already_existed)
  # check to see if an output file already exists, if it does add or replace test suite data
  if already_existed
    suites = REXML::XPath.first(doc, "testsuites")
    # remove old data
    suites.elements.each("testsuite") do |e|
      suites.delete_element e if /#{name}/.match?(e.name)
    end
  else
    suites = doc.add_element(REXML::Element.new('testsuites'))
  end
  return suites
end

.get_xml_contents(xml_file, name, stylesheet) ⇒ REXML::Document, REXML::Element

gets the xml doc & suites in order to build your xml output on top of



55
56
57
58
59
60
61
 
# File 'lib/beaker/logger_junit.rb', line 55

def self.get_xml_contents(xml_file, name, stylesheet)
  self.copy_stylesheet_into_xml_dir(stylesheet, xml_file)
  xml_file_already_exists = File.file?(xml_file)
  doc = self.get_doc_for_filename(xml_file, stylesheet, xml_file_already_exists)
  suites = self.get_testsuites_from_doc(doc, name, xml_file_already_exists)
  return doc, suites
end

.is_valid_xml(int) ⇒ Boolean

Determine if the provided number falls in the range of accepted xml unicode values See www.w3.org/TR/xml/#charsets for valid for valid character specifications.



144
145
146
147
148
149
150
151
 
# File 'lib/beaker/logger_junit.rb', line 144

def self.is_valid_xml(int)
  return (int == 0x9 or
    int == 0xA or
    (int >= 0x0020 and int <= 0xD7FF) or
    (int >= 0xE000 and int <= 0xFFFD) or
    (int >= 0x100000 and int <= 0x10FFFF)
         )
end

.write_xml(xml_file, stylesheet, &block) ⇒ Object

writes the xml created in the block to the xml file given

Note: Error Recovery should take place in the caller of this method in order to recover gracefully



20
21
22
23
24
25
26
27
28
29
30
31
32
33
 
# File 'lib/beaker/logger_junit.rb', line 20

def self.write_xml(xml_file, stylesheet, &block)
  doc, suites = self.get_xml_contents(xml_file, name, stylesheet)

  if block
    case block.arity
    when 2
      yield doc, suites
    else
      raise ArgumentError.new "write_xml block takes 2 arguments, not #{block.arity}"
    end
  end

  self.finish(doc, xml_file)
end