async-rspec 1.12.2 → 1.16.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 8eb3614b4300ca02cbb6299fbeb1606f70a6d090536cde9b5e66ecb53b52833d
-  data.tar.gz: 46f5a934a79a1cd5d500a1f5377439e248960ba63f2c79d52f72ffc7117d3dec
+  metadata.gz: 99cc9f67cfd8ad26555a5805dc8aa76b58876272ea0aecd18c07434ee8304c18
+  data.tar.gz: de573e462396eb1ccd28a45f7bd345bd2ecd0e8e6ea814ec300775b51d16280d
 SHA512:
-  metadata.gz: 7b18b90865cc67ab259e53905e87fc1c3038bcdb28fd4ed4db46f0adead7e082d07a9d8dfcc7e8774c828aa4d245331ae438b669293186d771345f2ac24cb682
-  data.tar.gz: 1b74916a02bb261d73978d7b06b9149d8b8d247a45b050ac026a17d3ef7137b168a2ab889def16c071c11ea56e3eaa9f50ff6afb592baab52d055c07565f7229
+  metadata.gz: ef693569b536cbba80c35d7805de22cd2cda1046f125cb3516c432b6548587042269cc802989fb79d94aa032c96eb45c589fd275f1c891462fb71337affdcdf5
+  data.tar.gz: fc6a89c0a9cecc214e140fb9a3b0887f6304f8612aeb1b6e771babda377d993fa56e6f2ef223fd02800123f24fd64005ceea500f0420df1fc29d7da039152d2a

data/lib/async/rspec/leaks.rb

@@ -18,31 +18,10 @@
 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 # THE SOFTWARE.
 
+require 'rspec/files'
+
 module Async
 	module RSpec
-		module Leaks
-			def current_ios(gc: GC.start)
-				all_ios = ObjectSpace.each_object(::IO).to_a.sort_by(&:object_id)
-				
-				# We are not interested in ios that have been closed already:
-				return all_ios.reject{|io| io.closed?}
-			end
-		end
-		
-		::RSpec.shared_context Leaks do
-			include Leaks
-			
-			let(:before_ios) {current_ios}
-			let(:after_ios) {current_ios}
-			
-			# We use around(:each) because it's the highest priority.
-			around(:each) do |example|
-				before_ios
-				
-				example.run.tap do
-					expect(after_ios).to be == before_ios
-				end
-			end
-		end
+		Leaks = ::RSpec::Files::Leaks
 	end
 end

data/lib/async/rspec/memory.rb

@@ -18,12 +18,10 @@
 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 # THE SOFTWARE.
 
-require_relative 'memory/limit_allocations'
+require 'rspec/memory'
 
 module Async
 	module RSpec
-		::RSpec.shared_context Memory do
-			include Memory
-		end
+		Memory = ::RSpec::Memory
 	end
 end

data/lib/async/rspec/reactor.rb

@@ -20,49 +20,56 @@
 
 require_relative 'leaks'
 
+require 'kernel/async'
 require 'async/reactor'
-require 'async/debug/selector'
 
 module Async
 	module RSpec
 		module Reactor
-			def run_example(reactor, example, duration)
+			def notify_failure(exception = $!)
+				::RSpec::Support.notify_failure(exception)
+			end
+			
+			def run_in_reactor(reactor, duration = nil)
 				result = nil
 				
+				timer_task = nil
+				
+				if duration
+					timer_task = reactor.async do |task|
+						# Wait for the timeout, at any point this task might be cancelled if the user code completes:
+						task.annotate("timer task duration=#{duration}")
+						task.sleep(duration)
+						
+						# The timeout expired, so generate an error:
+						buffer = StringIO.new
+						reactor.print_hierarchy(buffer)
+						
+						# Raise an error so it is logged:
+						raise TimeoutError, "run time exceeded duration #{duration}s:\r\n#{buffer.string}"
+					end
+				end
+				
 				reactor.run do |task|
 					task.annotate(self.class)
 					
