lita-digitalocean 0.0.1 → 1.0.0

data/spec/lita/handlers/digitalocean/domain_spec.rb

@@ -0,0 +1,85 @@
+require "spec_helper"
+
+describe Lita::Handlers::Digitalocean::Domain, lita_handler: true do
+  it { routes_command("do domains create example.com 10.10.10.10").to(:create) }
+  it { routes_command("do domains delete example.com").to(:delete) }
+  it { routes_command("do domains list").to(:list) }
+  it { routes_command("do domains show example.com").to(:show) }
+
+  let(:client) { instance_double("::DigitalOcean::API", domains: client_domains) }
+  let(:client_domains) { instance_double("::DigitalOcean::Resource::Domain") }
+
+  before do
+    Lita.config.handlers.digitalocean = Lita::Config.new
+    Lita.config.handlers.digitalocean.tap do |config|
+      config.client_id = "CLIENT_ID"
+      config.api_key = "API_KEY"
+    end
+
+    allow(Lita::Authorization).to receive(:user_in_group?).with(
+      user,
+      :digitalocean_admins
+    ).and_return(true)
+
+    allow(::DigitalOcean::API).to receive(:new).and_return(client)
+  end
+
+  describe "#create" do
+    it "creates a new DNS record set" do
+      allow(client_domains).to receive(:create).with(
+        name: "example.com",
+        ip_address: "10.0.0.0"
+      ).and_return(status: "OK", domain: { id: 123, name: "example.com" })
+      send_command("do domains create example.com 10.0.0.0")
+      expect(replies.last).to eq("Created new DNS record set for example.com.")
+    end
+  end
+
+  describe "#delete" do
+    it "deletes a DNS record set" do
+      allow(client_domains).to receive(:delete).with("123").and_return(status: "OK")
+      send_command("do domains delete 123")
+      expect(replies.last).to eq("Deleted DNS record set.")
+    end
+  end
+
+  describe "#list" do
+    it "lists all DNS record sets" do
+      allow(client_domains).to receive(:list).and_return(
+        status: "OK",
+        domains: [{
+          id: 123,
+          name: "example.com"
+        }, {
+          id: 456,
+          name: "another.example.com"
+        }]
+      )
+      send_command("do domains list")
+      expect(replies).to eq([
+        "ID: 123, Name: example.com",
+        "ID: 456, Name: another.example.com"
+      ])
+    end
+  end
+
+  describe "#show" do
+    it "responds with the details of the DNS record set" do
+      allow(client_domains).to receive(:show).with("123").and_return(
+        status: "OK",
+        domain: {
+          id: 123,
+          name: "example.com",
+          ttl: 1800,
+          live_zone_file: "LZF",
+          error: nil,
+          zone_file_with_error: nil
+        }
+      )
+      send_command("do domains show 123")
+      expect(replies.last).to eq(
+      "ID: 123, Name: example.com, TTL: 1800, Live Zone File: LZF, Error: , Zone File With Error: "
+      )
+    end
+  end
+end

data/spec/lita/handlers/digitalocean/droplet_spec.rb

