esp_sdk 1.1.0 → 2.0.0.rc1

data/test/esp/resources/suppression/signature_test.rb

@@ -0,0 +1,117 @@
+require File.expand_path(File.dirname(__FILE__) + '/../../../test_helper')
+
+module ESP
+  class Suppression
+    class SignatureTest < ActiveSupport::TestCase
+      context ESP::Suppression::Signature do
+        context '#find' do
+          should 'not be implemented' do
+            assert_raises ESP::NotImplementedError do
+              ESP::Suppression::Signature.find(4)
+            end
+          end
+        end
+
+        context '#update' do
+          should 'not be implemented' do
+            s = ESP::Suppression::Signature.new
+            assert_raises ESP::NotImplementedError do
+              s.update
+            end
+          end
+        end
+
+        context '#destroy' do
+          should 'not be implemented' do
+            s = ESP::Suppression::Signature.new
+            assert_raises ESP::NotImplementedError do
+              s.destroy
+            end
+          end
+        end
+
+        context '#create' do
+          should 'call the api' do
+            stub_request(:post, %r{suppressions/signatures.json*}).to_return(body: json(:suppression_signature))
+
+            suppression = ESP::Suppression::Signature.create(signature_ids: [4, 2], custom_signature_ids: [3], regions: ['us_east_1'], external_account_ids: [5], reason: 'because')
+
+            assert_requested(:post, %r{suppressions/signatures.json*}) do |req|
+              body = JSON.parse(req.body)
+              assert_equal 'because', body['data']['attributes']['reason']
+              assert_equal [4, 2], body['data']['attributes']['signature_ids']
+              assert_equal [3], body['data']['attributes']['custom_signature_ids']
+              assert_equal ['us_east_1'], body['data']['attributes']['regions']
+              assert_equal [5], body['data']['attributes']['external_account_ids']
+            end
+            assert_equal ESP::Suppression::Signature, suppression.class
+          end
+
+          context 'for alert' do
+            should 'call the api' do
+              stub_request(:post, %r{suppressions/alert/5/signatures.json*}).to_return(body: json(:suppression_signature))
+
+              suppression = ESP::Suppression::Signature.create(alert_id: 5, reason: 'because')
+
+              assert_requested(:post, %r{suppressions/alert/5/signatures.json*}) do |req|
+                body = JSON.parse(req.body)
+                assert_equal 'because', body['data']['attributes']['reason']
+              end
+              assert_equal ESP::Suppression::Signature, suppression.class
+            end
+          end
+        end
+
+        context 'live calls' do
+          setup do
+            skip "Make sure you run the live calls locally to ensure proper integration" if ENV['CI_SERVER']
+            WebMock.allow_net_connect!
+          end
+
+          teardown do
+            WebMock.disable_net_connect!
+          end
+
+          context '.create' do
+            should 'return error when reason is not supplied' do
+              signature_id = ESP::Signature.last.id
+              external_account_id = ESP::ExternalAccount.last.id
+
+              suppression = ESP::Suppression::Signature.create(signature_ids: [signature_id], custom_signature_ids: [], regions: ['us_east_1'], external_account_ids: [external_account_id])
+
+              assert_equal "Reason can't be blank", suppression.errors.full_messages.first
+            end
+
+            should 'return suppression' do
+              signature_id = ESP::Signature.last.id
+              external_account_id = ESP::ExternalAccount.last.id
+
+              suppression = ESP::Suppression::Signature.create(signature_ids: [signature_id], custom_signature_ids: [], reason: 'test', regions: ['us_east_1'], external_account_ids: [external_account_id])
+
+              assert_equal ESP::Suppression::Signature, suppression.class
+            end
+
+            context 'for_alert' do
+              should 'return error when reason is not supplied' do
+                alert_id = ESP::Report.last.alerts.last.id
+
+                suppression = ESP::Suppression::Signature.create(alert_id: alert_id)
+
+                assert_equal "Reason can't be blank", suppression.errors.full_messages.first
+              end
+
+              should 'return suppression' do
+                alert_id = ESP::Report.last.alerts.last.id
+
+                suppression = ESP::Suppression::Signature.create(alert_id: alert_id, reason: 'test')
+
+                assert_predicate suppression.errors, :blank?
+                assert_equal ESP::Suppression::Signature, suppression.class
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/test/esp/resources/suppression/unique_identifier_test.rb