-					reactor = task.reactor
-					timer = nil
-					
-					if duration
-						timer = reactor.async do |task|
-							task.annotate("timer task duration=#{duration}")
-							task.sleep(duration)
-							
-							buffer = StringIO.new
-							reactor.print_hierarchy(buffer)
-							
-							reactor.stop
-							
-							raise TimeoutError, "run time exceeded duration #{duration}s:\r\n#{buffer.string}"
-						end
-					end
-					
-					task.async do |spec_task|
-						spec_task.annotate("example runner")
+					spec_task = task.async do |spec_task|
+						spec_task.annotate("running example")
 						
-						result = example.run
+						result = yield
 						
-						if result.is_a? Exception
-							reactor.stop
-						else
-							spec_task.children&.each(&:wait)
-						end
-					end.wait
+						timer_task&.stop
+						
+						raise Async::Stop
+					end
 					
-					timer.stop if timer
-				end
+					begin
+						timer_task&.wait
+						spec_task.wait
+					ensure
+						spec_task.stop
+					end
+				end.wait
 				
 				return result
 			end
@@ -70,8 +77,7 @@ module Async
 		
 		::RSpec.shared_context Reactor do
 			include Reactor
-			
-			let(:reactor) {Async::Reactor.new(selector: Async::Debug::Selector.new)}
+			let(:reactor) {Async::Reactor.new}
 			
 			include_context Async::RSpec::Leaks
 			
@@ -79,7 +85,9 @@ module Async
 				duration = example.metadata.fetch(:timeout, 10)
 				
 				begin
-					run_example(reactor, example, duration)
+					run_in_reactor(reactor, duration) do
+						example.run
+					end
 				ensure
 					reactor.close
 				end

data/lib/async/rspec/ssl.rb

@@ -41,8 +41,8 @@ module Async
 		
 		::RSpec.shared_context SSL::CertificateAuthority do
 			# This key size is generally considered insecure, but it's fine for testing.
-			let(:certificate_authority_key) {OpenSSL::PKey::RSA.new(1024)}
-			let(:certificate_authority_name) {OpenSSL::X509::Name.parse("O=Test/CN=localhost")}
+			let(:certificate_authority_key) {OpenSSL::PKey::RSA.new(2048)}
+			let(:certificate_authority_name) {OpenSSL::X509::Name.parse("O=TestCA/CN=localhost")}
 
 			# The certificate authority is used for signing and validating the certificate which is used for communciation:
 			let(:certificate_authority) do
@@ -83,7 +83,7 @@ module Async
 			include_context SSL::CertificateAuthority
 			
 			# The private key to use on the server side:
-			let(:key) {OpenSSL::PKey::RSA.new(1024)}
+			let(:key) {OpenSSL::PKey::RSA.new(2048)}
 			let(:certificate_name) {OpenSSL::X509::Name.parse("O=Test/CN=localhost")}
 
 			# The certificate used for actual communication:
@@ -115,7 +115,7 @@ module Async
 			
 			let(:keys) do
 				Hash[
-					hosts.collect{|name| [name, OpenSSL::PKey::RSA.new(1024)]}
+					hosts.collect{|name| [name, OpenSSL::PKey::RSA.new(2048)]}
 				]
 			end
 			
@@ -177,8 +177,8 @@ module Async
 			include_context SSL::CertificateAuthority
 			
 			# The private key to use on the server side:
-			let(:key) {OpenSSL::PKey::RSA.new(1024)}
-			let(:invalid_key) {OpenSSL::PKey::RSA.new(1024)}
+			let(:key) {OpenSSL::PKey::RSA.new(2048)}
+			let(:invalid_key) {OpenSSL::PKey::RSA.new(2048)}
 			let(:certificate_name) {OpenSSL::X509::Name.parse("O=Test/CN=localhost")}
 
 			# The certificate used for actual communication:

data/lib/async/rspec/version.rb

@@ -20,6 +20,6 @@
 
 module Async
 	module RSpec
-		VERSION = "1.12.2"
+		VERSION = "1.16.0"
 	end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: async-rspec
 version: !ruby/object:Gem::Version