@@ -0,0 +1,238 @@
+require "spec_helper"
+
+describe Lita::Handlers::Digitalocean::Droplet, lita_handler: true do
+  it do
+    routes_command(
+      "do droplets create example.com 512mb centos-5-8-x64 nyc1"
+    ).to(:create)
+  end
+  it { routes_command("do droplets delete 123").to(:delete) }
+  it { routes_command("do droplets delete 123 --scrub").to(:delete) }
+  it { routes_command("do droplets list").to(:list) }
+  it { routes_command("do droplets password reset 123").to(:password_reset) }
+  it { routes_command("do droplets power cycle 123").to(:power_cycle) }
+  it { routes_command("do droplets power off 123").to(:power_off) }
+  it { routes_command("do droplets power on 123").to(:power_on) }
+  it { routes_command("do droplets reboot 123").to(:reboot) }
+  it { routes_command("do droplets rebuild 123 456").to(:rebuild) }
+  it { routes_command("do droplets resize 123 1gb").to(:resize) }
+  it { routes_command("do droplets restore 123 456").to(:restore) }
+  it { routes_command("do droplets show 123").to(:show) }
+  it { routes_command("do droplets shutdown 123").to(:shutdown) }
+  it { routes_command("do droplets snapshot 123 'My Snapshot'").to(:snapshot) }
+
+  let(:client) { instance_double("::DigitalOcean::API", droplets: client_droplets) }
+  let(:client_droplets) { instance_double("::DigitalOcean::Resource::Droplet") }
+
+  let(:do_ok) { { status: "OK" } }
+
+  before do
+    Lita.config.handlers.digitalocean = Lita::Config.new
+    Lita.config.handlers.digitalocean.tap do |config|
+      config.client_id = "CLIENT_ID"
+      config.api_key = "API_KEY"
+    end
+
+    allow(Lita::Authorization).to receive(:user_in_group?).with(
+      user,
+      :digitalocean_admins
+    ).and_return(true)
+
+    allow(::DigitalOcean::API).to receive(:new).and_return(client)
+  end
+
+  describe "#create" do
+    it "creates a new droplet using slugs" do
+      allow(client_droplets).to receive(:create).with(
+        name: "example.com",
+        size_slug: "512mb",
+        image_slug: "centos-5-8-x64",
+        region_slug: "nyc1"
+      ).and_return(status: "OK", droplet: { id: 123, name: "example.com" })
+      send_command("do droplets create example.com 512mb centos-5-8-x64 nyc1")
+      expect(replies.last).to eq("Created new droplet: 123 (example.com)")
+    end
+
+    it "creates a new droplet using IDs" do
+      allow(client_droplets).to receive(:create).with(
+        name: "example.com",
+        size_id: "1",
+        image_id: "2",
+        region_id: "3"
+      ).and_return(status: "OK", droplet: { id: 123, name: "example.com" })
+      send_command("do droplets create example.com 1 2 3")
+      expect(replies.last).to eq("Created new droplet: 123 (example.com)")
+    end
+
+    it "creates a new droplet with optional parameters" do
+      expect(client_droplets).to receive(:create).with(
+        hash_including(ssh_key_ids: "1,2,3", private_networking: true, backups_enabled: true)
+      ).and_return(status: "OK", droplet: { id: 123, name: "example.com" })
+      send_command <<-COMMAND.chomp
+do droplets create example.com 1 2 3 --ssh-key-ids 1,2,3 --private-networking --backups_enabled
+COMMAND
+    end
+  end
+
+  describe "#delete" do
+    it "deletes a droplet" do
+      allow(client_droplets).to receive(:delete).with("123", {}).and_return(do_ok)
+      send_command("do droplets delete 123")
+      expect(replies.last).to eq("Deleted droplet: 123")
+    end
+
+    it "scrubs the disk before deleting the droplet" do
+      allow(client_droplets).to receive(:delete).with(
+        "123", {
+          scrub_data: true
+        }).and_return(do_ok)
+      send_command("do droplets delete 123 --scrub")
+    end
+  end
+
+  describe "#list" do
+    let(:do_droplets) do
+      {
+        status: "OK",
+        droplets: [{
+          id: 123,
+          name: "image1",
+          ip_address: "1.2.3.4"
+        }, {
+          id: 456,
+          name: "image2",
+          ip_address: "5.6.7.8"
+        }]
+      }
+    end
+
+    it "lists all droplets" do
+      allow(client_droplets).to receive(:list).and_return(do_droplets)
+      send_command("do droplets list")
+      expect(replies).to eq([
+        "ID: 123, Name: image1, IP: 1.2.3.4",
+        "ID: 456, Name: image2, IP: 5.6.7.8"
+      ])
+    end
+  end
+
+  describe "#password_reset" do
+    it "resets the root password" do
+      allow(client_droplets).to receive(:password_reset).with("123").and_return(do_ok)
+      send_command("do droplets password reset 123")
+      expect(replies.last).to eq("Password reset for droplet: 123")
+    end
+  end
+
+  describe "#power_cycle" do
+    it "cycles the droplet's power" do
+      allow(client_droplets).to receive(:power_cycle).with("123").and_return(do_ok)
+      send_command("do droplets power cycle 123")
+      expect(replies.last).to eq("Power cycled for droplet: 123")
+    end
+  end
+
+  describe "#power_off" do
+    it "powers off the droplet" do
+      allow(client_droplets).to receive(:power_off).with("123").and_return(do_ok)
+      send_command("do droplets power off 123")
+      expect(replies.last).to eq("Powered off droplet: 123")
+    end
+  end
+
+  describe "#power_on" do
+    it "powers on the droplet" do
+      allow(client_droplets).to receive(:power_on).with("123").and_return(do_ok)
+      send_command("do droplets power on 123")
+      expect(replies.last).to eq("Powered on droplet: 123")
+    end
+  end
+
+  describe "#reboot" do
+    it "reboots the droplet" do
+      allow(client_droplets).to receive(:reboot).with("123").and_return(do_ok)
+      send_command("do droplets reboot 123")
+      expect(replies.last).to eq("Rebooted droplet: 123")
+    end
+  end
+
+  describe "#rebuild" do
+    it "rebuilds the droplet with the provided image" do
+      allow(client_droplets).to receive(:rebuild).with("123", image_id: "456").and_return(do_ok)
+      send_command("do droplets rebuild 123 456")
+      expect(replies.last).to eq("Rebuilt droplet: 123")
+    end
+  end
+
+  describe "#resize" do
+    it "resizes the droplet with the provided size ID" do
+      allow(client_droplets).to receive(:resize).with("123", size_id: "456").and_return(do_ok)
+      send_command("do droplets resize 123 456")
+      expect(replies.last).to eq("Resized droplet: 123")
+    end
+
+    it "resizes the droplet with the provided size slug" do
+      allow(client_droplets).to receive(:resize).with("123", size_slug: "1gb").and_return(do_ok)
+      send_command("do droplets resize 123 1gb")
+      expect(replies.last).to eq("Resized droplet: 123")
+    end
+  end
+
+  describe "#restore" do
+    it "restores the droplet to the provided image" do
+      allow(client_droplets).to receive(:restore).with("123", image_id: "456").and_return(do_ok)
+      send_command("do droplets restore 123 456")
+      expect(replies.last).to eq("Restored droplet: 123")
+    end
+  end
+
+  describe "#show" do
+    it "responds with the details of the droplet" do
+      allow(client_droplets).to receive(:show).with("123").and_return(
+        status: "OK",
+        droplet: {
+          id: 123,
+          image_id: 456,
+          name: "My Droplet",
+          region_id: 1,
+          size_id: 33,
+          backups_active: true,
+          backups: [],
+          snapshots: [],
+          ip_address: "1.2.3.4",
+          private_ip_address: "5.6.7.8",
+          locked: false,
+          status: "active"
+        }
+      )
+      send_command("do droplets show 123")
+      expect(replies.last).to eq <<-DROPLET.chomp
+ID: 123, Image ID: 456, Name: My Droplet, Region ID: 1, Size ID: 33, Backups active: true, \
+Backups: [], Snapshots: [], IP address: 1.2.3.4, Private IP address: 5.6.7.8, Locked: false, \
+Status: active
+DROPLET
+    end
+  end
+
+  describe "#shutdown" do
+    it "shuts down the droplet" do
+      allow(client_droplets).to receive(:shutdown).with("123").and_return(do_ok)
+      send_command("do droplets shutdown 123")
+      expect(replies.last).to eq("Shut down droplet: 123")
+    end
+  end
+
+  describe "#snapshot" do
+    it "takes a snapshot of the droplet" do
+      allow(client_droplets).to receive(:snapshot).with("123", {}).and_return(do_ok)
+      send_command("do droplets snapshot 123")
+      expect(replies.last).to eq("Snapshotted droplet: 123")
+    end
+
+    it "takes a named snapshot of the droplet" do
+      allow(client_droplets).to receive(:snapshot).with("123", name: "My Droplet").and_return(do_ok)
+      send_command("do droplets snapshot 123 'My Droplet'")
+      expect(replies.last).to eq("Snapshotted droplet: 123")
+    end
+  end
+end