@@ -0,0 +1,79 @@
+require File.expand_path(File.dirname(__FILE__) + '/../../../test_helper')
+
+module ESP
+  class Suppression
+    class UniqueIdentifierTest < ActiveSupport::TestCase
+      context ESP::Suppression::UniqueIdentifier do
+        context '#find' do
+          should 'not be implemented' do
+            assert_raises ESP::NotImplementedError do
+              ESP::Suppression::UniqueIdentifier.find(4)
+            end
+          end
+        end
+
+        context '#update' do
+          should 'not be implemented' do
+            s = ESP::Suppression::UniqueIdentifier.new
+            assert_raises ESP::NotImplementedError do
+              s.update
+            end
+          end
+        end
+
+        context '#destroy' do
+          should 'not be implemented' do
+            s = ESP::Suppression::UniqueIdentifier.new
+            assert_raises ESP::NotImplementedError do
+              s.destroy
+            end
+          end
+        end
+
+        context '#create' do
+          should 'call the api' do
+            stub_request(:post, %r{suppressions/alert/5/unique_identifiers.json*}).to_return(body: json(:suppression_unique_identifier))
+
+            suppression = ESP::Suppression::UniqueIdentifier.create(alert_id: 5, reason: 'because')
+
+            assert_requested(:post, %r{suppressions/alert/5/unique_identifiers.json*}) do |req|
+              body = JSON.parse(req.body)
+              assert_equal 'because', body['data']['attributes']['reason']
+            end
+            assert_equal ESP::Suppression::UniqueIdentifier, suppression.class
+          end
+        end
+
+        context 'live calls' do
+          setup do
+            skip "Make sure you run the live calls locally to ensure proper integration" if ENV['CI_SERVER']
+            WebMock.allow_net_connect!
+          end
+
+          teardown do
+            WebMock.disable_net_connect!
+          end
+
+          context '.create' do
+            should 'return error when reason is not supplied' do
+              alert_id = ESP::Report.last.alerts.last.id
+
+              suppression = ESP::Suppression::UniqueIdentifier.create(alert_id: alert_id)
+
+              assert_equal "Reason can't be blank", suppression.errors.full_messages.first
+            end
+
+            should 'return suppression' do
+              alert_id = ESP::Report.last.alerts.last.id
+
+              suppression = ESP::Suppression::UniqueIdentifier.create(alert_id: alert_id, reason: 'test')
+
+              assert_predicate suppression.errors, :blank?
+              assert_equal ESP::Suppression::UniqueIdentifier, suppression.class
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/test/esp/resources/suppression_test.rb

