Refactor save! and load! methods.

lib/covered/persist.rb

@@ -24,7 +24,7 @@
 
 module Covered
 	class Persist < Wrapper
-		def initialize(output, path: ".covered")
+		def initialize(output, path = ".covered")
 			super(output)
 
 			@path = path
@@ -46,9 +46,7 @@ def serialize(coverage)
 			}
 		end
 
-		def enable
-			super
-
+		def load!(path = @path)
 			return unless File.exist?(@path)
 
 			# Load existing coverage information and mark all files:
@@ -57,9 +55,7 @@ def enable
 			end
 		end
 
-		def disable
-			super
-
+		def save!(path = @path)
 			# Dump all coverage:
 			File.open(@path, "w") do |file|
 				packer = make_packer(file)
@@ -72,6 +68,18 @@ def disable
 			end
 		end
 
+		def enable
+			super
+
+			load!
+		end
+
+		def disable
+			super
+
+			save!
+		end
+
 		def make_packer(io)
 			packer = MessagePack::Packer.new(io)
 			packer.register_type(0x00, Symbol, :to_msgpack_ext)