esp_sdk 2.2.0 → 2.3.0

data/test/esp/resources/alert_test.rb

@@ -225,99 +225,6 @@ module ESP
           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!
-          @report = ESP::Report.all.detect { |r| r.status == 'complete' }
-          skip "Live DB does not have any reports.  Add a report and run tests again." if @report.blank?
-          @alert = @report.alerts.last
-        end
-
-        teardown do
-          WebMock.disable_net_connect!
-        end
-
-        context '#external_account' do
-          should 'return an external_account' do
-            external_account = @alert.external_account
-
-            assert_equal ESP::ExternalAccount, external_account.class
-            assert_equal @alert.external_account_id, external_account.id
-          end
-        end
-
-        context '#region' do
-          should 'return a region' do
-            region = @alert.region
-
-            assert_equal ESP::Region, region.class
-            assert_equal @alert.region_id, region.id
-          end
-        end
-
-        context '#signature' do
-          should 'return a signature' do
-            signature = @alert.signature
-
-            assert_equal ESP::Signature, signature.class
-            assert_equal @alert.signature_id, signature.id
-          end
-        end
-
-        context '#custom_signature' do
-          should 'return a custom_signature' do
-            assert_nothing_raised do
-              @alert.attributes['custom_signature_id'] ||= 1
-              @alert.custom_signature
-            end
-          end
-        end
-
-        context '#suppression' do
-          should 'return a suppression' do
-            assert_nothing_raised do
-              @alert.attributes['suppression_id'] ||= 1
-              @alert.suppression
-            end
-          end
-        end
-
-        context '#cloud_trail_events' do
-          should 'return cloud_trail_events' do
-            assert_nothing_raised do
-              @alert.cloud_trail_events
-            end
-          end
-        end
-
-        context '#tags' do
-          should 'return tags' do
-            assert_nothing_raised do
-              @alert.tags
-            end
-          end
-        end
-
-        context '.find' do
-          should 'return an alert by id' do
-            alert = ESP::Alert.find(@alert.id.to_i)
-
-            assert_equal ESP::Alert, alert.class
-            assert_equal @alert.id, alert.id
-          end
-        end
-
-        context '.where' do
-          should 'return alert objects' do
-            alerts = ESP::Alert.where(report_id: @report.id, id_eq: @alert.id)
-
-            assert_equal ESP::Alert, alerts.resource_class
-            assert_equal @alert.id, alerts.first.id
-          end
-        end
-      end
     end
   end
 end

data/test/esp/resources/cloud_trail_event_test.rb

@@ -81,26 +81,6 @@ module ESP
           assert_equal ESP::CloudTrailEvent, events.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 events for alert id' do
-            report = ESP::Report.all.detect { |r| r.status == 'complete' }
-            events = ESP::CloudTrailEvent.for_alert(report.alerts.last.id)
-
-            assert_equal ESP::CloudTrailEvent, events.resource_class
-          end
-        end
-      end
     end
   end
 end

data/test/esp/resources/contact_request_test.rb

@@ -53,29 +53,6 @@ module ESP
           assert_equal ESP::ContactRequest, contact_request.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 '#CRUD' do
-          should 'be able to create, update and destroy' do
-            contact_request = ESP::ContactRequest.new(user_id: 5, request_type: 'feature', title: 'My great feature idea', description: 'This is my idea for a really useful feature...')
-
-            assert_predicate contact_request, :new?
-
-            contact_request.save
-
-            refute_predicate contact_request, :new?
-          end
-        end
-      end
     end
   end
 end

data/test/esp/resources/custom_signature_test.rb

@@ -14,6 +14,20 @@ module ESP
         end
       end
 
+      context '#teams' do
+        should 'call the api' do
+          custom_signature = build(:custom_signature, team_id: 1)
+          stub_request(:put, /teams.json*/).to_return(body: json_list(:team, 2))
+
+          custom_signature.teams
+
+          assert_requested(:put, /teams.json*/) do |req|
+            body = JSON.parse(req.body)
+            assert_equal custom_signature.id, body['filter']['custom_signatures_id_eq']
+          end
+        end
+      end
+
       context '.run_sanity_test!' do
         should 'call the api and pass params' do
           custom_signature = build(:custom_signature, external_account_id: 3)
