Module: BenchmarkRunner::DataGenerator

Defined in:
lib/tasks/benchmark_runner.rb

Overview

Test data generators

Class Method Summary collapse

Class Method Details

.build_xml_element(items, depth, prefix, with_attrs, ns_decl) ⇒ Object



281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
# File 'lib/tasks/benchmark_runner.rb', line 281

def build_xml_element(items, depth, prefix, with_attrs, ns_decl)
  attrs = with_attrs ? " id=\"#{rand(1000)}\" status=\"active\"" : ""
  ns_attr = ns_decl.empty? ? "" : " #{ns_decl}"

  if depth <= 1
    children = Array.new(items) do |i|
      inner_attrs = with_attrs ? " index=\"#{i}\"" : ""
      "<#{prefix}item#{inner_attrs}>Item #{i} content with some text</#{prefix}item>"
    end.join
    "<#{prefix}root#{ns_attr}#{attrs}>#{children}</#{prefix}root>"
  else
    child = build_xml_element(items / 2, depth - 1, prefix, with_attrs,
                              "")
    "<#{prefix}root#{ns_attr}#{attrs}>#{child}</#{prefix}root>"
  end
end

.generate_html(items: DEFAULT_ITEMS, with_scripts: false, with_tables: false) ⇒ Object



298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
# File 'lib/tasks/benchmark_runner.rb', line 298

def generate_html(items: DEFAULT_ITEMS, with_scripts: false,
                  with_tables: false)
  scripts = if with_scripts
              <<~HTML
                <script type="text/javascript">
                  // Some inline script
                  function test() { return true; }
                </script>
                <style>
                  body { margin: 0; }
                </style>
              HTML
            else
              ""
            end

  tables = if with_tables
             rows = Array.new(items) do |i|
               "<tr><td>Cell #{i}A</td><td>Cell #{i}B</td></tr>"
             end.join
             "<table>#{rows}</table>"
           else
             ""
           end

  list_items = Array.new(items) do |i|
    "<li class=\"item-#{i}\">List item #{i} with bold text</li>"
  end.join("\n              ")

  nav_item_count = [(items / 5), 2].max
  nav_items = items.times.first(nav_item_count).map do |i|
    "<li class=\"nav-#{i}\">Nav #{i}</li>"
  end.join("\n                ")

  <<~HTML
    <!DOCTYPE html>
    <html lang="en">
    <head>
      <meta charset="UTF-8">
      <title>Benchmark Document</title>
      #{scripts}
    </head>
    <body>
      <header>
        <h1>Benchmark Test Document</h1>
        <nav>
          <ul>
          #{nav_items}
          </ul>
        </nav>
      </header>
      <main>
        <section id="content">
          <p>This is a paragraph with emphasized and strong text.</p>
          <ul>
          #{list_items}
          </ul>
          #{tables}
        </section>
      </main>
      <footer>
        <p>Copyright 2024</p>
      </footer>
    </body>
    </html>
  HTML
end

.generate_json(items: DEFAULT_ITEMS) ⇒ Object



366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
# File 'lib/tasks/benchmark_runner.rb', line 366

def generate_json(items: DEFAULT_ITEMS)
  data = {
    metadata: { version: "1.0", generated: Time.now.iso8601 },
    items: Array.new(items) do |i|
      {
        id: i,
        name: "Item #{i}",
        value: rand * 1000,
        tags: ["tag1", "tag2", "tag#{i % 10}"],
        nested: {
          level1: { level2: { data: "deeply nested value #{i}" } },
        },
      }
    end,
  }
  JSON.generate(data)
end

.generate_large_xml(items: 500) ⇒ Object



389
390
391
392
# File 'lib/tasks/benchmark_runner.rb', line 389

def generate_large_xml(items: 500)
  generate_xml(items: items, depth: 3, with_namespaces: true,
               with_attributes: true)
end

.generate_xml(items: DEFAULT_ITEMS, depth: 1, with_namespaces: false, with_attributes: true) ⇒ Object



273
274
275
276
277
278
279
# File 'lib/tasks/benchmark_runner.rb', line 273

def generate_xml(items: DEFAULT_ITEMS, depth: 1, with_namespaces: false,
                 with_attributes: true)
  ns = with_namespaces ? 'xmlns:ns="http://example.org"' : ""
  prefix = with_namespaces ? "ns:" : ""

  build_xml_element(items, depth, prefix, with_attributes, ns)
end

.generate_yaml(items: DEFAULT_ITEMS) ⇒ Object



384
385
386
387
# File 'lib/tasks/benchmark_runner.rb', line 384

def generate_yaml(items: DEFAULT_ITEMS)
  data = JSON.parse(generate_json(items: items))
  data.to_yaml
end