@@ -0,0 +1,226 @@
+require File.expand_path(File.dirname(__FILE__) + '/../../test_helper')
+
+module ESP
+  class SuppressionTest < ActiveSupport::TestCase
+    context ESP::Suppression do
+      context '#create' do
+        should 'not be implemented' do
+          assert_raises ESP::NotImplementedError do
+            ESP::User.create(name: 'test')
+          end
+        end
+      end
+
+      context '#update' do
+        should 'not be implemented' do
+          u = build(:user)
+          assert_raises ESP::NotImplementedError do
+            u.save
+          end
+        end
+      end
+
+      context '#destroy' do
+        should 'not be implemented' do
+          u = build(:user)
+          assert_raises ESP::NotImplementedError do
+            u.destroy
+          end
+        end
+      end
+
+      context '#organization' do
+        should 'call the api' do
+          s = build(:suppression, organization_id: 1)
+          stub_org = stub_request(:get, %r{organizations/#{s.organization_id}.json*}).to_return(body: json(:organization))
+
+          s.organization
+
+          assert_requested(stub_org)
+        end
+      end
+
+      context '#created_by' do
+        should 'call the api' do
+          s = build(:suppression, created_by_id: 1)
+          stub_user = stub_request(:get, %r{users/#{s.created_by_id}.json*}).to_return(body: json(:user))
+
+          s.created_by
+
+          assert_requested(stub_user)
+        end
+      end
+
+      context '#regions' do
+        should 'call the api for the report and the passed in params' do
+          suppression = build(:suppression)
+          stub_request(:get, /regions.json*/).to_return(body: json_list(:region, 2))
+
+          suppression.regions
+
+          assert_requested(:get, /regions.json*/) do |req|
+            assert_equal "filter[suppressions_id_eq]=#{suppression.id}", URI.unescape(req.uri.query)
+          end
+        end
+      end
+
+      context '#external_accounts' do
+        should 'call the api for the report and the passed in params' do
+          suppression = build(:suppression)
+          stub_request(:get, /external_accounts.json*/).to_return(body: json_list(:external_account, 2))
+
+          suppression.external_accounts
+
+          assert_requested(:get, /external_accounts.json*/) do |req|
+            assert_equal "filter[suppressions_id_eq]=#{suppression.id}", URI.unescape(req.uri.query)
+          end
+        end
+      end
+
+      context '#signatures' do
+        should 'call the api for the report and the passed in params' do
+          suppression = build(:suppression, signature_ids: [1, 2])
+          stub_request(:get, /signatures.json*/).to_return(body: json_list(:signature, 2))
+
+          suppression.signatures
+
+          assert_requested(:get, /signatures.json*/) do |req|
+            assert_equal "filter[id_in][0]=#{suppression.signature_ids.first}&filter[id_in][1]=#{suppression.signature_ids.second}", URI.unescape(req.uri.query)
+          end
+        end
+      end
+
+      context '#custom_signatures' do
+        should 'call the api for the report and the passed in params' do
+          suppression = build(:suppression, custom_signature_ids: [1, 2])
+          stub_request(:get, /custom_signatures.json*/).to_return(body: json_list(:custom_signature, 2))
+
+          suppression.custom_signatures
+
+          assert_requested(:get, /custom_signatures.json*/) do |req|
+            assert_equal "filter[id_in][0]=#{suppression.custom_signature_ids.first}&filter[id_in][1]=#{suppression.custom_signature_ids.second}", URI.unescape(req.uri.query)
+          end
+        end
+      end
+
+      context '#deactivate' do
+        should 'call the api and not throw an error if an error is returned' do
+          s = build(:suppression)
+          error = ActiveResource::BadRequest.new('')
+          error_response = json(:error)
+          response = mock(body: error_response, code: '400')
+          error.stubs(:response).returns(response)
+          s.expects(:patch).raises(error)
+
+          assert_nothing_raised do
+            refute_predicate s, :deactivate
+          end
+
+          assert_equal JSON.parse(error_response)['errors'].first['title'], s.errors.full_messages.first
+        end
+      end
+
+      context '#deactivate!' do
+        should 'call the api and throw an error if an error is returned' do
+          s = build(:suppression)
+          error = ActiveResource::BadRequest.new('')
+          error_response = json(:error)
+          response = mock(body: error_response, code: '400')
+          error.stubs(:response).returns(response)
+          s.expects(:patch).raises(error)
+
+          error = assert_raises ActiveResource::ResourceInvalid do
+            s.deactivate!
+          end
+          assert_equal "Failed.  Response code = 400.  Response message = #{JSON.parse(error_response)['errors'].first['title']}.", error.message
+        end
+      end
+
+      context 'live calls' do
+        setup do
+          skip "Make sure you run the live calls locally to ensure proper integration" if ENV['CI_SERVER']
+          WebMock.allow_net_connect!
+          @s = ESP::Suppression.first
+          skip "Live DB does not have any suppressions.  Add a suppression and run tests again." if @s.blank?
+        end
+
+        teardown do
+          WebMock.disable_net_connect!
+        end
+
+        context '#organization' do
+          should 'return an organization' do
+            org = @s.organization
+
+            assert_equal @s.organization_id, org.id
+          end
+        end
+
+        context '#created_by' do
+          should 'return a user' do
+            u = @s.created_by
+
+            assert_equal @s.created_by_id, u.id
+          end
+        end
+
+        context '#regions' do
+          should 'return regions' do
+            r = @s.regions
+
+            assert_equal @s.relationships.regions.data.map(&:id).sort, r.map(&:id).sort
+          end
+        end
+
+        context '#external_accounts' do
+          should 'return external_accounts' do
+            e = @s.external_accounts
+
+            assert_equal @s.relationships.external_accounts.data.map(&:id).sort, e.map(&:id).sort
+          end
+        end
+
+        context '#signatures' do
+          should 'return signatures' do
+            s = @s.signatures
+
+            assert_equal @s.relationships.signatures.data.map(&:id).sort, s.map(&:id).sort
+          end
+        end
+
+        context '#custom_signatures' do
+          should 'return custom_signatures' do
+            cs = @s.custom_signatures
+
+            assert_equal @s.relationships.custom_signatures.data.map(&:id).sort, cs.map(&:id).sort
+          end
+        end
+
+        context '#deactivate' do
+          should 'deactivate the suppression' do
+            status = @s.status
+
+            @s.deactivate
+
+            assert_equal 'inactive', @s.status
+            assert_contains @s.errors.full_messages.first, 'Access Denied' if status == 'inactive'
+          end
+        end
+
+        context '#deactivate!' do
+          should 'raise an error if already inactive' do
+            status = @s.status
+
+            if status == 'inactive'
+              assert_raises ActiveResource::ResourceInvalid do
+                @s.deactivate!
+              end
+              assert_equal 'inactive', @s.status
+              assert_contains @s.errors.full_messages.first, 'Access Denied' if status == 'inactive'
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/test/esp/resources/tag_test.rb

@@ -0,0 +1,98 @@
+require File.expand_path(File.dirname(__FILE__) + '/../../test_helper')
+
+module ESP
+  class TagTest < ActiveSupport::TestCase
+    context ESP::Tag do
+      context '#create' do
+        should 'not be implemented' do
+          assert_raises ESP::NotImplementedError do
+            ESP::Tag.create(name: 'test')
+          end
+        end
+      end
+
+      context '#update' do
+        should 'not be implemented' do
+          tag = build(:tag)
+          assert_raises ESP::NotImplementedError do
+            tag.save
+          end
+        end
+      end
+
+      context '#destroy' do
+        should 'not be implemented' do
+          tag = build(:tag)
+          assert_raises ESP::NotImplementedError do
+            tag.destroy
+          end
+        end
+      end
+
+      context '.for_alert' do
+        should 'throw an error if alert id is not supplied' do
+          error = assert_raises ArgumentError do
+            ESP::Tag.for_alert
+          end
+          assert_equal 'You must supply an alert id.', error.message
+        end
+
+        should 'call the api' do
+          stub_tag = stub_request(:get, %r{alerts/5/tags.json*}).to_return(body: json_list(:tag, 2))
+
+          tags = ESP::Tag.for_alert(5)
+
+          assert_requested(stub_tag)
+          assert_equal ESP::Tag, tags.resource_class
+        end
+      end
+
+      context '.find' do
+        should 'throw an error if alert_id is not supplied' do
+          error = assert_raises ArgumentError do
+            ESP::Tag.find(:all, params: { id: 3 })
+          end
+          assert_equal 'You must supply an alert id.', error.message
+        end
+
+        should 'call the show api and return a tag if searching by id' do
+          stub_tag = stub_request(:get, %r{tags/5.json*}).to_return(body: json(:tag))
+
+          tag = ESP::Tag.find(5)
+
+          assert_requested(stub_tag)
+          assert_equal ESP::Tag, tag.class
+        end
+
+        should 'call the api and return tags when alert_id is supplied' do
+          stub_tag = stub_request(:get, %r{alerts/5/tags.json*}).to_return(body: json_list(:tag, 2))
+
+          tags = ESP::Tag.find(:all, params: { alert_id: 5 })
+
+          assert_requested(stub_tag)
+          assert_equal ESP::Tag, tags.resource_class
+        end
+      end
+
+      context 'live calls' do
+        setup do
+          skip "Make sure you run the live calls locally to ensure proper integration" if ENV['CI_SERVER']
+          WebMock.allow_net_connect!
+        end
+
+        teardown do
+          WebMock.disable_net_connect!
+        end
+
+        context '.for_alert' do
+          should 'return tags for alert id' do
+            report = ESP::Report.last
+            events = ESP::Tag.for_alert(report.alerts.last.id)
+
+            assert_equal ESP::Tag, events.resource_class
+          end
+        end
+      end
+    end
+  end
+end

data/test/esp/resources/team_test.rb

@@ -0,0 +1,140 @@
+require File.expand_path(File.dirname(__FILE__) + '/../../test_helper')
+
+module ESP
+  class TeamTest < ActiveSupport::TestCase
+    context ESP::Team do
+      context '#organization' do
+        should 'call the api' do
+          team = build(:team, organization_id: 4)
+          stub_org = stub_request(:get, %r{organizations/#{team.organization_id}.json*}).to_return(body: json(:organization))
+
+          team.organization
+
+          assert_requested(stub_org)
+        end
+      end
+
+      context '#sub_organization' do
+        should 'call the api' do
+          team = build(:team, sub_organization_id: 4)
+          stub_sub_org = stub_request(:get, %r{sub_organizations/#{team.sub_organization_id}.json*}).to_return(body: json(:sub_organization))
+
+          team.sub_organization
+
+          assert_requested(stub_sub_org)
+        end
+      end
+
+      context '#external_accounts' do
+        should 'call the api' do
+          team = build(:team)
+          stub_request(:get, /external_accounts.json*/).to_return(body: json_list(:external_account, 2))
+
+          team.external_accounts
+
+          assert_requested(:get, /external_accounts.json*/) do |req|
+            assert_equal "filter[team_id_eq]=#{team.id}", URI.unescape(req.uri.query)
+          end
+        end
+      end
+
+      context '#reports' do
+        should 'call the api' do
+          team = build(:team)
+          stub_request(:get, /reports.json*/).to_return(body: json_list(:report, 2))
+
+          team.reports
+
+          assert_requested(:get, /reports.json*/) do |req|
+            assert_equal "filter[team_id_eq]=#{team.id}", URI.unescape(req.uri.query)
+          end
+        end
+      end
+
+      context '#create_report' do
+        should 'call Report.create_for_team' do
+          Report.stubs(:create_for_team)
+          team = build(:team)
+
+          team.create_report
+
+          assert_received(Report, :create_for_team) do |expects|
+            expects.with do |id|
+              assert_equal team.id, id
+            end
+          end
+        end
+      end
+
+      context 'live calls' do
+        setup do
+          skip "Make sure you run the live calls locally to ensure proper integration" if ENV['CI_SERVER']
+          WebMock.allow_net_connect!
+          @team = ESP::Team.last
+          skip "Live DB does not have any teams.  Add a team and run tests again." if @team.blank?
+        end
+
+        teardown do
+          WebMock.disable_net_connect!
+        end
+
+        context '#organization' do
+          should 'return an organization' do
+            org = @team.organization
+
+            assert_equal @team.organization_id, org.id
+          end
+        end
+
+        context '#sub_organization' do
+          should 'return a sub_organization' do
+            sub_org = @team.sub_organization
+
+            assert_equal @team.sub_organization_id, sub_org.id
+          end
+        end
+
+        context '#external_accounts' do
+          should 'return an array of external_accounts' do
+            external_accounts = @team.external_accounts
+
+            assert_equal ESP::ExternalAccount, external_accounts.resource_class
+          end
+        end
+
+        context '#reports' do
+          should 'return an array of reports' do
+            reports = @team.reports
+
+            assert_equal ESP::Report, reports.resource_class
+          end
+        end
+
+        context '#CRUD' do
+          should 'be able to create, update and destroy' do
+            team = ESP::Team.new(name: 'bob', organization_id: @team.organization_id, sub_organization_id: @team.sub_organization_id)
+
+            assert_predicate team, :new?
+
+            team.save
+
+            refute_predicate team, :new?
+
+            team.name = 'jim'
+            team.save
+
+            assert_nothing_raised do
+              ESP::Team.find(team.id)
+            end
+
+            team.destroy
+
+            assert_raises ActiveResource::ResourceNotFound do
+              ESP::Team.find(team.id)
+            end
+          end
+        end
+      end
+    end
+  end
+end