mbt-gen 0.0.8 → 0.0.10

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 763d26c653134a8a5432753091e286e1df8aecf8c603a5eba5bc6980ed5c3e6f
-  data.tar.gz: e5e86c5368af05ff682fbd2f7bf7fed6a9e7909d7d92523728956ee3fee71946
+  metadata.gz: 77cbf93998b731d074f62e0b364c20f87f324c8409e0e223193e9d8c0a0ddc52
+  data.tar.gz: 98985c6b4ace9d25b36fc73d9110f50368bc8badff62918b52a09557c0c52bd3
 SHA512:
-  metadata.gz: 5c32f100cae06aa9f7e97d7e1b214f27a9d0747948feafddd01530392f45df0a9966e4f6ad3ef5be9a719b87f9e0bd91f966523415231b539026503aa1659064
-  data.tar.gz: 485bd1dfe0f01f7e553087603c90e73c582b7dd29de72cd79032f63f1b35df32af15c66369c540507ae4ed8427244159710076221dcf033696e7ce8732191ff7
+  metadata.gz: aad6c3b1ea3c63b7510cfc214a020fc8cf91e8f2cb97f2da44c39543d350524a62999440c2a97ea9ecfbdc3c036365667d8853150f6eeda485bf07d464b09fa1
+  data.tar.gz: e8d4adb8a2ea41d6a5200ca4002245afe66ab5779539567d18c206323d1c7c45226ac9273569609e2ed12ccfb3bc434c31ab8c4717bc81d639a4f1c5020cdb0f

data/lib/mbt-gen.rb

@@ -18,6 +18,7 @@ DEFAULT_Z3_PATH = "z3"
 
 CONFIG_FILE = "config.yml"
 OUTPUT_FILE = "result.xml"
+BASE_FILENAME = "doc-base.xml"
 CONTINUATION_FILE = "continuation.log"
 COMBINATION_LOG = "combinations.log"
 VARS_LOG = "vars.log"
@@ -51,6 +52,19 @@ GENERATING A SINGLE MESSAGE:
     -negate <validation_rule> - negates the named validation rule. This causes the resulting message to still satisfy all other validation rules, but to violate this one.
   NOTE: the result can be found in #{OUTPUT_FILE}
   
+GENERATING MESSAGES FOR SPECIAL REQUIREMENTS:
+
+  ruby #{$PROGRAM_NAME} -generate-max
+  - generates a message of maximum size. The message will be valid XML and will adhere to the schema, but it will
+    not neccessarily satisfy all validation rules. It will contain all structures, all optional fields and all lists of maximum size.
+
+  ruby #{$PROGRAM_NAME} -generate-equality-test-data
+  - generates a base message of maximum size (see -generate-max option) as well as the following test messages for each field and structure
+    - one test message in which the field contains a different value (only for fields)
+    - one test message in which the field/structure is empty in case it is optional
+    NOTE: all the other fields and structures will be the same as in the base message. This kind of test data is useful for
+          testing implementations of equality and comparison functionality on the specified messages.
+
 GENERATING MESSAGES FOR ALL COMBINATIONS:
 
   ruby #{$PROGRAM_NAME} -list-keys
@@ -772,6 +786,7 @@ def parse_field_value(fielddef, value)
   when :enum
     return solver_value_to_string(value)
   when :date
+    puts "DEBUG: parsing date #{value.inspect}"
     return Date.parse(value)
   when :timestamp
     return Time.new(value)
@@ -799,6 +814,91 @@ def format_solver_value(fielddef, solver_value)
   end
 end
 
