esp_sdk 2.2.0 → 2.3.0

data/test/esp/integration/external_account_integration_test.rb

@@ -0,0 +1,68 @@
+require File.expand_path(File.dirname(__FILE__) + '/../../test_helper')
+
+module ESP::Integration
+  class ExternalAccountTest < ESP::Integration::TestCase
+    context ESP::ExternalAccount do
+      context 'live calls' do
+        setup do
+          @external_account = ESP::ExternalAccount.last
+          fail "Live DB does not have any external_accounts.  Add a external_account and run tests again." if @external_account.blank?
+        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/integration/json_api_format_integration_test.rb

@@ -0,0 +1,61 @@
+require File.expand_path(File.dirname(__FILE__) + '/../../test_helper')
+
+module ESP::Integration
+  module Formats
+    class JsonAPIFormatTest < ESP::Integration::TestCase
+      context ActiveResource::Formats::JsonAPIFormat do
+        context 'live calls' do
+          should 'merge included objects' do
+            report = ESP::Report.first
+            alert = ESP::Alert.where(report_id: report.id, include: 'external_account.team.organization,region,signature,custom_signature').first
+
+            assert_not_nil alert.attributes['external_account']
+            assert_equal alert.external_account_id, alert.external_account.id
+            assert_not_nil alert.external_account.attributes['organization']
+            assert_equal alert.external_account.organization_id, alert.external_account.organization.id
+            assert_not_nil alert.external_account.attributes['team']
+            assert_equal alert.external_account.team_id, alert.external_account.team.id
+            assert_not_nil alert.external_account.team.attributes['organization']
+            assert_equal alert.external_account.team.organization_id, alert.external_account.team.organization.id
+            assert_not_nil alert.attributes['region']
+            assert_equal alert.region_id, alert.region.id
+            if alert.signature.present?
+              assert_not_nil alert.attributes['signature']
+              assert_equal alert.signature_id, alert.signature.id
+            else
+              assert_not_nil alert.attributes['custom_signature']
+              assert_equal alert.custom_signature_id, alert.custom_signature.id
+            end
+          end
+
+          should 'assign foreign key for a belongs_to relationship' do
+            user = ESP::User.last
+
+            assert_not_nil user.organization_id
+          end
+
+          should 'assign foreign key for a has_many relationship' do
+            user = ESP::User.last
+
+            assert_not_nil user.sub_organization_ids
+          end
+        end
+      end
+
+      context ActiveResource::ConnectionError do
+        context 'live calls' do
+          context "with ESP::Team" do
+            context '.initialize' do
+              should 'parse the error and return a descriptive error message' do
+                error = assert_raises ActiveResource::BadRequest do
+                  ESP::Team.create
+                end
+                assert_equal 'Failed.  Response code = 400.  Response message = param is missing or the value is empty: attributes.', error.message
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/test/esp/integration/metadata_integration_test.rb

@@ -0,0 +1,18 @@
+require File.expand_path(File.dirname(__FILE__) + '/../../test_helper')
+
+module ESP::Integration
+  class MetadataTest < ESP::Integration::TestCase
+    context ESP::Metadata do
+      context 'live calls' do
+        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/integration/organization_integration_test.rb

@@ -0,0 +1,81 @@
+require File.expand_path(File.dirname(__FILE__) + '/../../test_helper')
+
+module ESP::Integration
+  class OrganizationTest < ESP::Integration::TestCase
+    context ESP::Organization do
+      context 'live calls' do
+        setup do
+          @organization = ESP::Organization.last
+          fail "Live DB does not have any sub organizations.  Add a sub organization and run tests again." if @organization.blank?
+        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/integration/paginated_collection_integration_test.rb

