esp_sdk 1.1.0 → 2.0.0.rc1

data/test/esp/resources/stat_service_test.rb

@@ -0,0 +1,114 @@
+require File.expand_path(File.dirname(__FILE__) + '/../../test_helper')
+
+module ESP
+  class Stat
+    class ServiceTest < ActiveSupport::TestCase
+      context ESP::StatService do
+        context '.for_stat' do
+          should 'throw an error if stat id is not supplied' do
+            error = assert_raises ArgumentError do
+              ESP::StatService.for_stat
+            end
+            assert_equal 'You must supply a stat id.', error.message
+          end
+
+          should 'call the api' do
+            stub_service = stub_request(:get, %r{stats/5/services.json*}).to_return(body: json_list(:stat_service, 2))
+
+            services = ESP::StatService.for_stat(5)
+
+            assert_requested(stub_service)
+            assert_equal ESP::StatService, services.resource_class
+          end
+        end
+        context '.find' do
+          should 'throw an error if stat_id is not supplied' do
+            error = assert_raises ArgumentError do
+              ESP::StatService.find(:all, params: { id: 3 })
+            end
+            assert_equal 'You must supply a stat id.', error.message
+          end
+
+          should 'call the show api and return a service if searching by id' do
+            stub_service = stub_request(:get, %r{services/5.json*}).to_return(body: json(:stat_service))
+
+            service = ESP::StatService.find(5)
+
+            assert_requested(stub_service)
+            assert_equal ESP::StatService, service.class
+          end
+
+          should 'call the api and return services when stat_id is supplied' do
+            stub_service = stub_request(:get, %r{stats/5/services.json*}).to_return(body: json_list(:stat_service, 2))
+
+            services = ESP::StatService.find(:all, params: { stat_id: 5 })
+
+            assert_requested(stub_service)
+            assert_equal ESP::StatService, services.resource_class
+          end
+        end
+
+        context '#create' do
+          should 'not be implemented' do
+            assert_raises ESP::NotImplementedError do
+              ESP::StatService.create(name: 'test')
+            end
+          end
+        end
+
+        context '#update' do
+          should 'not be implemented' do
+            service = ESP::StatService.new
+            assert_raises ESP::NotImplementedError do
+              service.save
+            end
+          end
+        end
+
+        context '#destroy' do
+          should 'not be implemented' do
+            s = ESP::StatService.new
+            assert_raises ESP::NotImplementedError do
+              s.destroy
+            end
+          end
+        end
+
+        context '#service' do
+          should 'call the show api and return the service' do
+            service_stat = ESP::StatService.new(service_id: 3)
+
+            stub_service = stub_request(:get, %r{services/#{service_stat.service_id}.json*}).to_return(body: json(:service))
+
+            service = service_stat.service
+
+            assert_requested(stub_service)
+            assert_equal ESP::Service, service.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_stat' do
+            should 'return tags for stat id' do
+              report = ESP::Report.find(:first, params: { status: 'complete' })
+              skip "make sure you have a complete report" unless report.present?
+              stat_id = report.stat.id
+              stats = ESP::StatService.for_stat(stat_id)
+
+              assert_equal ESP::StatService, stats.resource_class
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/test/esp/resources/stat_signature_test.rb

@@ -0,0 +1,115 @@
+require File.expand_path(File.dirname(__FILE__) + '/../../test_helper')
+
+module ESP
+  class Stat
+    class SignatureTest < ActiveSupport::TestCase
+      context ESP::StatSignature do
+        context '.for_stat' do
+          should 'throw an error if stat id is not supplied' do
+            error = assert_raises ArgumentError do
+              ESP::StatSignature.for_stat
+            end
+            assert_equal 'You must supply a stat id.', error.message
+          end
+
+          should 'call the api' do
+            stub_signature = stub_request(:get, %r{stats/5/signatures.json*}).to_return(body: json_list(:stat_signature, 2))
+
+            signatures = ESP::StatSignature.for_stat(5)
+
+            assert_requested(stub_signature)
+            assert_equal ESP::StatSignature, signatures.resource_class
+          end
+        end
+
+        context '.find' do
+          should 'throw an error if stat_id is not supplied' do
+            error = assert_raises ArgumentError do
+              ESP::StatSignature.find(:all, params: { id: 3 })
+            end
+            assert_equal 'You must supply a stat id.', error.message
+          end
+
+          should 'call the show api and return a signature if searching by id' do
+            stub_signature = stub_request(:get, %r{signatures/5.json*}).to_return(body: json(:stat_signature))
+
+            signature = ESP::StatSignature.find(5)
+
+            assert_requested(stub_signature)
+            assert_equal ESP::StatSignature, signature.class
+          end
+
+          should 'call the api and return signatures when stat_id is supplied' do
+            stub_signature = stub_request(:get, %r{stats/5/signatures.json*}).to_return(body: json_list(:stat_signature, 2))
+
+            signatures = ESP::StatSignature.find(:all, params: { stat_id: 5 })
+
+            assert_requested(stub_signature)
+            assert_equal ESP::StatSignature, signatures.resource_class
+          end
+        end
+
+        context '#create' do
+          should 'not be implemented' do
+            assert_raises ESP::NotImplementedError do
+              ESP::StatSignature.create(name: 'test')
+            end
+          end
+        end
+
+        context '#update' do
+          should 'not be implemented' do
+            signature = ESP::StatSignature.new
+            assert_raises ESP::NotImplementedError do
+              signature.save
+            end
+          end
+        end
+
+        context '#destroy' do
+          should 'not be implemented' do
+            s = ESP::StatSignature.new
+            assert_raises ESP::NotImplementedError do
+              s.destroy
+            end
+          end
+        end
+
+        context '#signature' do
+          should 'call the show api and return the signature' do
+            signature_stat = ESP::StatSignature.new(signature_id: 3)
+
+            stub_signature = stub_request(:get, %r{signatures/#{signature_stat.signature_id}.json*}).to_return(body: json(:signature))
+
+            signature = signature_stat.signature
+
+            assert_requested(stub_signature)
+            assert_equal ESP::Signature, signature.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_stat' do
+            should 'return tags for stat id' do
+              report = ESP::Report.find(:first, params: { status: 'complete' })
+              skip "make sure you have a complete report" unless report.present?
+              stat_id = report.stat.id
+              stats = ESP::StatSignature.for_stat(stat_id)
+
+              assert_equal ESP::StatSignature, stats.resource_class
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/test/esp/resources/stat_test.rb

@@ -0,0 +1,159 @@
+require File.expand_path(File.dirname(__FILE__) + '/../../test_helper')
+
+module ESP
+  class StatTest < ActiveSupport::TestCase
+    context ESP::Stat do
+      context '#report' do
+        should 'call the api' do
+          stat = build(:stat, report_id: 3)
+          stubbed_report = stub_request(:get, %r{reports/#{stat.report_id}.json*}).to_return(body: json(:report))
+
+          stat.report
+
+          assert_requested(stubbed_report)
+        end
+      end
+
+      context '#regions' do
+        should 'call the api for the stat' do
+          stat = build(:stat)
+          stubbed_regions = stub_request(:get, %r{stats/#{stat.id}/regions.json*}).to_return(body: json_list(:stat_region, 2))
+
+          stat.regions
+
+          assert_requested(stubbed_regions)
+        end
+      end
+
+      context '#services' do
+        should 'call the api for the stat' do
+          stat = build(:stat)
+          stubbed_services = stub_request(:get, %r{stats/#{stat.id}/services.json*}).to_return(body: json_list(:stat_service, 2))
+
+          stat.services
+
+          assert_requested(stubbed_services)
+        end
+      end
+
+      context '#signatures' do
+        should 'call the api for the stat' do
+          stat = build(:stat)
+          stubbed_signatures = stub_request(:get, %r{stats/#{stat.id}/signatures.json*}).to_return(body: json_list(:stat_signature, 2))
+
+          stat.signatures
+
+          assert_requested(stubbed_signatures)
+        end
+      end
+
+      context '#custom_signatures' do
+        should 'call the api for the stat' do
+          stat = build(:stat)
+          stubbed_custom_signatures = stub_request(:get, %r{stats/#{stat.id}/custom_signatures.json*}).to_return(body: json_list(:stat_custom_signature, 2))
+
+          stat.custom_signatures
+
+          assert_requested(stubbed_custom_signatures)
+        end
+      end
+
+      context '#find' do
+        should 'not be implemented' do
+          assert_raises ESP::NotImplementedError do
+            ESP::Stat.find(1)
+          end
+        end
+      end
+
+      context '#create' do
+        should 'not be implemented' do
+          assert_raises ESP::NotImplementedError do
+            ESP::Stat.create(report_id: 1)
+          end
+        end
+      end
+
+      context '#update' do
+        should 'not be implemented' do
+          s = build(:stat)
+
+          assert_raises ESP::NotImplementedError do
+            s.save
+          end
+        end
+      end
+
+      context '#destroy' do
+        should 'not be implemented' do
+          s = build(:stat)
+
+          assert_raises ESP::NotImplementedError do
+            s.destroy
+          end
+        end
+      end
+
+      context '.for_report' do
+        should 'raise an error if report_id is not provided' do
+          error = assert_raises ArgumentError do
+            ESP::Stat.for_report
+          end
+          assert_equal "You must supply a report id.", error.message
+        end
+
+        should 'call the api and return a stat' do
+          stub_stat = stub_request(:get, %r{reports/5/stats.json*}).to_return(body: json(:stat))
+
+          stat = ESP::Stat.for_report(5)
+
+          assert_requested(stub_stat)
+          assert_equal ESP::Stat, stat.class
+        end
+      end
+
+      context '.latest_for_teams' do
+        should 'call the api and return a collection of stats' do
+          stub_stat = stub_request(:get, %r{stats/latest_for_teams.json*}).to_return(body: json_list(:stat, 2))
+
+          stats = ESP::Stat.latest_for_teams
+
+          assert_requested(stub_stat)
+          assert_equal ESP::Stat, stats.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!
+          @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 'for_report' do
+          should 'return the stat for the report' do
+            report = ESP::Report.last
+            skip "Live DB does not have any reports.  Add a report with stats and run tests again." if report.blank?
+
+            stat = ESP::Stat.for_report(report.id)
+
+            assert_equal report.id, stat.report.id
+          end
+        end
+
+        context '.latest_for_teams' do
+          should 'return a collection of stats' do
+            stats = ESP::Stat.latest_for_teams
+
+            assert_equal ESP::Stat, stats.resource_class
+          end
+        end
+      end
+    end
+  end
+end

data/test/esp/resources/sub_organization_test.rb

@@ -0,0 +1,127 @@
+require File.expand_path(File.dirname(__FILE__) + '/../../test_helper')
+
+module ESP
+  class SubOrganizationTest < ActiveSupport::TestCase
+    context ESP::SubOrganization do
+      context '#organization' do
+        should 'call the api' do
+          sub_organization = build(:sub_organization, organization_id: 4)
+          stub_org = stub_request(:get, %r{organizations/#{sub_organization.organization_id}.json*}).to_return(body: json(:organization))
+
+          sub_organization.organization
+
+          assert_requested(stub_org)
+        end
+      end
+
+      context '#teams' do
+        should 'call the api' do
+          sub_organization = build(:sub_organization, team_id: 4)
+          stub_request(:get, /teams.json*/).to_return(body: json_list(:sub_organization, 2))
+
+          sub_organization.teams
+
+          assert_requested(:get, /teams.json*/) do |req|
+            assert_equal "filter[sub_organization_id_eq]=#{sub_organization.id}", URI.unescape(req.uri.query)
+          end
+        end
+      end
+
+      context '#external_accounts' do
+        should 'call the api' do
+          sub_organization = build(:sub_organization)
+          stub_request(:get, /external_accounts.json*/).to_return(body: json_list(:external_account, 2))
+
+          sub_organization.external_accounts
+
+          assert_requested(:get, /external_accounts.json*/) do |req|
+            assert_equal "filter[sub_organization_id_eq]=#{sub_organization.id}", URI.unescape(req.uri.query)
+          end
+        end
+      end
+
+      context '#reports' do
+        should 'call the api' do
+          sub_organization = build(:sub_organization)
+          stub_request(:get, /reports.json*/).to_return(body: json_list(:report, 2))
+
+          sub_organization.reports
+
+          assert_requested(:get, /reports.json*/) do |req|
+            assert_equal "filter[sub_organization_id_eq]=#{sub_organization.id}", URI.unescape(req.uri.query)
+          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!
+          @sub_organization = ESP::SubOrganization.last
+          skip "Live DB does not have any sub organizations.  Add a sub organization and run tests again." if @sub_organization.blank?
+        end
+
+        teardown do
+          WebMock.disable_net_connect!
+        end
+
+        context '#organization' do
+          should 'return an organization' do
+            org = @sub_organization.organization
+
+            assert_equal @sub_organization.organization_id, org.id
+          end
+        end
+
+        context '#teams' do
+          should 'return a sub_organization' do
+            teams = @sub_organization.teams
+
+            assert_equal ESP::Team, teams.resource_class
+          end
+        end
+
+        context '#external_accounts' do
+          should 'return an array of external_accounts' do
+            external_accounts = @sub_organization.external_accounts
+
+            assert_equal ESP::ExternalAccount, external_accounts.resource_class
+          end
+        end
+
+        context '#reports' do
+          should 'return an array of reports' do
+            reports = @sub_organization.reports
+
+            assert_equal ESP::Report, reports.resource_class
+          end
+        end
+
+        context '#CRUD' do
+          should 'be able to create, update and destroy' do
+            sub_organization = ESP::SubOrganization.new(name: 'bob', organization_id: @sub_organization.organization_id)
+
+            assert_predicate sub_organization, :new?
+
+            sub_organization.save
+
+            refute_predicate sub_organization, :new?
+
+            sub_organization.name = 'jim'
+            sub_organization.save
+
+            assert_nothing_raised do
+              ESP::SubOrganization.find(sub_organization.id)
+            end
+
+            sub_organization.destroy
+
+            assert_raises ActiveResource::ResourceNotFound do
+              ESP::SubOrganization.find(sub_organization.id)
+            end
+          end
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,115 @@
+require File.expand_path(File.dirname(__FILE__) + '/../../../test_helper')
+
+module ESP
+  class Suppression
+    class RegionTest < ActiveSupport::TestCase
+      context ESP::Suppression::Region do
+        context '#find' do
+          should 'not be implemented' do
+            assert_raises ESP::NotImplementedError do
+              ESP::Suppression::Region.find(4)
+            end
+          end
+        end
+
+        context '#update' do
+          should 'not be implemented' do
+            s = ESP::Suppression::Region.new
+            assert_raises ESP::NotImplementedError do
+              s.update
+            end
+          end
+        end
+
+        context '#destroy' do
+          should 'not be implemented' do
+            s = ESP::Suppression::Region.new
+            assert_raises ESP::NotImplementedError do
+              s.destroy
+            end
+          end
+        end
+
+        context '#create' do
+          should 'call the api' do
+            stub_request(:post, %r{suppressions/regions.json*}).to_return(body: json(:suppression_region))
+
+            suppression = ESP::Suppression::Region.create(regions: ['us_east_1'], external_account_ids: [5], reason: 'because')
+
+            assert_requested(:post, %r{suppressions/regions.json*}) do |req|
+              body = JSON.parse(req.body)
+              assert_equal 'because', body['data']['attributes']['reason']
+              assert_equal ['us_east_1'], body['data']['attributes']['regions']
+              assert_equal [5], body['data']['attributes']['external_account_ids']
+            end
+            assert_equal ESP::Suppression::Region, suppression.class
+          end
+
+          context 'for alert' do
+            should 'call the api' do
+              stub_request(:post, %r{suppressions/alert/5/regions.json*}).to_return(body: json(:suppression_region))
+
+              suppression = ESP::Suppression::Region.create(alert_id: 5, reason: 'because')
+
+              assert_requested(:post, %r{suppressions/alert/5/regions.json*}) do |req|
+                body = JSON.parse(req.body)
+                assert_equal 'because', body['data']['attributes']['reason']
+              end
+              assert_equal ESP::Suppression::Region, 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::Region.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::Region.create(signature_ids: [signature_id], custom_signature_ids: [], reason: 'test', regions: ['us_east_1'], external_account_ids: [external_account_id])
+
+              assert_equal ESP::Suppression::Region, 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::Region.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::Region.create(alert_id: alert_id, reason: 'test')
+
+                assert_predicate suppression.errors, :blank?
+                assert_equal ESP::Suppression::Region, suppression.class
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+end