@@ -159,91 +173,6 @@ module ESP
           assert_equal ESP::Suppression::Signature, 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!
-          @custom_signature = ESP::CustomSignature.last
-          skip "Live DB does not have any custom_signatures.  Add a custom_signature and run tests again." if @custom_signature.blank?
-        end
-
-        teardown do
-          WebMock.disable_net_connect!
-        end
-
-        context '#organization' do
-          should 'return an organization' do
-            organization = @custom_signature.organization
-
-            assert_equal @custom_signature.organization_id, organization.id
-            assert_equal ESP::Organization, organization.class
-          end
-        end
-
-        context '.run' do
-          should 'return alerts' do
-            external_account_id = ESP::ExternalAccount.last.id
-            alerts = ESP::CustomSignature.run(external_account_id: external_account_id, regions: 'us_east_1', language: @custom_signature.language, signature: @custom_signature.signature)
-
-            assert_equal ESP::Alert, alerts.resource_class
-          end
-
-          should 'return errors' do
-            signature = ESP::CustomSignature.run(external_account_id: 999_999_999_999, regions: 'us_east_1', language: @custom_signature.language, signature: @custom_signature.signature)
-
-            assert_equal "Couldn't find ExternalAccount", signature.errors.full_messages.first
-          end
-        end
-
-        context '#run' do
-          should 'return alerts' do
-            external_account_id = ESP::ExternalAccount.last.id
-            alerts = @custom_signature.run(external_account_id: external_account_id, regions: ['us_east_1'])
-
-            assert_equal ESP::Alert, alerts.resource_class
-          end
-
-          should 'return errors' do
-            @custom_signature.run(external_account_id: 999_999_999_999)
-
-            assert_equal "Couldn't find ExternalAccount", @custom_signature.errors.full_messages.first
-          end
-        end
-
-        context '.where' do
-          should 'return custom_signature objects' do
-            custom_signatures = ESP::CustomSignature.where(id_eq: @custom_signature.id)
-
-            assert_equal ESP::CustomSignature, custom_signatures.resource_class
-          end
-        end
-
-        context '#CRUD' do
-          should 'be able to create, update and destroy' do
-            custom_signature = ESP::CustomSignature.new(@custom_signature.attributes)
-
-            assert_predicate custom_signature, :new?
-
-            custom_signature.save
-
-            refute_predicate custom_signature, :new?
-
-            custom_signature.identifier = 'new identifier'
-            custom_signature.save
-
-            assert_nothing_raised do
-              ESP::CustomSignature.find(custom_signature.id)
-            end
-
-            custom_signature.destroy
-
-            assert_raises ActiveResource::ResourceNotFound do
-              ESP::CustomSignature.find(custom_signature.id)
-            end
-          end
-        end
-      end
     end
   end
 end

data/test/esp/resources/dashboard_test.rb

@@ -55,25 +55,6 @@ module ESP
           assert_equal ESP::Dashboard, dashboard.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 '.recent' do
-          should 'return an array of contact_requests' do
-            dashboards = ESP::Dashboard.recent
-
-            assert_equal ESP::Dashboard, dashboards.resource_class
-          end
-        end
-      end
     end
   end
 end

data/test/esp/resources/external_account_test.rb

@@ -72,73 +72,6 @@ module ESP
           assert_received(ESP::ExternalAccount.any_instance, :generate_external_id)
         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!