@@ -0,0 +1,67 @@
+require File.expand_path(File.dirname(__FILE__) + '/../../test_helper')
+
+module ESP::Integration
+  class PaginatedCollectionTest < ESP::Integration::TestCase
+    context ActiveResource::PaginatedCollection do
+      context 'live calls' do
+        should 'always return the correct page and update itself when using the ! methods' do
+          report = ESP::Report.all.detect { |r| r.status == 'complete' }
+          alerts = report.alerts
+          last_page_number = alerts.last_page_number
+
+          assert_equal '1', alerts.current_page_number
+          first_id = alerts.first.id
+
+          alerts.next_page!
+          assert_equal '2', alerts.current_page_number
+          refute_equal first_id, alerts.first.id
+          first_id = alerts.first.id
+
+          alerts.last_page!
+          assert_equal last_page_number, alerts.current_page_number
+          refute_equal first_id, alerts.first.id
+          first_id = alerts.first.id
+
+          alerts.previous_page!
+          assert_equal last_page_number.to_i - 1, alerts.current_page_number.to_i
+          refute_equal first_id, alerts.first.id
+          assert_equal 20, alerts.count # make sure the size did not get messed up while on the last page
+          first_id = alerts.first.id
+
+          alerts.page!(3)
+          assert_equal '3', alerts.current_page_number
+          refute_equal first_id, alerts.first.id
+        end
+
+        should 'always return the correct page when not using the ! methods' do
+          report = ESP::Report.all.detect { |r| r.status == 'complete' }
+          alerts = report.alerts
+          last_page_number = alerts.last_page_number
+
+          assert_equal '1', alerts.current_page_number
+          first_id = alerts.first.id
+
+          page = alerts.next_page
+          assert_equal '2', page.current_page_number
+          refute_equal first_id, page.first.id
+          first_id = page.first.id
+
+          page = page.last_page
+          assert_equal last_page_number, page.current_page_number
+          refute_equal first_id, page.first.id
+          first_id = page.first.id
+
+          page = page.previous_page
+          assert_equal last_page_number.to_i - 1, page.current_page_number.to_i
+          refute_equal first_id, page.first.id
+          assert_equal 20, page.count # make sure the size did not get messed up while on the last page
+          first_id = page.first.id
+
+          page = alerts.page(3)
+          assert_equal '3', page.current_page_number
+          refute_equal first_id, page.first.id
+        end
+      end
+    end
+  end
+end

data/test/esp/integration/region_integration_test.rb

@@ -0,0 +1,32 @@
+require File.expand_path(File.dirname(__FILE__) + '/../../test_helper')
+
+module ESP::Integration
+  class RegionTest < ESP::Integration::TestCase
+    context ESP::Region do
+      context 'live calls' do
+        setup do
+          @region = ESP::Region.last
+          fail "Live DB does not have any regions.  Add a region and run tests again." if @region.blank?
+        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/integration/report_export_integration_integration_test.rb

@@ -0,0 +1,24 @@
+require File.expand_path(File.dirname(__FILE__) + '/../../test_helper')
+
+module ESP::Integration
+  class Report
+    module Export
+      class IntegrationTest < ESP::Integration::TestCase
+        context ESP::Report::Export::Integration do
+          context 'live calls' do
+            context '#create' do
+              should 'return regions' do
+                report = ESP::Report.last
+                fail "Live DB does not have any reports.  Add a report with stats and run tests again." if report.blank?
+
+                response = ESP::Report::Export::Integration.create(integration_id: 1, report_ids: [report.id])
+
+                assert_predicate response.errors, :blank?
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/test/esp/integration/report_integration_test.rb

@@ -0,0 +1,71 @@
+require File.expand_path(File.dirname(__FILE__) + '/../../test_helper')
+
+module ESP::Integration
+  class ReportTest < ESP::Integration::TestCase
+    context ESP::Report do
+      context 'live calls' do
+        setup do
+          @report = ESP::Report.all.detect { |r| r.status == 'complete' }
+          fail "Live DB does not have any reports.  Add a report and run tests again." if @report.blank?
+        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/integration/resource_integration_test.rb

@@ -0,0 +1,54 @@
+require File.expand_path(File.dirname(__FILE__) + '/../../test_helper')
+
+module ESP::Integration
+  class ResourceTest < ESP::Integration::TestCase
+    context ESP::Resource do
+      context 'live calls' do
+        setup do
+          @team = ESP::Team.last
+          fail "Live DB does not have any teams.  Add a team and run tests again." if @team.blank?
+        end
+
+        context 'with ESP::Team' do
+          context '.find' do
+            should 'find a team' do
+              t = ESP::Team.find(@team.id)
+
+              assert_equal ESP::Team, t.class
+              assert_equal @team.id, t.id
+            end
+
+            should 'find teams' do
+              t = ESP::Team.find(:all, params: { id: @team.id })
+
+              assert_equal ESP::Team, t.resource_class
+              assert_equal 1, t.count
+            end
+          end
+
+          context '.where' do
+            should 'return teams' do
+              teams = ESP::Team.where(id: @team.id)
+
+              assert_equal ESP::Team, teams.resource_class
+              assert_equal 1, teams.count
+            end
+
+            should 'return included object on first call' do
+              teams = ESP::Team.where(id: @team.id, include: 'organization')
+
+              assert_equal ESP::Organization, teams.first.attributes['organization'].class
+            end
+
+            should 'return multiple included objects on first call' do
+              teams = ESP::Team.where(id: @team.id, include: 'sub_organization, organization')
+
+              assert_equal ESP::Organization, teams.first.attributes['organization'].class
+              assert_equal ESP::SubOrganization, teams.first.attributes['sub_organization'].class
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/test/esp/integration/scan_interval_integration_test.rb

