quokkadb 1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 5581c6ea0323e66a9427c900c403859df29cf962
+  data.tar.gz: 8c9c7a4d7eee93b1d3875d8a66bffd16499f2db9
+SHA512:
+  metadata.gz: de037adf0b869b13fb19085a467675d0dff6e00a7a498c3237f176a681ff9b627734dd7837ed4412aac89835c74a246a851c6cecf4a1472c4e6de2d5ff734c42
+  data.tar.gz: b0a3d681a7644461ccbc6f3ae42031a9b9f5be0607e5f96a00d19df4864a34d59b0a1e28f4d078187ebfb80f828276850905b54adc00f088aff291a56a74481c

data/lib/quokkadb/adb.rb

@@ -0,0 +1,37 @@
+module ADB
+
+  def ADB.find_tests(runner:, package:, serial: nil, annotation: nil, shards: nil, index: nil)
+    puts("adb #{serial ? "-s #{serial}" : ''} shell am instrument #{annotation ? "-e annotation #{annotation}" : ''} -e numShards 4 -e shardIndex 0 -r -w #{package}/#{runner}\n")
+    `adb #{serial ? "-s #{serial}" : ''} shell am instrument #{annotation ? "-e annotation #{annotation}" : ''} #{shards ? "-e numShards #{shards}" : ''} #{index ? "-e shardIndex #{index}" : ''} -e log true -r -w #{package}/#{runner}`
+  end
+
+  def ADB.execute_test(runner:, package:, test_name:, class_name:)
+    puts("adb shell am instrument -w -e class #{class_name}##{test_name} #{package}/#{runner}")
+    `adb shell am instrument -w -e class #{class_name}##{test_name} #{package}/#{runner}`
+  end
+
+  def ADB.mass_execute(runner:, package:, tests:)
+    tests_list = []
+    tests.each do |test_info|
+      tests_list.push("#{test_info.class_name}##{test_info.name}")
+    end
+
+    puts("adb shell am instrument -w -e class #{tests_list.join(',')} #{package}/#{runner}")
+    `adb shell am instrument -w -e class #{tests_list.join(',')} -r #{package}/#{runner}`
+  end
+
+  def ADB.run_tests(runner:, package:, serial: nil, annotation: nil, shards: nil, index: nil, &callback)
+    command = "adb #{serial ? "-s #{serial}" : ''} shell am instrument #{annotation ? "-e annotation #{annotation}" : ''} #{ADB.generate_sharding_options(shards: shards, index: index)} -r -w #{package}/#{runner}"
+    puts("Command: #{command}")
+    pipe = IO.popen(command)
+    while (line = pipe.gets)
+      callback.call(line)
+    end
+  end
+
+  def ADB.generate_sharding_options(shards:, index:)
+    return "" unless shards
+
+    "#{shards != '' ? "-e numShards #{shards}" : ''} #{index && index != '' ? "-e shardIndex #{index}" : ''}"
+  end
+end

data/lib/quokkadb/adb_test_parser.rb

@@ -0,0 +1,102 @@
+require_relative './test.rb'
+require_relative './test_result.rb'
+require 'colorize'
+
+class ADBTestParser
+
+  PREFIX_TEST = 'INSTRUMENTATION_STATUS: test='
+  PREFIX_CLASS = 'INSTRUMENTATION_STATUS: class='
+  PREFIX_CODE = 'INSTRUMENTATION_STATUS_CODE: '
+  PREFIX_STREAM = 'INSTRUMENTATION_STATUS: stream='
+  PREFIX_STACK = 'INSTRUMENTATION_STATUS: stack='
+
+  attr_accessor :test_results
+
+  def initialize
+    clear
+    @test_results = []
+    @lock = Mutex.new
+  end
+
+  def parse_line(line:, background_black: false)
+    @lock.synchronize do
+      # If starts with prefix, we are not parsing an error any more (if we were)
+      @parsing_error = false if line.include?('INSTRUMENTATION_STATUS')
+
+      if (@parsing_error)
+        # if we have started with the stack trace continue with it
+        if (@current_result.stack_trace)
+          @current_result.stack_trace += "\n#{line.gsub("\n", '')}"
+        else
+          @current_result.failure_output += "\n#{line.gsub("\n", '')}"
+        end
+        return
+      end
+
+      if (line.include?('INSTRUMENTATION_STATUS: id='))
+        # if we have already started parsing a second block for the test, this
+        # is a new test
+        if (@test_occurences > 1)
+          @current_result.original_test = @current_test
+          @current_result.passed = @current_result.result_code == "0"
+          @test_results.push(@current_result)
+
+          print_result(@current_result, background_black: background_black)
+          clear
+        end
+
+        @test_occurences += 1
+      elsif line.include?(PREFIX_TEST)
+        @current_test.name = strip(line, PREFIX_TEST)
+      elsif line.include?(PREFIX_CLASS)
+        @current_test.class_name = strip(line, PREFIX_CLASS)
+      elsif line.include?(PREFIX_CODE)
+        @current_result.result_code = strip(line, PREFIX_CODE)
+      elsif line.include?(PREFIX_STREAM)
+        @parsing_error = true
+        @current_result.failure_output += "#{strip(line, PREFIX_STREAM)}"
+      elsif line.include?(PREFIX_STACK)
+        @parsing_error = true
+        @current_result.stack_trace = "#{strip(line, PREFIX_STACK)}"
+      end
+    end
+  end
+
+  def clear
+    @parsing_error = false
+    @test_occurences = 0
+    @current_test = Test.new()
+    @current_result = TestResult.new()
+  end
+
+  def strip(str, substring)
+    str.sub(substring, '').gsub("\n", '').gsub("\r", '')
+  end
+
+  def print_result(result, background_black: false)
+    class_name = result.original_test.class_name.split('.').last
+    class_name = '%30.30s' % class_name
+    print("#{class_name} ")
+
+    passed_string = " #{result.passed ? "P" : "F"} ".black
+    if (result.passed)
+      passed_string = passed_string.on_green
+    else
+      passed_string = passed_string.on_red
+    end
+    print passed_string
+
+    test_name_string = " #{result.original_test.name} ".yellow
+    if (background_black)
+      test_name_string = test_name_string.yellow.on_light_black
+    end
+    puts(test_name_string)
+
+    if (!result.passed)
+      result.failure_output.each_line do |line|
+        print("#{'%30.30s' % ""}     #{line}".red)
+      end
+    end
+  end
+
+end

data/lib/quokkadb/adbparser.rb

@@ -0,0 +1,39 @@
+require_relative './test.rb'
+
+module ADBParser
+
+  PREFIX_INST_STATUS = 'INSTRUMENTATION_STATUS: '
+  PREFIX_TEST = 'test='
+  PREFIX_CLASS = 'class='
+
+  def ADBParser.generate_tests(logs)
+    if (!logs.respond_to?('each_line'))
+      raise 'Logs are not enumerable'
+    end
+
+    tests = []
+    active_test = nil
+    logs.each_line do |line|
+      line.gsub!("\n", '')
+      if (!line.include?(PREFIX_INST_STATUS))
+        next
+      end
+
+      # Remove beginning of line
+      line.sub!(PREFIX_INST_STATUS, '')
+
+      if (line.start_with?(PREFIX_TEST))
+        active_test = Test.new(line.sub(PREFIX_TEST, ''))
+      elsif (line.start_with?(PREFIX_CLASS))
+        active_test.class_name = line.sub(PREFIX_CLASS, '')
+      end
+
+      if (active_test && active_test.data_complete?)
+        tests.push(active_test)
+        active_test = nil
+      end
+    end
+    tests
+  end
+
+end

data/lib/quokkadb.rb

@@ -0,0 +1,40 @@
+require 'quokkadb/adb.rb'
+require 'quokkadb/test.rb'
+require 'quokkadb/test_result.rb'
+require 'quokkadb/adb_test_parser.rb'
+require 'colorize'
+
+module QuokkADB
+
+  def QuokkADB.run(runner:, package:, annotation: nil, shards: nil, index: nil, serial: nil, background_black: false)
+    full_log = ""
+    parser = ADBTestParser.new()
+    ADB.run_tests(
+      runner: runner,
+      package: package,
+      annotation: annotation,
+      shards: shards,
+      index: index,
+      serial: serial
+    ) { |line|
+      full_log += line
+      parser.parse_line(line: line, background_black: background_black)
+    }
+
+    full_log.each_line do |log|
+      if (log.include?("FAILURES!!!") or log.include?("Process crashed while executing") or log.include?("INSTRUMENTATION_FAILED"))
+        puts("exit 1")
+        exit(1)
+      end
+    end
+
+    parser.test_results.each do |result|
+      if (!result.passed)
+        puts("exit 1")
+        exit(1)
+      end
+    end
+    puts("exit 0")
+  end
+
+end

metadata

@@ -0,0 +1,47 @@
+--- !ruby/object:Gem::Specification
+name: quokkadb
+version: !ruby/object:Gem::Version
+  version: '1.0'
+platform: ruby
+authors:
+- Joe Beveridge
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2017-02-15 00:00:00.000000000 Z
+dependencies: []
+description: ADB test parsing libraryr
+email: joe.beveridge@shopify.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/quokkadb.rb
+- lib/quokkadb/adb.rb
+- lib/quokkadb/adb_test_parser.rb
+- lib/quokkadb/adbparser.rb
+homepage: http://rubygems.org/gems/quokka
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.5.1
+signing_key: 
+specification_version: 4
+summary: ADB test parsing library
+test_files: []