1 files changed, 105 insertions, 0 deletions
diff --git a/test/rubygems/test_gem_commands_owner_command.rb b/test/rubygems/test_gem_commands_owner_command.rb
new file mode 100644
index 0000000000..fa4281c6a7
--- /dev/null
+++ b/test/rubygems/test_gem_commands_owner_command.rb
@@ -0,0 +1,105 @@
+require_relative 'gemutilities'
+require 'rubygems/commands/owner_command'
+
+class TestGemCommandsOwnerCommand < RubyGemTestCase
+
+  def setup
+    super
+
+    @fetcher = Gem::FakeFetcher.new
+    Gem::RemoteFetcher.fetcher = @fetcher
+    Gem.configuration.rubygems_api_key = "ed244fbf2b1a52e012da8616c512fa47f9aa5250"
+
+    @cmd = Gem::Commands::OwnerCommand.new
+  end
+
+  def test_show_owners
+    response = <<EOF
+---
+- email: [email protected]
+- email: [email protected]
+EOF
+
+    @fetcher.data["https://rubygems.org/api/v1/gems/freewill/owners.yaml"] = [response, 200, 'OK']
+
+    use_ui @ui do
+      @cmd.show_owners("freewill")
+    end
+
+    assert_equal Net::HTTP::Get, @fetcher.last_request.class
+    assert_equal Gem.configuration.rubygems_api_key, @fetcher.last_request["Authorization"]
+
+    assert_match %r{Owners for gem: freewill}, @ui.output
+    assert_match %r{- [email protected]}, @ui.output
+    assert_match %r{- [email protected]}, @ui.output
+  end
+
+  def test_show_owners_denied
+    response = "You don't have permission to push to this gem"
+    @fetcher.data["https://rubygems.org/api/v1/gems/freewill/owners.yaml"] = [response, 403, 'Forbidden']
+
+    assert_raises MockGemUi::TermError do
+      use_ui @ui do
+        @cmd.show_owners("freewill")
+      end
+    end
+
+    assert_match response, @ui.output
+  end
+
+  def test_add_owners
+    response = "Owner added successfully."
+    @fetcher.data["https://rubygems.org/api/v1/gems/freewill/owners"] = [response, 200, 'OK']
+
+    use_ui @ui do
+      @cmd.add_owners("freewill", ["[email protected]"])
+    end
+
+    assert_equal Net::HTTP::Post, @fetcher.last_request.class
+    assert_equal Gem.configuration.rubygems_api_key, @fetcher.last_request["Authorization"]
+    assert_equal "email=user-new1%40example.com", @fetcher.last_request.body
+
+    assert_match response, @ui.output
+  end
+
+  def test_add_owners_denied
+    response = "You don't have permission to push to this gem"
+    @fetcher.data["https://rubygems.org/api/v1/gems/freewill/owners"] = [response, 403, 'Forbidden']
+
+    assert_raises MockGemUi::TermError do
+      use_ui @ui do
+        @cmd.add_owners("freewill", ["[email protected]"])
+      end
+    end
+
+    assert_match response, @ui.output
+  end
+
+  def test_remove_owners
+    response = "Owner removed successfully."
+    @fetcher.data["https://rubygems.org/api/v1/gems/freewill/owners"] = [response, 200, 'OK']
+
+    use_ui @ui do
+      @cmd.remove_owners("freewill", ["[email protected]"])
+    end
+
+    assert_equal Net::HTTP::Delete, @fetcher.last_request.class
+    assert_equal Gem.configuration.rubygems_api_key, @fetcher.last_request["Authorization"]
+    assert_equal "email=user-remove1%40example.com", @fetcher.last_request.body
+
+    assert_match response, @ui.output
+  end
+
+  def test_remove_owners_denied
+    response = "You don't have permission to push to this gem"
+    @fetcher.data["https://rubygems.org/api/v1/gems/freewill/owners"] = [response, 403, 'Forbidden']
+
+    assert_raises MockGemUi::TermError do
+      use_ui @ui do
+        @cmd.remove_owners("freewill", ["[email protected]"])
+      end
+    end
+
+    assert_match response, @ui.output
+  end
+end