-  version: 1.12.2
+  version: 1.16.0
 platform: ruby
 authors:
 - Samuel Williams
-autorequire: 
+autorequire:
 bindir: bin
 cert_chain: []
-date: 2019-04-28 00:00:00.000000000 Z
+date: 2021-04-26 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rspec
@@ -24,22 +24,50 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '3.0'
+- !ruby/object:Gem::Dependency
+  name: rspec-files
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+- !ruby/object:Gem::Dependency
+  name: rspec-memory
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
 - !ruby/object:Gem::Dependency
   name: async
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '1.8'
+        version: '1.24'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '1.8'
+        version: '1.24'
 - !ruby/object:Gem::Dependency
-  name: covered
+  name: async-io
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
@@ -67,48 +95,38 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '0'
 - !ruby/object:Gem::Dependency
-  name: rake
+  name: covered
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: '10.0'
+        version: '0'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: '10.0'
-description: 
+        version: '0'
+description:
 email:
-- samuel.williams@oriontransfer.co.nz
 executables: []
 extensions: []
 extra_rdoc_files: []
 files:
-- ".editorconfig"
-- ".gitignore"
-- ".rspec"
-- ".travis.yml"
-- Gemfile
-- README.md
-- Rakefile
-- async-rspec.gemspec
 - lib/async/rspec.rb
 - lib/async/rspec/buffer.rb
 - lib/async/rspec/leaks.rb
 - lib/async/rspec/memory.rb
-- lib/async/rspec/memory/limit_allocations.rb
-- lib/async/rspec/memory/trace.rb
 - lib/async/rspec/profile.rb
 - lib/async/rspec/reactor.rb
 - lib/async/rspec/ssl.rb
 - lib/async/rspec/version.rb
 homepage: https://github.com/socketry/async-rspec
-licenses: []
+licenses:
+- MIT
 metadata: {}
-post_install_message: 
+post_install_message:
 rdoc_options: []
 require_paths:
 - lib
@@ -123,8 +141,8 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.0.3
-signing_key: 
+rubygems_version: 3.2.3
+signing_key:
 specification_version: 4
 summary: Helpers for writing specs against the async gem.
 test_files: []

data/.editorconfig

@@ -1,6 +0,0 @@
-root = true
-
-[*]
-indent_style = tab
-indent_size = 2
-

data/.gitignore

@@ -1,12 +0,0 @@
-/.bundle/
-/.yardoc
-/Gemfile.lock
-/_yardoc/
-/coverage/
-/doc/
-/pkg/
-/spec/reports/
-/tmp/
-
-# rspec failure tracking
-.rspec_status

data/.rspec

@@ -1,3 +0,0 @@
---format documentation
---warnings
---require spec_helper

data/.travis.yml

@@ -1,19 +0,0 @@
-language: ruby
-cache: bundler
-
-matrix:
-  include:
-    - rvm: 2.3
-    - rvm: 2.4
-    - rvm: 2.5
-    - rvm: 2.6
-    - rvm: 2.6
-      env: COVERAGE=BriefSummary,Coveralls
-    - rvm: ruby-head
-    - rvm: jruby-head
-      env: JRUBY_OPTS="--debug -X+O"
-    - rvm: truffleruby
-  allow_failures:
-    - rvm: ruby-head
-    - rvm: jruby-head
-    - rvm: truffleruby

data/Gemfile

@@ -1,8 +0,0 @@
-source 'https://rubygems.org'
-
-# Specify your gem's dependencies in async-rspec.gemspec
-gemspec
-
-group :test do
-	gem "ruby-prof", platform: :mri
-end

data/README.md

