esp_sdk 2.4.0 → 2.5.0

data/test/esp/resources/custom_signature/result/alert_test.rb

@@ -0,0 +1,84 @@
+require File.expand_path(File.dirname(__FILE__) + '/../../../../test_helper')
+
+module ESP
+  class CustomSignature
+    class Result
+      class AlertTest < ActiveSupport::TestCase
+        context ESP::CustomSignature::Result::Alert do
+          context '#region' do
+            should 'call the api' do
+              alert = build(:result_alert, region_id: 4)
+              stubbed_region = stub_request(:get, %r{regions/#{alert.region_id}.json*}).to_return(body: json(:region))
+
+              alert.region
+
+              assert_requested(stubbed_region)
+            end
+          end
+
+          context '#external_account' do
+            should 'call the api' do
+              alert = build(:result_alert, external_account_id: 4)
+              stubbed_external_account = stub_request(:get, %r{external_accounts/#{alert.external_account_id}.json*}).to_return(body: json(:external_account))
+
+              alert.external_account
+
+              assert_requested(stubbed_external_account)
+            end
+          end
+
+          context '#custom_signature' do
+            should 'call the api' do
+              alert = build(:result_alert, custom_signature_id: 4)
+              stubbed_custom_signature = stub_request(:get, %r{custom_signatures/#{alert.custom_signature_id}.json*}).to_return(body: json(:custom_signature))
+
+              alert.custom_signature
+
+              assert_requested(stubbed_custom_signature)
+            end
+          end
+
+          context '.find' do
+            should 'raise ESP::NotImplementedError' do
+              assert_raises ESP::NotImplementedError do
+                ESP::CustomSignature::Result::Alert.find(1)
+              end
+            end
+          end
+
+          context '.where' do
+            should 'raise ESP::NotImplementedError' do
+              assert_raises ESP::NotImplementedError do
+                ESP::CustomSignature::Result::Alert.where(id_eq: 1)
+              end
+            end
+          end
+
+          context '#create' do
+            should 'raise ESP::NotImplementedError' do
+              assert_raises ESP::NotImplementedError do
+                ESP::CustomSignature::Result::Alert.new.create
+              end
+            end
+          end
+
+          context '#update' do
+            should 'raise ESP::NotImplementedError' do
+              assert_raises ESP::NotImplementedError do
+                ESP::CustomSignature::Result::Alert.new.update
+              end
+            end
+          end
+
+          context '#destroy' do
+            should 'raise ESP::NotImplementedError' do
+              assert_raises ESP::NotImplementedError do
+                ESP::CustomSignature::Result::Alert.new.destroy
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/test/esp/resources/custom_signature/result_test.rb

@@ -0,0 +1,91 @@
+require File.expand_path(File.dirname(__FILE__) + '/../../../test_helper')
+
+module ESP
+  class CustomSignature
+    class ResultTest < ActiveSupport::TestCase
+      context ESP::CustomSignature::Result do
+        context '#definition' do
+          should 'call the api' do
+            result = build(:result, definition_id: 4)
+            stubbed_definition = stub_request(:get, %r{custom_signature_definitions/#{result.definition_id}.json*}).to_return(body: json(:definition))
+
+            result.definition
+
+            assert_requested(stubbed_definition)
+          end
+        end
+
+        context '#region' do
+          should 'call the api' do
+            result = build(:result, region_id: 4)
+            stubbed_region = stub_request(:get, %r{regions/#{result.region_id}.json*}).to_return(body: json(:region))
+
+            result.region
+
+            assert_requested(stubbed_region)
+          end
+        end
+
+        context '#external_account' do
+          should 'call the api' do
+            result = build(:result, external_account_id: 4)
+            stubbed_external_account = stub_request(:get, %r{external_accounts/#{result.external_account_id}.json*}).to_return(body: json(:external_account))
+
+            result.external_account
+
+            assert_requested(stubbed_external_account)
+          end
+        end
+
+        context '#alerts' do
+          should 'call the api' do
+            result = build(:result)
+            stub_request(:get, %r{custom_signature_results/#{result.id}/alerts.json}).to_return(body: json_list(:result_alert, 2))
+
+            result.alerts
+
+            assert_requested(:get, %r{custom_signature_results/#{result.id}/alerts.json})
+          end
+        end
+
+        context 'activate' do
+          should 'call the api' do
+            definition = build(:definition)
+            stubbed_defintion = stub_request(:patch, %r{custom_signature_definitions/#{definition.id}/activate.json}).to_return(body: json(:definition))
+
+            definition.activate
+
+            assert_requested stubbed_defintion
+          end
+        end
+
+        context 'archive' do
+          should 'call the api' do
+            definition = build(:definition)
+            stubbed_defintion = stub_request(:patch, %r{custom_signature_definitions/#{definition.id}/archive.json}).to_return(body: json(:definition))
+
+            definition.archive
+
+            assert_requested stubbed_defintion
+          end
+        end
+
+        context '#update' do
+          should 'raise ESP::NotImplementedError' do
+            assert_raises ESP::NotImplementedError do
+              ESP::CustomSignature::Result::Alert.new.update
+            end
+          end
+        end
+
+        context '#destroy' do
+          should 'raise ESP::NotImplementedError' do
+            assert_raises ESP::NotImplementedError do
+              ESP::CustomSignature::Result::Alert.new.destroy
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/test/esp/resources/custom_signature_test.rb

@@ -28,130 +28,15 @@ module ESP
         end
       end
 
-      context '.run_sanity_test!' do
-        should 'call the api and pass params' do
-          custom_signature = build(:custom_signature, external_account_id: 3)
-          stub_request(:post, %r{custom_signatures/run.json*}).to_return(body: json_list(:alert, 2))
-
-          alerts = ESP::CustomSignature.run!(external_account_id: 3, regions: 'param2', language: custom_signature.language, signature: custom_signature.signature)
-
-          assert_requested(:post, %r{custom_signatures/run.json*}) do |req|
-            body = JSON.parse req.body
-            assert_equal false, body['data'].key?('id')
-            assert_equal ['param2'], body['data']['attributes']['regions']
-            assert_equal custom_signature.language, body['data']['attributes']['language']
-            assert_equal custom_signature.signature, body['data']['attributes']['signature']
-          end
-          assert_equal ESP::Alert, alerts.resource_class
-        end
-
-        should 'throw an error if an error is returned' do
-          custom_signature = build(:custom_signature, external_account_id: 3)
-          error = ActiveResource::BadRequest.new('')
-          error_response = json(:error)
-          response = mock(body: error_response, code: '400')
-          error.stubs(:response).returns(response)
-          ActiveResource::Connection.any_instance.expects(:post).raises(error)
-          stub_request(:post, %r{custom_signatures/run.json*}).to_return(body: json_list(:alert, 2))
-
-          error = assert_raises ActiveResource::ResourceInvalid do
-            ESP::CustomSignature.run!(external_account_id: 3, regions: 'param2', language: custom_signature.language, signature: custom_signature.signature)
-          end
-          assert_equal "Failed.  Response code = 400.  Response message = #{JSON.parse(error_response)['errors'].first['title']}.", error.message
-        end
-      end
-
-      context '.run_sanity_test' do
-        should 'call the api and pass params' do
-          custom_signature = build(:custom_signature, external_account_id: 3)
-          stub_request(:post, %r{custom_signatures/run.json*}).to_return(body: json_list(:alert, 2))
-
-          alerts = ESP::CustomSignature.run(external_account_id: 3, regions: 'param2', language: custom_signature.language, signature: custom_signature.signature)
-
-          assert_requested(:post, %r{custom_signatures/run.json*}) do |req|
-            body = JSON.parse req.body
-            assert_equal false, body['data'].key?('id')
-            assert_equal ['param2'], body['data']['attributes']['regions']
-            assert_equal custom_signature.language, body['data']['attributes']['language']
-            assert_equal custom_signature.signature, body['data']['attributes']['signature']
-          end
-          assert_equal ESP::Alert, alerts.resource_class
-        end
-
-        should 'not throw an error if an error is returned' do
-          custom_signature = build(:custom_signature, external_account_id: 3)
-          error = ActiveResource::BadRequest.new('')
-          error_response = json(:error)
-          response = mock(body: error_response, code: '400')
-          error.stubs(:response).returns(response)
-          ActiveResource::Connection.any_instance.expects(:post).raises(error)
-          stub_request(:post, %r{custom_signatures/run.json*}).to_return(body: json_list(:alert, 2))
-
-          assert_nothing_raised do
-            result = ESP::CustomSignature.run(external_account_id: 3, regions: 'param2', language: custom_signature.language, signature: custom_signature.signature)
-            assert_equal JSON.parse(error_response)['errors'].first['title'], result.errors.full_messages.first
-          end
-        end
-      end
-
-      context '#run!' do
-        should 'call the api and pass params' do
-          custom_signature = build(:custom_signature, external_account_id: 3)
-          stub_request(:post, %r{custom_signatures/#{custom_signature.id}/run.json*}).to_return(body: json_list(:alert, 2))
-
-          alerts = custom_signature.run!(regions: 'param2')
-
-          assert_requested(:post, %r{custom_signatures/#{custom_signature.id}/run.json*}) do |req|
-            body = JSON.parse req.body
-            assert_equal custom_signature.id, body['data']['id']
-            assert_equal ['param2'], body['data']['attributes']['regions']
-          end
-          assert_equal ESP::Alert, alerts.resource_class
-        end
-
-        should 'throw an error if an error is returned' do
-          custom_signature = build(:custom_signature, external_account_id: 3)
-          error = ActiveResource::BadRequest.new('')
-          error_response = json(:error)
-          response = mock(body: error_response, code: '400')
-          error.stubs(:response).returns(response)
-          ActiveResource::Connection.any_instance.expects(:post).raises(error)
-          stub_request(:post, %r{custom_signatures/#{custom_signature.id}/run.json*}).to_return(body: json_list(:alert, 2))
-
-          error = assert_raises ActiveResource::ResourceInvalid do
-            custom_signature.run!(regions: 'param2')
-          end
-          assert_equal "Failed.  Response code = 400.  Response message = #{JSON.parse(error_response)['errors'].first['title']}.", error.message
-        end
-      end
-
-      context '#run' do
-        should 'call the api and pass params' do
-          custom_signature = build(:custom_signature, external_account_id: 3)
-          stub_request(:post, %r{custom_signatures/#{custom_signature.id}/run.json*}).to_return(body: json_list(:alert, 2))
-
-          alerts = custom_signature.run(regions: 'param2')
-
-          assert_requested(:post, %r{custom_signatures/#{custom_signature.id}/run.json*}) do |req|
-            body = JSON.parse req.body
-            assert_equal custom_signature.id, body['data']['id']
-            assert_equal ['param2'], body['data']['attributes']['regions']
-          end
-          assert_equal ESP::Alert, alerts.resource_class
-        end
+      context '#definitions' do
+        should 'call the api' do
+          custom_signature = build(:custom_signature, team_id: 1)
+          stub_request(:get, /custom_signature_definitions.json*/).to_return(body: json_list(:definition, 2))
 
-        should 'not throw an error if an error is returned' do
-          custom_signature = build(:custom_signature, external_account_id: 3)
-          error = ActiveResource::BadRequest.new('')
-          error_response = json(:error)
-          response = mock(body: error_response, code: '400')
-          error.stubs(:response).returns(response)
-          ActiveResource::Connection.any_instance.expects(:post).raises(error)
-          stub_request(:post, %r{custom_signatures/#{custom_signature.id}/run.json*}).to_return(body: json_list(:alert, 2))
+          custom_signature.definitions
 
-          assert_nothing_raised do
-            custom_signature.run(regions: 'param2')
-            assert_equal JSON.parse(error_response)['errors'].first['title'], custom_signature.errors.full_messages.first
+          assert_requested(:get, /custom_signature_definitions.json*/) do |req|
+            assert_equal "filter[custom_signature_id_eq]=#{custom_signature.id}", URI.unescape(req.uri.query)
           end
         end
       end

data/test/esp/resources/service_test.rb

@@ -24,6 +24,14 @@ module ESP
         end
       end
 
+      context '.where' do
+        should 'not be implemented' do
+          assert_raises ESP::NotImplementedError do
+            ESP::Service.where(id_eq: 1)
+          end
+        end
+      end
+
       context '#create' do
         should 'not be implemented' do
           assert_raises ESP::NotImplementedError do

data/test/factories/custom_signature/definitions.rb

@@ -0,0 +1,30 @@
+FactoryGirl.define do
+  factory :definition, class: 'ESP::CustomSignature::Definition' do
+    skip_create
+
+    sequence(:id) { |n| n }
+    code 'abc'
+    created_at { Time.current }
+    language 'javascript'
+    status "active"
+    updated_at { Time.current }
+
+    relationships do
+      { custom_signature: {
+        data: {
+          type: "custom_signatures",
+          id: "1"
+        },
+        links: {
+          related: "http://localhost:3000/api/v2/custom_signatures/1.json"
+        }
+      },
+        results: {
+          links: {
+            related: "http://localhost:3000/api/v2/custom_signature_results.json?filter%5Bdefinition_id_eq%5D=#{id}"
+          }
+        }
+      }
+    end
+  end
+end

data/test/factories/custom_signature/result_alerts.rb

@@ -0,0 +1,94 @@
+FactoryGirl.define do
+  factory :result_alert, class: 'ESP::CustomSignature::Result::Alert' do
+    skip_create
+
+    sequence(:id) { |n| n }
+    type "alerts"
+    created_at { Time.current }
+    status "fail"
+    resource "resource-3"
+    updated_at nil
+    started_at { Time.current }
+    ended_at nil
+    metadata { { abc: '123' } }
+    tags { [{ key: 'Name', value: 'i-abc123' }] }
+    relationships do
+      {
+        external_account: {
+          data: {
+            type: "external_accounts",
+            id: "1015"
+          },
+          links: {
+            related: "http://test.host/api/v2/external_accounts/1015.json"
+          }
+        },
+        region: {
+          data: {
+            type: "regions",
+            id: "1014"
+          }
+        },
+        custom_signature: {
+          data: nil
+        }
+      }
+    end
+    included do
+      [
+        {
+          id: "1015",
+          type: "external_accounts",
+          attributes: {
+            account: "5",
+            arn: "arn:aws:iam::5:role/test_sts_role",
+            created_at: "2015-09-11T21:12:15.183Z",
+            external_id: "test_sts_external_id_1",
+            name: nil,
+            updated_at: nil,
+            user_attribution_role: nil
+          },
+          relationships: {
+            organization: {
+              data: {
+                type: "organizations",
+                id: "5"
+              },
+              links: {
+                related: "http://test.host/api/v2/organizations/5.json"
+              }
+            },
+            sub_organization: {
+              data: {
+                type: "sub_organizations",
+                id: "5"
+              },
+              links: {
+                related: "http://test.host/api/v2/sub_organizations/5.json"
+              }
+            },
+            team: {
+              data: {
+                type: "teams",
+                id: "5"
+              },
+              links: {
+                related: "http://test.host/api/v2/teams/5.json"
+              }
+            },
+            user_attribution_role: {
+              data: nil
+            }
+          }
+        },
+        {
+          id: "1014",
+          type: "regions",
+          attributes: {
+            code: "us_east_test_3"
+          }
+        }
+      ]
+    end
+  end
+end