-          @external_account = ESP::ExternalAccount.last
-          skip "Live DB does not have any external_accounts.  Add a external_account and run tests again." if @external_account.blank?
-        end
-
-        teardown do
-          WebMock.disable_net_connect!
-        end
-
-        context '#organization' do
-          should 'return an organization' do
-            org = @external_account.organization
-
-            assert_equal @external_account.organization_id, org.id
-          end
-        end
-
-        context '#sub_organization' do
-          should 'return a sub_organization' do
-            sub_org = @external_account.sub_organization
-
-            assert_equal @external_account.sub_organization_id, sub_org.id
-          end
-        end
-
-        context '#team' do
-          should 'return a team' do
-            team = @external_account.team
-
-            assert_equal ESP::Team, team.class
-          end
-        end
-
-        context '.where' do
-          should 'return external_account objects' do
-            external_accounts = ESP::ExternalAccount.where(id_eq: @external_account.id)
-
-            assert_equal ESP::ExternalAccount, external_accounts.resource_class
-          end
-        end
-
-        context '#CRUD' do
-          should 'be able to create, update and destroy' do
-            skip "There are to many dependencies to validate an external account to create or update one. Besides esp_web, esp_query has to be running and there must be valid AWS keys assigned as well."
-
-            external_account = ESP::ExternalAccount.create(name: 'bob', arn: @external_account.arn, sub_organization_id: @external_account.sub_organization_id, team_id: @external_account.team_id)
-
-            assert_predicate external_account, :new?
-            assert_contains external_account.errors, "The account for this ARN is already being checked by Dev"
-
-            refute_predicate @external_account, :new?
-            @external_account.name = @external_account.name
-
-            assert_predicate @external_account, :save
-
-            external_account = build(:external_account, id: 999)
-
-            assert_raises ActiveResource::ResourceNotFound do
-              external_account.destroy
-            end
-          end
-        end
-      end
     end
   end
 end

data/test/esp/resources/metadata_test.rb

@@ -73,26 +73,6 @@ module ESP
           assert_equal ESP::Metadata, metadata.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 metadata for alert id' do
-            report = ESP::Report.all.detect { |r| r.status == 'complete' }
-            metadata = ESP::Metadata.for_alert(report.alerts.last.id)
-
-            assert_equal ESP::Metadata, metadata.class
-          end
-        end
-      end
     end
   end
 end

data/test/esp/resources/organization_test.rb

@@ -98,86 +98,6 @@ module ESP
           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!
-          @organization = ESP::Organization.last
-          skip "Live DB does not have any sub organizations.  Add a sub organization and run tests again." if @organization.blank?
-        end
-
-        teardown do
-          WebMock.disable_net_connect!
-        end
-
-        context '#teams' do
-          should 'return teams' do
-            teams = @organization.teams
-
-            assert_equal ESP::Team, teams.resource_class
-          end
-        end
-
-        context '#sub_organizations' do
-          should 'return a sub_organization' do
-            sub_organizations = @organization.sub_organizations
-
-            assert_equal ESP::SubOrganization, sub_organizations.resource_class
-          end
-        end
-
-        context '#users' do
-          should 'return an array of users' do
-            users = @organization.users
-
-            assert_equal ESP::User, users.resource_class
-          end
-        end
-
-        context '#reports' do
-          should 'return an array of reports' do
-            reports = @organization.reports
-
-            assert_equal ESP::Report, reports.resource_class
-          end
-        end
-
-        context '#external_accounts' do
-          should 'return an array of external_accounts' do
-            external_accounts = @organization.external_accounts
-
-            assert_equal ESP::ExternalAccount, external_accounts.resource_class
-          end
-        end
-
-        context '#custom_signatures' do
-          should 'return an array of custom_signatures' do
-            custom_signatures = @organization.custom_signatures
-
-            assert_equal ESP::CustomSignature, custom_signatures.resource_class
-          end
-        end
-
-        context '.where' do
-          should 'return organization objects' do
-            organizations = ESP::Organization.where(id_eq: @organization.id)
-
-            assert_equal ESP::Organization, organizations.resource_class
-          end
-        end
-
-        context '#CRUD' do
-          should 'be able to update' do
-            @organization.name = @organization.name
-            @organization.save
-
-            assert_nothing_raised do
-              ESP::Organization.find(@organization.id)
-            end
-          end
-        end
-      end
     end
   end
 end

data/test/esp/resources/region_test.rb