@@ -1,144 +0,0 @@
-# Async::RSpec
-
-Provides useful `RSpec.shared_context`s for testing code that builds on top of [async].
-
-[async]: https://github.com/socketry/async
-
-[![Build Status](https://secure.travis-ci.org/socketry/async-rspec.svg)](http://travis-ci.org/socketry/async-rspec)
-[![Code Climate](https://codeclimate.com/github/socketry/async-rspec.svg)](https://codeclimate.com/github/socketry/async-rspec)
-[![Coverage Status](https://coveralls.io/repos/socketry/async-rspec/badge.svg)](https://coveralls.io/r/socketry/async-rspec)
-
-## Installation
-
-Add this line to your application's Gemfile:
-
-```ruby
-gem 'async-rspec'
-```
-
-And then execute:
-
-	$ bundle
-
-Or install it yourself as:
-
-	$ gem install async-rspec
-	
-Finally, add this require statement to the top of `spec/spec_helper.rb`
-
-```ruby
-require 'async/rspec'
-```
-
-## Usage
-
-### Leaks
-
-Leaking sockets and other kinds of IOs are a problem for long running services. `Async::RSpec::Leaks` tracks all open sockets both before and after the spec. If any are left open, a `RuntimeError` is raised and the spec fails.
-
-```ruby
-RSpec.describe "leaky ios" do
-	include_context Async::RSpec::Leaks
-	
-	# The following fails:
-	it "leaks io" do
-		@input, @output = IO.pipe
-	end
-end
-```
-
-In some cases, the Ruby garbage collector will close IOs. In the above case, it's possible that just writing `IO.pipe` will not leak as Ruby will garbage collect the resulting IOs immediately. It's still incorrect to not close IOs, so don't depend on this behaviour.
-
-### Allocations
-
-Allocating large amounts of objects can lead to memery problems. `Async::RSpec::Memory` adds a `limit_allocations` matcher, which tracks the number of allocations and memory size for each object type and allows you to specify expected limits.
-
-```ruby
-RSpec.describe "memory allocations" do
-	include_context Async::RSpec::Memory
-	
-	it "limits allocation counts" do
-		expect do
-			6.times{String.new}
-		end.to limit_allocations(String => 10) # 10 strings can be allocated
-	end
-	
-	it "limits allocation counts (hash)" do
-		expect do
-			6.times{String.new}
-		end.to limit_allocations(String => {count: 10}) # 10 strings can be allocated
-	end
-	
-	it "limits allocation size" do
-		expect do
-			6.times{String.new("foo")}
-		end.to limit_allocations(String => {size: 1024}) # 1 KB of strings can be allocated
-	end
-end
-```
-
-### Reactor
-
-Many specs need to run within a reactor. A shared context is provided which includes all the relevant bits, including the above leaks checks. If your spec fails to run in less than 10 seconds, an `Async::TimeoutError` raises to prevent your test suite from hanging.
-
-```ruby
-require 'async/io'
-
-RSpec.describe Async::IO do
-	include_context Async::RSpec::Reactor
-	
-	let(:pipe) {IO.pipe}
-	let(:input) {Async::IO::Generic.new(pipe.first)}
-	let(:output) {Async::IO::Generic.new(pipe.last)}
-	
-	it "should send and receive data within the same reactor" do
-		message = nil
-		
-		output_task = reactor.async do
-			message = input.read(1024)
-		end
-		
-		reactor.async do
-			output.write("Hello World")
-		end
-		
-		output_task.wait
-		expect(message).to be == "Hello World"
-		
-		input.close
-		output.close
-	end
-end
-```
-
-## Contributing
-
-1. Fork it
-2. Create your feature branch (`git checkout -b my-new-feature`)
-3. Commit your changes (`git commit -am 'Add some feature'`)
-4. Push to the branch (`git push origin my-new-feature`)
-5. Create new Pull Request
-
-## License
-
-Released under the MIT license.
-
-Copyright, 2017, by [Samuel G. D. Williams](http://www.codeotaku.com/samuel-williams).
-
-Permission is hereby granted, free of charge, to any person obtaining a copy
-of this software and associated documentation files (the "Software"), to deal
-in the Software without restriction, including without limitation the rights
-to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-copies of the Software, and to permit persons to whom the Software is
-furnished to do so, subject to the following conditions:
-
-The above copyright notice and this permission notice shall be included in
-all copies or substantial portions of the Software.
-
-THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
-THE SOFTWARE.

data/Rakefile

@@ -1,6 +0,0 @@
-require "bundler/gem_tasks"
-require "rspec/core/rake_task"
-
-RSpec::Core::RakeTask.new(:test)
-
-task :default => :test

data/async-rspec.gemspec

@@ -1,27 +0,0 @@
-# coding: utf-8
-require_relative 'lib/async/rspec/version'
-
-Gem::Specification.new do |spec|
-	spec.name          = "async-rspec"
-	spec.version       = Async::RSpec::VERSION
-	spec.authors       = ["Samuel Williams"]
-	spec.email         = ["samuel.williams@oriontransfer.co.nz"]
-
-	spec.summary       = "Helpers for writing specs against the async gem."
-	spec.homepage      = "https://github.com/socketry/async-rspec"
-
-	spec.files         = `git ls-files -z`.split("\x0").reject do |f|
-		f.match(%r{^(test|spec|features)/})
-	end
-	
-	spec.require_paths = ["lib"]
-	
-	spec.add_dependency "rspec", "~> 3.0"
-	
-	# Since we test the shared contexts, we need some bits of async:
-	spec.add_development_dependency "async", "~> 1.8"
-	
-	spec.add_development_dependency "covered"
-	spec.add_development_dependency "bundler"
-	spec.add_development_dependency "rake", "~> 10.0"
-end

data/lib/async/rspec/memory/limit_allocations.rb

@@ -1,115 +0,0 @@
-# Copyright, 2018, by Samuel G. D. Williams. <http://www.codeotaku.com>
-# Copyright, 2018, by Janko Marohnić.
-# 
-# Permission is hereby granted, free of charge, to any person obtaining a copy
-# of this software and associated documentation files (the "Software"), to deal
-# in the Software without restriction, including without limitation the rights
-# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-# copies of the Software, and to permit persons to whom the Software is
-# furnished to do so, subject to the following conditions:
-# 
-# The above copyright notice and this permission notice shall be included in
-# all copies or substantial portions of the Software.
-# 
-# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
-# THE SOFTWARE.
-
-require_relative 'trace'
-
-require 'rspec/expectations'
-
-module Async
-	module RSpec
-		module Memory
-			# expect{...}.to allocate < 10.megabytes
-			#
-			class LimitAllocations
-				include ::RSpec::Matchers::Composable
-				
-				def initialize(allocations = {}, count: nil, size: nil)
-					@count = count
-					@size = size
-					
-					@allocations = {}
-					@errors = []
-					
-					allocations.each do |klass, count|
-						self.of(klass, count: count)
-					end
-				end
-				
-				def supports_block_expectations?
-					true
-				end
-				
-				def of(klass, **limits)
-					@allocations[klass] = limits
-					
-					return self
-				end
-				
-				private def check(value, limit)
-					case limit
-					when Range
-						unless limit.include? value
-							yield "expected within #{limit}"
-						end
-					when Integer
-						unless value == limit
-							yield "expected exactly #{limit}"
-						end
-					end
-				end
-				
-				def matches?(given_proc)
-					return true unless trace = Trace.capture(@allocations.keys, &given_proc)
-					
-					if @count or @size
-						# If the spec specifies a total limit, we have a limit which we can enforce which takes all allocations into account:
-						total = trace.total
-						
-						check(total.count, @count) do |expected|
-							@errors << "allocated #{total.count} instances, #{total.size} bytes, #{expected} instances"
-						end if @count
-						
-						check(total.size, @size) do |expected|
-							@errors << "allocated #{total.count} instances, #{total.size} bytes, #{expected} bytes"
-						end if @size
-					else
-						# Otherwise unspecified allocations are considered an error:
-						trace.ignored.each do |klass, allocation|
-							@errors << "allocated #{allocation.count} #{klass} instances, #{allocation.size} bytes, but it was not specified"
-						end
-					end
-					
-					trace.allocated.each do |klass, allocation|
-						next unless acceptable = @allocations[klass]
-						
-						check(allocation.count, acceptable[:count]) do |expected|
-							@errors << "allocated #{allocation.count} #{klass} instances, #{allocation.size} bytes, #{expected} instances"
-						end
-						
-						check(allocation.size, acceptable[:size]) do |expected|
-							@errors << "allocated #{allocation.count} #{klass} instances, #{allocation.size} bytes, #{expected} bytes"
-						end
-					end
-					
-					return @errors.empty?
-				end
-				
-				def failure_message
-					"exceeded allocation limit: #{@errors.join(', ')}"
-				end
-			end
-			
-			def limit_allocations(*args)
-				LimitAllocations.new(*args)
-			end
-		end
-	end
-end

data/lib/async/rspec/memory/trace.rb

@@ -1,140 +0,0 @@
-# Copyright, 2017, by Samuel G. D. Williams. <http://www.codeotaku.com>
-# 
-# Permission is hereby granted, free of charge, to any person obtaining a copy
-# of this software and associated documentation files (the "Software"), to deal
-# in the Software without restriction, including without limitation the rights
-# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-# copies of the Software, and to permit persons to whom the Software is
-# furnished to do so, subject to the following conditions:
-# 
-# The above copyright notice and this permission notice shall be included in
-# all copies or substantial portions of the Software.
-# 
-# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
-# THE SOFTWARE.
-
-require 'objspace'
-
-module Async
-	module RSpec
-		module Memory
-			Allocation = Struct.new(:count, :size) do
-				SLOT_SIZE = 40
-				
-				def << object
-					self.count += 1
-					
-					# We don't want to force specs to take the slot size into account.
-					self.size += ObjectSpace.memsize_of(object) - SLOT_SIZE
-				end
-				
-				def self.default_hash
-					Hash.new{|h,k| h[k] = Allocation.new(0, 0)}
-				end
-			end
-			
-			class Trace
-				def self.supported?
-					# There are issues on truffleruby-1.0.0rc9
-					return false if RUBY_ENGINE == "truffleruby"
-					
-					ObjectSpace.respond_to?(:trace_object_allocations)
-				end
-				
-				if supported?
-					def self.capture(*args, &block)
-						self.new(*args).tap do |trace|
-							trace.capture(&block)
-						end
-					end
-				else
-					def self.capture(*args, &block)
-						yield
-						
-						return nil
-					end
-				end
-				
-				def initialize(klasses)
-					@klasses = klasses
-					
-					@allocated = Allocation.default_hash
-					@retained = Allocation.default_hash
-					
-					@ignored = Allocation.default_hash
-					
-					@total = Allocation.new(0, 0)
-				end
-				
-				attr :allocated
-				attr :retained
-				
-				attr :ignored
-				
-				attr :total
-				
-				def current_objects(generation)
-					allocations = []
-					
-					ObjectSpace.each_object do |object|
-						if ObjectSpace.allocation_generation(object) == generation
-							allocations << object
-						end
-					end
-					
-					return allocations
-				end
-				
-				def find_base(object)
-					@klasses.find{|klass| object.is_a? klass}
-				end
-				
-				def capture(&block)
-					GC.start
-					
-					begin
-						GC.disable
-						
-						generation = GC.count
-						ObjectSpace.trace_object_allocations(&block)
-						
-						allocated = current_objects(generation)
-					ensure
-						GC.enable
-					end
-					
-					GC.start
-					retained = current_objects(generation)
-					
-					# All allocated objects, including those freed in the last GC:
-					allocated.each do |object|
-						if klass = find_base(object)
-							@allocated[klass] << object
-						else
-							# If the user specified classes, but we can't pin this allocation to a specific class, we issue a warning.
-							if @klasses.any?
-								warn "Ignoring allocation of #{object.class} at #{ObjectSpace.allocation_sourcefile(object)}:#{ObjectSpace.allocation_sourceline(object)}"
-							end
-							
-							@ignored[object.class] << object
-						end
-						
-						@total << object
-					end
-					
-					# Retained objects are still alive after a final GC:
-					retained.each do |object|
-						if klass = find_base(object)
-							@retained[klass] << object
-						end
-					end
-				end
-			end
-		end
-	end
-end