test/test_helper.rb in vite_rails-1.0.8 vs test/test_helper.rb in vite_rails-1.0.9

@@ -1,34 +1,68 @@
 # frozen_string_literal: true
 
+require 'simplecov'
+SimpleCov.start {
+  add_filter '/test/'
+}
+
 require 'minitest/autorun'
 require 'rails'
 require 'rails/test_help'
-require 'byebug'
+require 'pry-byebug'
 
 require_relative 'test_app/config/environment'
 
 Rails.env = 'production'
 
-ViteRails.instance = ::ViteRails.new
+ViteRails.instance = ViteRails.new
 
-class ViteRails::Test < Minitest::Test
-private
-
-  def reloaded_config
-    ViteRails.instance.instance_variable_set(:@config, nil)
-    ViteRails.instance.instance_variable_set(:@dev_server, nil)
-    ViteRails.env = {}
-    ViteRails.config
-    ViteRails.dev_server
+module ViteRailsTestHelpers
+  def refresh_config(env_variables = ViteRails.load_env_variables)
+    ViteRails.env = env_variables
+    (ViteRails.instance = ViteRails.new).config
   end
 
   def with_rails_env(env)
     original = Rails.env
     Rails.env = ActiveSupport::StringInquirer.new(env)
-    reloaded_config
-    yield
+    yield(refresh_config)
   ensure
     Rails.env = ActiveSupport::StringInquirer.new(original)
-    reloaded_config
+    refresh_config
+  end
+
+  def test_app_path
+    File.expand_path('test_app', __dir__)
+  end
+
+  def with_dev_server_running(&block)
+    ViteRails.instance.stub(:dev_server_running?, true, &block)
+  end
+end
+
+class ViteRails::Test < Minitest::Test
+  include ViteRailsTestHelpers
+
+private
+
+  def assert_run_command(*argv, use_yarn: false, flags: [])
+    command = use_yarn ? %w[yarn vite] : ["#{ test_app_path }/node_modules/.bin/vite"]
+    cwd = Dir.pwd
+    Dir.chdir(test_app_path)
+
+    klass = ViteRails::Runner
+    instance = klass.new(argv)
+    mock = Minitest::Mock.new
+    mock.expect(:call, nil, [ViteRails.env, *command, *argv, *flags])
+
+    klass.stub(:new, instance) do
+      instance.stub(:executable_exists?, !use_yarn) do
+        Kernel.stub(:exec, mock) { ViteRails.run(argv) }
+      end
+    end
+
+    mock.verify
+  ensure
+    Dir.chdir(cwd)
   end
 end