+def build_max_doc(progress, struct)
+  doc = Nokogiri::XML(XML_TEMPLATE)
+  add_max_struct_to_doc(progress, struct, doc)
+  return doc
+end
+
+def pick_solver_value_for_field(field_desc)
+  if field_desc[:datatype] == :int then
+    if field_desc[:min] then
+      field_desc[:min].to_s
+    elsif field_desc[:max] then
+      field_desc[:max].to_s
+    else
+      "0"
+    end
+  elsif field_desc[:datatype] == :string then
+    # TODO: when a string field has a :regexp, use the solver to find a value satisfying it
+    if field_desc[:maxLength] then
+      "a" * field_desc[:maxLength]
+    else
+      "default"
+    end
+  elsif field_desc[:datatype] == :enum then
+    field_desc[:values].first.to_s
+  elsif field_desc[:datatype] == :date then
+    Time.at(0).to_date.to_s
+  elsif field_desc[:datatype] == :timestamp then
+    Time.at(0).to_s
+  else
+    raise RuntimeError.new("encountered unknown :datatype #{field_desc[:datatype].inspect}")
+  end
+end
+
+def add_max_struct_to_doc(progress, struct, doc, prefix = nil)
+  struct.each do |key, value|
+    next if key == :validation_rules || key == :predicates || key == :parent_link || key == :struct_xpath_prefix || key == :additional_smtlib
+    if prefix then
+      xpath = "#{prefix}/#{key}"
+    else
+      xpath = key.to_s
+    end
+    if value[:type] == :field then
+      val = pick_solver_value_for_field(value)
+      fvalue_str = format_solver_value(value, val)
+      doc.add_child("<#{key}>#{fvalue_str}</#{key}>")
+    elsif value[:type] == :list then
+      node = doc.add_child("<#{key}></#{key}>")
+      if node.is_a?(Nokogiri::XML::NodeSet) then
+        node = node.first 
+      end
+      xpath_element = value[:xpath_element]
+      if Integer(value[:model_maxLength]) > 0 then
+        Integer(value[:model_maxLength]).times do |i|
+          node.add_child("<#{xpath_element}>#{format_solver_value(value, pick_solver_value_for_field(value))}</#{xpath_element}>")
+        end
+      end
+    elsif value[:type] == :objlist then
+      node = doc.add_child("<#{key}></#{key}>")
+      if node.is_a?(Nokogiri::XML::NodeSet) then
+        node = node.first 
+      end
+      xpath_element = value[:xpath_element]
+      value[:model_maxLength].times do |index|
+        cnode = node.add_child("<#{xpath_element}></#{xpath_element}>")
+        if cnode.is_a?(Nokogiri::XML::NodeSet) then
+          cnode = cnode.first 
+        end
+        add_max_struct_to_doc(progress, value[:ref], cnode, "#{xpath}/#{index}")
+      end
+    elsif value[:type] == :structure then
+      unless value[:ref] then
+        raise RuntimeError.new("struct is broken: :structure #{key.inspect} does not have a :ref")
+      end
+      node = doc.add_child("<#{key}></#{key}>")
+      if node.is_a?(Nokogiri::XML::NodeSet) then
+        node = node.first 
+      end
+      add_max_struct_to_doc(progress, value[:ref], node, xpath)
+    else
+      raise RuntimeError.new("struct is broken: encountered unknown :type #{value[:type].inspect}")
+    end    
+  end
+end
+
+
 def build_document_from_model(progress, struct, model)
   doc = Nokogiri::XML(XML_TEMPLATE)
   add_model_struct_to_doc(progress, struct, doc, model)
@@ -888,6 +988,148 @@ def set_standard_options(progress, solver)
   solver.to_solver(progress, "")
 end
 
