require "time"

module Petitest
  module SubscriberConcerns
    module TimeConcern
      # @return [Time, nil]
      attr_accessor :finished_at

      # @return [Time, nil]
      attr_accessor :started_at

      # @note Override
      def after_running_test_plan(test_plan)
        super
        self.finished_at = ::Time.now
      end

      # @note Override
      def before_running_test_plan(test_plan)
        super
        self.started_at = ::Time.now
      end
    end
  end
end