File: in_sub_process.rb

package info (click to toggle)
ruby-rspec 3.12.0c0e1m1s0-1
  • links: PTS, VCS
  • area: main
  • in suites: bookworm
  • size: 6,752 kB
  • sloc: ruby: 69,818; sh: 1,861; makefile: 99
file content (67 lines) | stat: -rw-r--r-- 2,177 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
module RSpec
  module Support
    module InSubProcess
      if Process.respond_to?(:fork) && !(Ruby.jruby? && RUBY_VERSION == '1.8.7')

        UnmarshableObject = Struct.new(:error)

        # Useful as a way to isolate a global change to a subprocess.

        def in_sub_process(prevent_warnings=true) # rubocop:disable Metrics/MethodLength, Metrics/AbcSize
          exception_reader, exception_writer = IO.pipe
          result_reader, result_writer = IO.pipe

          pid = Process.fork do
            warning_preventer = $stderr = RSpec::Support::StdErrSplitter.new($stderr)

            begin
              result = yield
              warning_preventer.verify_no_warnings! if prevent_warnings
              # rubocop:disable Lint/HandleExceptions
            rescue Support::AllExceptionsExceptOnesWeMustNotRescue => exception
              # rubocop:enable Lint/HandleExceptions
            end

            exception_writer.write marshal_dump_with_unmarshable_object_handling(exception)
            exception_reader.close
            exception_writer.close

            result_writer.write marshal_dump_with_unmarshable_object_handling(result)
            result_reader.close
            result_writer.close

            exit! # prevent at_exit hooks from running (e.g. minitest)
          end

          exception_writer.close
          result_writer.close
          Process.waitpid(pid)

          exception = Marshal.load(exception_reader.read)
          exception_reader.close
          raise exception if exception

          result = Marshal.load(result_reader.read)
          result_reader.close
          result
        end
        alias :in_sub_process_if_possible :in_sub_process

        def marshal_dump_with_unmarshable_object_handling(object)
          Marshal.dump(object)
        rescue TypeError => error
          Marshal.dump(UnmarshableObject.new(error))
        end
      else
        def in_sub_process(*)
          skip "This spec requires forking to work properly, " \
               "and your platform does not support forking"
        end

        def in_sub_process_if_possible(*)
          yield
        end
      end
    end
  end
end