data/spec/lita/handlers/digitalocean/image_spec.rb

@@ -0,0 +1,121 @@
+require "spec_helper"
+
+describe Lita::Handlers::Digitalocean::Image, lita_handler: true do
+  it { routes_command("do images delete 123").to(:delete) }
+  it { routes_command("do images list").to(:list) }
+  it { routes_command("do images list filter").to(:list) }
+  it { routes_command("do images show 123").to(:show) }
+
+  let(:client) { instance_double("::DigitalOcean::API", images: client_images) }
+  let(:client_images) { instance_double("::DigitalOcean::Resource::Image") }
+
+  before do
+    Lita.config.handlers.digitalocean = Lita::Config.new
+    Lita.config.handlers.digitalocean.tap do |config|
+      config.client_id = "CLIENT_ID"
+      config.api_key = "API_KEY"
+    end
+
+    allow(Lita::Authorization).to receive(:user_in_group?).with(
+      user,
+      :digitalocean_admins
+    ).and_return(true)
+
+    allow(::DigitalOcean::API).to receive(:new).and_return(client)
+  end
+
+  let(:public_image) do
+    {
+      id: 1601,
+      name: "CentOS 5.8 x64",
+      slug: "centos-5-8-x64",
+      distribution: "CentOS",
+      public: true,
+      regions: [1, 2, 3, 4, 5, 6],
+      region_slugs: %w(nyc1 ams1 sfo1 nyc2 ams2 sgp1)
+    }
+  end
+
+    let(:public_image_message) do
+      <<-IMAGE.chomp
+ID: 1601, Name: CentOS 5.8 x64, Slug: centos-5-8-x64, Distribution: CentOS, Public: true, \
+Regions: [1,2,3,4,5,6], Region Slugs: [nyc1,ams1,sfo1,nyc2,ams2,sgp1]
+IMAGE
+    end
+
+    let(:do_delete) { { status: "OK" } }
+
+  describe "#delete" do
+    it "responds with a success message" do
+      allow(client_images).to receive(:delete).with("123").and_return(do_delete)
+      send_command("do images delete 123")
+      expect(replies.last).to eq("Deleted image: 123")
+    end
+  end
+
+  describe "#list" do
+    let(:private_image) do
+      {
+        "id" => 1602,
+        name: "CentOS 5.8 x32",
+        slug: "centos-5-8-x32",
+        distribution: "CentOS",
+        public: false,
+        regions: [1, 2, 3],
+        region_slugs: %w(nyc1 ams1 sfo1)
+      }
+    end
+
+    let(:do_images) do
+      {
+        status: "OK",
+        images: [public_image, private_image]
+      }
+    end
+
+    let(:do_public_images) do
+      {
+        status: "OK",
+        images: [public_image]
+      }
+    end
+
+    let(:do_private_images) do
+      {
+        status: "OK",
+        images: [private_image]
+      }
+    end
+
+    let(:private_image_message) do
+      <<-IMAGE.chomp
+ID: 1602, Name: CentOS 5.8 x32, Slug: centos-5-8-x32, Distribution: CentOS, Public: false, \
+Regions: [1,2,3], Region Slugs: [nyc1,ams1,sfo1]
+IMAGE
+    end
+
+    it "responds with a list of all available images" do
+      allow(client_images).to receive(:list).and_return(do_images)
+      send_command("do images list")
+      expect(replies).to eq([public_image_message, private_image_message])
+    end
+
+    it "responds with a list of only public images" do
+      allow(client_images).to receive(:list).with(
+        filter: "global"
+      ).and_return(do_public_images)
+      send_command("do images list global")
+      expect(replies.last).to eq(public_image_message)
+    end
+  end
+
+  describe "#show" do
+    it "responds with the details of the image" do
+      allow(client_images).to receive(:show).with("123").and_return(
+        { status: "OK", image: public_image }
+      )
+      send_command("do images show 123")
+      expect(replies.last).to eq(public_image_message)
+    end
+  end
+end