+def each_structure_and_field_in(struct, prefix = nil, &block)
+  struct.each do |key, value|
+    next if key == :validation_rules || key == :predicates || key == :parent_link || key == :struct_xpath_prefix || key == :additional_smtlib
+
+    if prefix then
+      xpath = "#{prefix}/#{key}"
+    else
+      xpath = key
+    end
+    
+    if value[:type] == :field then
+      block.call(value, xpath)
+    elsif value[:type] == :list then
+      block.call(value, xpath)
+      value[:model_maxLength].times do |i|
+        cvalue = value.clone()
+        cvalue[:element] = true
+        block.call(cvalue, xpath + "/#{value[:xpath_element]}[#{i+1}]")
+      end
+    elsif value[:type] == :objlist then
+      block.call(value, xpath)
+      value[:model_maxLength].times do |i|
+        cxpath = xpath + "/#{value[:xpath_element]}[#{i+1}]"
+        cvalue = value.clone()
+        cvalue[:element] = true
+        block.call(cvalue, cxpath)
+        each_structure_and_field_in(value[:ref], cxpath, &block)      
+      end
+    elsif value[:type] == :structure then
+      block.call(value, xpath)
+      each_structure_and_field_in(value[:ref], xpath, &block)      
+    else
+      raise RuntimeError.new("encountered unknown :type #{value[:type].inspect}")
+    end
+  end
+end
+
+def xpath_to_filename(xpath)
+  xpath.gsub("/", "-").gsub(/\[\d*\]/) {|s| if s =~ /\[(\d*)\]/ then "-#{$1}-" else s end }
+end
+
+def generate_equality_test_data(progress, message, output_dir, config, options)
+  unless Dir.exist?(output_dir)
+    FileUtils.mkdir_p(output_dir)
+  end
+
+  base_filename = "#{output_dir}/#{BASE_FILENAME}"
+  generate_max_doc(progress, message, base_filename, config, options)
+
+  base_doc = Nokogiri::XML(File.read(base_filename))
+
+  each_structure_and_field_in(message) do |desc, xpath|
+    if desc[:type] == :field then
+      mutated_doc = base_doc.clone()
+      node = mutated_doc.at_xpath(xpath)
+      unless node then
+        raise RuntimeError.new("xpath #{xpath.inspect} could not be found in base_doc")
+      end
+      node.inner_html = node.inner_html + "x"
+      File.open("#{output_dir}/doc-mutated-#{xpath_to_filename(xpath)}.xml", "w") do |f|
+        f.puts(mutated_doc.to_xml)
+      end
+      if desc[:optional] == true then
+        mutated_doc = base_doc.clone()
+        node = mutated_doc.at_xpath(xpath)
+        unless node then
+          raise RuntimeError.new("xpath #{xpath.inspect} could not be found in base_doc")
+        end
+        node.remove()
+        File.open("#{output_dir}/doc-removed-#{xpath_to_filename(xpath)}.xml", "w") do |f|
+          f.puts(mutated_doc.to_xml)
+        end
+      end
+    elsif desc[:type] == :list then
+      if desc[:element] then
+        # list elements
+        mutated_doc = base_doc.clone()
+        node = mutated_doc.at_xpath(xpath)
+        unless node then
+          raise RuntimeError.new("xpath #{xpath.inspect} could not be found in base_doc")
+        end
+        node.inner_html = node.inner_html + "x"
+        File.open("#{output_dir}/doc-mutated-#{xpath_to_filename(xpath)}.xml", "w") do |f|
+          f.puts(mutated_doc.to_xml)
+        end
+      else
+        # the list itself
+        mutated_doc = base_doc.clone()
+        node = mutated_doc.at_xpath(xpath)
+        unless node then
+          raise RuntimeError.new("xpath #{xpath.inspect} could not be found in base_doc")
+        end
+        node.remove()
+        File.open("#{output_dir}/doc-removed-#{xpath_to_filename(xpath)}.xml", "w") do |f|
+          f.puts(mutated_doc.to_xml)
+        end
+      end
+    elsif desc[:type] == :objlist then
+      if desc[:element] then
+        # list elements
+        # no additional test data neccessary as the element objects will contain fields that are modified already
+      else
+        # the list itself
+        mutated_doc = base_doc.clone()
+        node = mutated_doc.at_xpath(xpath)
+        unless node then
+          raise RuntimeError.new("xpath #{xpath.inspect} could not be found in base_doc")
+        end
+        node.remove()
+        File.open("#{output_dir}/doc-removed-#{xpath_to_filename(xpath)}.xml", "w") do |f|
+          f.puts(mutated_doc.to_xml)
+        end
+      end
+    elsif desc[:type] == :structure then
+      if desc[:optional] == true then
+        mutated_doc = base_doc.clone()
+        node = mutated_doc.at_xpath(xpath)
+        unless node then
+          raise RuntimeError.new("xpath #{xpath.inspect} could not be found in base_doc")
+        end
+        node.remove()
+        File.open("#{output_dir}/doc-removed-#{xpath_to_filename(xpath)}.xml", "w") do |f|
+          f.puts(mutated_doc.to_xml)
+        end
+      end
+    else
+      raise RuntimeError.new("encountered unknown :type #{desc[:type].inspect}")
+    end
+  end    
+end
+
+def generate_max_doc(progress, message, output_filename, config, options)
+  result_doc = build_max_doc(progress, message)
+  dir = File.dirname(output_filename)
+  FileUtils.mkdir_p(dir) unless Dir.exist?(dir)
+  File.open(output_filename, "w") do |f|
+    f.puts result_doc.to_xml()
+  end
+  progress.print_line "DONE. Generated XML was written to #{output_filename}."
+  return true
+end
+
 def generate_doc(progress, message, output_filename, config, options)
   raise RuntimeError.new("option :solverFactory is required") unless options[:solverFactory]
   log = options[:log] || nil