@@ -0,0 +1,49 @@
+require File.expand_path(File.dirname(__FILE__) + '/../../test_helper')
+
+module ESP::Integration
+  class ScanIntervalTest < ESP::Integration::TestCase
+    context ESP::ScanInterval do
+      context 'live calls' do
+        setup do
+          @external_account = ESP::ExternalAccount.last
+          @service = ESP::Service.last
+          fail "Live DB does not have any external_accounts.  Add an external_account and run tests again." if @external_account.blank?
+          fail "Live DB does not have any services.  Add an service and run tests again." if @service.blank?
+        end
+
+        context '#CRUD' do
+          should 'be able to create, update and destroy' do
+            @scan_interval = ESP::ScanInterval.new(interval: 15, service_id: @service.id, external_account_id: @external_account.id)
+
+            # Create
+            assert_predicate @scan_interval, :new?
+            @scan_interval.save
+            refute_predicate @scan_interval, :new?
+
+            # Update
+            @scan_interval.interval = 30
+            @scan_interval.save
+            assert_nothing_raised do
+              ESP::ScanInterval.find(@scan_interval.id.to_i)
+            end
+
+            # Service Relationship
+            service = @scan_interval.service
+            assert_equal service.id, @service.id
+
+            # External Account Relationship
+            external_account = @scan_interval.external_account
+            assert_equal external_account.id, @external_account.id
+
+            # Destroy
+            @scan_interval.destroy
+
+            assert_raises ActiveResource::ResourceNotFound do
+              ESP::ScanInterval.find(@scan_interval.id.to_i)
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/test/esp/integration/service_integration_test.rb

@@ -0,0 +1,28 @@
+require File.expand_path(File.dirname(__FILE__) + '/../../test_helper')
+
+module ESP::Integration
+  class ServiceTest < ESP::Integration::TestCase
+    context ESP::Service do
+      context 'live calls' do
+        setup do
+          @service = ESP::Service.last
+          fail "Live DB does not have any services.  Add a service and run tests again." if @service.blank?
+        end
+
+        context '#signatures' do
+          should 'return an array of signatures' do
+            signatures = @service.signatures
+
+            assert_equal ESP::Signature, signatures.resource_class
+          end
+        end
+
+        context '#CRUD' do
+          should 'be able to read' do
+            assert_not_nil @service, @service.inspect
+          end
+        end
+      end
+    end
+  end
+end

data/test/esp/integration/signature_integration_test.rb

@@ -0,0 +1,62 @@
+require File.expand_path(File.dirname(__FILE__) + '/../../test_helper')
+
+module ESP::Integration
+  class SignatureTest < ESP::Integration::TestCase
+    context ESP::Signature do
+      context 'live calls' do
+        setup do
+          @signature = ESP::Signature.where(name_cont: 'heartbleed').last
+          fail "Live DB does not have any signatures.  Add a signature and run tests again." if @signature.blank?
+        end
+
+        context '#service' do
+          should 'return a service' do
+            service = @signature.service
+
+            assert_equal @signature.service_id, service.id
+            assert_equal ESP::Service, service.class
+          end
+        end
+
+        context '#run' do
+          should 'return alerts' do
+            skip "Can't run sigs on CI" if ENV['CI_SERVER']
+            external_account_id = ESP::ExternalAccount.last.id
+
+            alerts = @signature.run(external_account_id: external_account_id, region: 'us_east_1')
+
+            assert_equal ESP::Alert, alerts.resource_class
+          end
+
+          should 'return errors' do
+            external_account_id = 999_999_999_999
+
+            @signature.run(external_account_id: external_account_id, region: 'us_east_1')
+
+            assert_equal "Couldn't find ExternalAccount", @signature.errors.full_messages.first
+          end
+        end
+
+        context '.where' do
+          should 'return signature objects' do
+            signatures = ESP::Signature.where(id_eq: @signature.id)
+
+            assert_equal ESP::Signature, signatures.resource_class
+          end
+        end
+
+        context '#CRUD' do
+          should 'be able to read' do
+            signature = ESP::Signature.last
+
+            assert_not_nil signature
+
+            signature = ESP::Signature.find(signature.id)
+
+            assert_not_nil signature
+          end
+        end
+      end
+    end
+  end
+end

data/test/esp/integration/stat_custom_signature_integration_test.rb

@@ -0,0 +1,22 @@
+require File.expand_path(File.dirname(__FILE__) + '/../../test_helper')
+
+module ESP::Integration
+  class Stat
+    class CustomSignatureTest < ESP::Integration::TestCase
+      context ESP::StatCustomSignature do
+        context 'live calls' do
+          context '.for_stat' do
+            should 'return tags for stat id' do
+              report = ESP::Report.all.detect { |r| r.status == 'complete' }
+              fail "make sure you have a complete report" unless report.present?
+              stat_id = report.stat.id
+              stats = ESP::StatCustomSignature.for_stat(stat_id)
+
+              assert_equal ESP::StatCustomSignature, stats.resource_class
+            end
+          end
+        end
+      end
+    end
+  end
+end