data/spec/lita/handlers/digitalocean/region_spec.rb

@@ -0,0 +1,44 @@
+require "spec_helper"
+
+describe Lita::Handlers::Digitalocean::Region, lita_handler: true do
+  it { routes_command("do regions list").to(:list) }
+
+  let(:client) { instance_double("::DigitalOcean::API", regions: client_regions) }
+  let(:client_regions) { instance_double("::DigitalOcean::Resource::Region") }
+
+  before do
+    Lita.config.handlers.digitalocean = Lita::Config.new
+    Lita.config.handlers.digitalocean.tap do |config|
+      config.client_id = "CLIENT_ID"
+      config.api_key = "API_KEY"
+    end
+
+    allow(Lita::Authorization).to receive(:user_in_group?).with(
+      user,
+      :digitalocean_admins
+    ).and_return(true)
+
+    allow(::DigitalOcean::API).to receive(:new).and_return(client)
+  end
+
+  let(:do_list) do
+    {
+      status: "OK",
+      regions: [
+        { id: 1, name: "New York 1", slug: "nyc1" },
+        { id: 2, name: "Amsterdam 1", slug: "ams1" }
+      ]
+    }
+  end
+
+  describe "#list" do
+    it "responds with a list of all regions" do
+      allow(client_regions).to receive(:list).and_return(do_list)
+      send_command("do regions list")
+      expect(replies).to eq ([
+        "ID: 1, Name: New York 1, Slug: nyc1",
+        "ID: 2, Name: Amsterdam 1, Slug: ams1"
+      ])
+    end
+  end
+end