@@ -1978,7 +2220,7 @@ def translate_struct_in_doc_to_SMTLIB(progress, solver, struct, doc, prefix, var
           :assertion_type => :doc_field_empty,
           :xpath => xpath,
         }
-        assertionID = "doc-#{xpath.gsub('/', '-')}"
+        assertionID = "doc-#{var_xpath.gsub('/', '-')}"
         solver.associateAssertionIDWith(assertionID, info)
         solver.to_solver(progress, "(assert (! (= #{filled_var_for_field(var_xpath)} false) :named #{assertionID}))")
       end
@@ -1995,7 +2237,7 @@ def translate_struct_in_doc_to_SMTLIB(progress, solver, struct, doc, prefix, var
           :xpath => xpath,
           :values => fvalues
         }
-        assertionID = "doc-#{xpath.gsub('/', '-')}"
+        assertionID = "doc-#{var_xpath.gsub('/', '-')}"
         solver.associateAssertionIDWith(assertionID, info)
         clauses = []
         fvalues.each_with_index do |fvalue, index|
@@ -2007,7 +2249,7 @@ def translate_struct_in_doc_to_SMTLIB(progress, solver, struct, doc, prefix, var
           :assertion_type => :doc_field_empty,
           :xpath => xpath,
         }
-        assertionID = "doc-#{xpath.gsub('/', '-')}"
+        assertionID = "doc-#{var_xpath.gsub('/', '-')}"
         solver.associateAssertionIDWith(assertionID, info)
         solver.to_solver(progress, "(assert (! (= #{filled_var_for_field(var_xpath)} false) :named #{assertionID}))")
       end
@@ -2047,7 +2289,7 @@ def translate_struct_in_doc_to_SMTLIB(progress, solver, struct, doc, prefix, var
           :size => num_elements,
           :xpath => xpath
         }
-        assertionID = "doc-#{xpath.gsub('/', '-')}"
+        assertionID = "doc-#{var_xpath.gsub('/', '-')}"
         solver.associateAssertionIDWith(assertionID, info)
         clauses = []
         solver.to_solver(progress, "(assert (! (and (= #{filled_var_for_field(var_xpath)} true) (= #{size_var_for_list(var_xpath)} #{num_elements})) :named #{assertionID}))")
@@ -2056,7 +2298,7 @@ def translate_struct_in_doc_to_SMTLIB(progress, solver, struct, doc, prefix, var
           :assertion_type => :doc_field_empty,
           :xpath => xpath,
         }
-        assertionID = "doc-#{xpath.gsub('/', '-')}"
+        assertionID = "doc-#{var_xpath.gsub('/', '-')}"
         solver.associateAssertionIDWith(assertionID, info)
         solver.to_solver(progress, "(assert (! (= #{filled_var_for_field(var_xpath)} false) :named #{assertionID}))")
       end
@@ -2132,6 +2374,10 @@ def generate(message, argv, message_config_filename = nil)
       else
         options[:validate_files] = [argv.shift]
       end
+    elsif opt == '-generate-max' then
+      options[:generate_max] = true
+    elsif opt == '-generate-equality-test-data' then
+      options[:generate_equality_test_data] = true
     else
       puts "FATAL: unknown option #{opt.inspect}."
       puts USAGE
@@ -2192,6 +2438,12 @@ def generate(message, argv, message_config_filename = nil)
       list_fields_and_structures(message, options[:list])
     elsif options[:list_validation_rules] then
       list_validation_rules(message)
+    elsif options[:generate_max] then
+      validate_model(message)
+      generate_max_doc(progress, message, OUTPUT_FILE, config, options)
+    elsif options[:generate_equality_test_data] then
+      validate_model(message)
+      generate_equality_test_data(progress, message, OUTPUT_DIR, config, options)
     else
       progress.print_line "using #{options[:date_now]} as [Date.now]"
       progress.print_line "using #{options[:timestamp_now]} as [Timestamp.now]"

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: mbt-gen
 version: !ruby/object:Gem::Version
-  version: 0.0.8
+  version: 0.0.10
 platform: ruby
 authors:
 - FM-enthusiast
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2026-08-08 00:00:00.000000000 Z
+date: 2026-09-01 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: nokogiri