More test coverage.

Author: ioquatix

lib/memory/leak/cluster.rb

@@ -94,6 +94,8 @@ def sample!
 			#
 			# @yields {|process_id, monitor| ...} each process ID and monitor that is leaking or exceeds the memory limit.
 			def check!(&block)
+				return to_enum(__method__) unless block_given?
+				
 				self.sample!
 
 				leaking = []

lib/memory/leak/monitor.rb

@@ -4,6 +4,7 @@
 # Copyright, 2025, by Samuel Williams.
 
 require "console"
+require_relative "system"
 
 module Memory
 	module Leak

lib/memory/leak/system.rb

@@ -15,8 +15,6 @@ def self.total_memory_size
 							return total.to_i * 1024
 						end
 					end
-					
-					return nil
 				end
 			elsif RUBY_PLATFORM =~ /darwin/
 				def self.total_memory_size

test/memory/leak/cluster.rb

@@ -13,6 +13,20 @@
 
 	let(:cluster) {subject.new}
 
+	with "#as_json" do
+		it "generates a JSON representation" do
+			expect(cluster.as_json).to have_keys(
+				processes: be_a(Hash),
+				total_size: be_nil,
+				total_size_limit: be_nil,
+			)
+		end
+		
+		it "generates a JSON string" do
+			expect(JSON.dump(cluster)).to be == cluster.to_json
+		end
+	end
+	
 	with "a leaking child process" do
 		before do
 			@children = 3.times.map do
@@ -65,6 +79,8 @@
 			# 100 MiB limit:
 			cluster.total_size_limit = 1024*1024*100
 
+			expect(cluster.check!.to_a).to be(:empty?)
+			
 			big_child = children.values.first
 			big_monitor = cluster.processes[big_child.process_id]
 			small_child = children.values.last

test/memory/leak/monitor.rb

@@ -9,6 +9,24 @@
 describe Memory::Leak::Monitor do
 	let(:monitor) {subject.new}
 
+	with "#as_json" do
+		it "generates a JSON representation" do
+			expect(monitor.as_json).to have_keys(
+				process_id: be_a(Integer),
+				current_size: be_nil,
+				maximum_size: be_nil,
+				maximum_size_limit: be_nil,
+				threshold_size: be_a(Integer),
+				increase_count: be == 0,
+				increase_limit: be_a(Integer)
+			)
+		end
+		
+		it "generates a JSON string" do
+			expect(JSON.dump(monitor)).to be == monitor.to_json
+		end
+	end
+	
 	with "#sample!" do
 		it "can capture samples" do
 			3.times do