From be056f544c43a3b59147fc28eb51ecdb90f59c7c Mon Sep 17 00:00:00 2001
From: Stephen Margheim <stephen.margheim@gmail.com>
Date: Tue, 23 Jan 2024 01:53:11 +0100
Subject: [PATCH] Add some basic Rake task tests

---
 test/tasks/test_litestream_tasks.rb | 69 +++++++++++++++++++++++++++++
 1 file changed, 69 insertions(+)
 create mode 100644 test/tasks/test_litestream_tasks.rb

diff --git a/test/tasks/test_litestream_tasks.rb b/test/tasks/test_litestream_tasks.rb
new file mode 100644
index 0000000..22da216
--- /dev/null
+++ b/test/tasks/test_litestream_tasks.rb
@@ -0,0 +1,69 @@
+# test/tasks/rake_task_file_test.rb
+
+require 'test_helper'
+require 'rake'
+
+class RakeTaskFileTaskTest < ActiveSupport::TestCase
+  def setup
+    Rake.application.rake_require "tasks/litestream_tasks"
+    Rake::Task.define_task(:environment)
+    Rake::Task["litestream:env"].reenable
+    Rake::Task["litestream:replicate"].reenable
+  end
+
+  def teardown
+    Litestream.configuration = nil
+    ARGV.replace []
+  end
+
+  def test_env_task_when_nothing_configured_warns
+    out, err = capture_io do
+      Rake.application.invoke_task "litestream:env"
+    end
+
+    assert_equal "", out
+    assert_equal "You have not configured the Litestream gem with any values to generate ENV variables\n", err
+  end
+
+  def test_env_task_when_configured_outputs_env_variables
+    Litestream.configure do |config|
+      config.database_path = "path/to/database"
+    end
+
+    out, err = capture_io do
+      Rake.application.invoke_task "litestream:env"
+    end
+
+    assert_equal "LITESTREAM_DATABASE_PATH=path/to/database\nLITESTREAM_REPLICA_URL=\nLITESTREAM_ACCESS_KEY_ID=\nLITESTREAM_SECRET_ACCESS_KEY=\n", out
+    assert_equal "", err
+  end
+
+  def test_replicate_task_with_no_arguments
+    fake = Minitest::Mock.new
+    fake.expect :call, nil, [{}]
+    Litestream::Commands.stub :replicate, fake do
+      Rake.application.invoke_task "litestream:replicate"
+    end
+    fake.verify
+  end
+
+  def test_replicate_task_with_arguments
+    ARGV.replace ["--", "--no-expand-env"]
+    fake = Minitest::Mock.new
+    fake.expect :call, nil, [{"--no-expand-env"=>nil}]
+    Litestream::Commands.stub :replicate, fake do
+      Rake.application.invoke_task "litestream:replicate"
+    end
+    fake.verify
+  end
+
+  def test_replicate_task_with_arguments_without_separator
+    ARGV.replace ["--no-expand-env"]
+    fake = Minitest::Mock.new
+    fake.expect :call, nil, [{}]
+    Litestream::Commands.stub :replicate, fake do
+      Rake.application.invoke_task "litestream:replicate"
+    end
+    fake.verify
+  end
+end