@@ -56,37 +56,6 @@ module ESP
           assert_equal ESP::Suppression::Region, 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!
-          @region = ESP::Region.last
-          skip "Live DB does not have any regions.  Add a region and run tests again." if @region.blank?
-        end
-
-        teardown do
-          WebMock.disable_net_connect!
-        end
-
-        context '.where' do
-          should 'return region objects' do
-            regions = ESP::Region.where(id_eq: @region.id)
-
-            assert_equal ESP::Region, regions.resource_class
-          end
-        end
-
-        context '#CRUD' do
-          should 'be able to read' do
-            assert_not_nil @region
-
-            region = ESP::Region.find(@region.id)
-
-            assert_not_nil region
-          end
-        end
-      end
     end
   end
 end

data/test/esp/resources/report_test.rb

@@ -114,76 +114,6 @@ module ESP
           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!
-          @report = ESP::Report.all.detect { |r| r.status == 'complete' }
-          skip "Live DB does not have any reports.  Add a report and run tests again." if @report.blank?
-        end
-
-        teardown do
-          WebMock.disable_net_connect!
-        end
-
-        context '#organization' do
-          should 'return an organization' do
-            org = @report.organization
-
-            assert_equal @report.organization_id, org.id
-          end
-        end
-
-        context '#sub_organization' do
-          should 'return a sub_organization' do
-            sub_org = @report.sub_organization
-
-            assert_equal @report.sub_organization_id, sub_org.id
-          end
-        end
-
-        context '#team' do
-          should 'return a team' do
-            team = @report.team
-
-            assert_equal ESP::Team, team.class
-          end
-        end
-
-        context '#alerts' do
-          should 'return an array of alerts' do
-            alerts = @report.alerts
-
-            assert_equal ESP::Alert, alerts.resource_class
-          end
-        end
-
-        context '#stat' do
-          should 'return a stat' do
-            stat = @report.stat
-
-            assert_equal ESP::Stat, stat.class
-          end
-        end
-
-        context '.where' do
-          should 'return report objects' do
-            reports = ESP::Report.where(id_eq: @report.id)
-
-            assert_equal ESP::Report, reports.resource_class
-          end
-        end
-
-        context '.create' do
-          should 'return an error if a bad team_id is passed' do
-            assert_nothing_raised do
-              r = ESP::Report.create(team_id: 999)
-              assert_equal "Couldn't find Team", r.errors.full_messages.first
-            end
-          end
-        end
-      end
     end
   end
 end

data/test/esp/resources/reports/export/integration_test.rb

@@ -0,0 +1,55 @@
+require File.expand_path(File.dirname(__FILE__) + '/../../../../test_helper')
+
+module ESP
+  class Report
+    module Export
+      class IntegrationTest < ActiveSupport::TestCase
+        context ESP::Report::Export::Integration do
+          context '.where' do
+            should 'not be implemented' do
+              assert_raises ESP::NotImplementedError do
+                ESP::Report::Export::Integration.where(id_eq: 2)
+              end
+            end
+          end
+
+          context '#find' do
+            should 'not be implemented' do
+              assert_raises ESP::NotImplementedError do
+                ESP::Report::Export::Integration.find(4)
+              end
+            end
+          end
+
+          context '#update' do
+            should 'not be implemented' do
+              s = ESP::Report::Export::Integration.new
+              assert_raises ESP::NotImplementedError do
+                s.update
+              end
+            end
+          end
+
+          context '#destroy' do
+            should 'not be implemented' do
+              s = ESP::Report::Export::Integration.new
+              assert_raises ESP::NotImplementedError do
+                s.destroy
+              end
+            end
+          end
+
+          context '#create' do
+            should 'call the api' do
+              stub_request(:post, %r{reports/export/integrations.json*}).to_return(body: { success: 'Your export has been started' }.to_json)
+
+              ESP::Report::Export::Integration.create(integration_id: 1, region_ids: [1])
+
+              assert_requested(:post, %r{reports/export/integrations.json*})
+            end
+          end
+        end
+      end
+    end
+  end
+end