tcell_agent 0.2.12 → 0.2.13

data/spec/lib/tcell_agent/policies/appsensor/request_size_sensor_spec.rb

@@ -0,0 +1,164 @@
+require 'spec_helper'
+
+module TCellAgent
+  module Policies
+
+    describe RequestSizeSensor do
+      context "#initialize" do
+        context "default sensor" do
+          it "should have properties set to defaults" do
+            sensor = RequestSizeSensor.new
+            expect(sensor.enabled).to eq(false)
+            expect(sensor.limit).to eq(524288)
+            expect(sensor.exclude_routes).to eq({})
+            expect(sensor.dp_code).to eq(RequestSizeSensor::DP_UNUSUAL_REQUEST_SIZE)
+          end
+        end
+
+        context "setting enabled on sensor" do
+          it "should have properties set to defaults" do
+            sensor = RequestSizeSensor.new({"enabled" => true})
+            expect(sensor.enabled).to eq(true)
+            expect(sensor.limit).to eq(524288)
+            expect(sensor.exclude_routes).to eq({})
+            expect(sensor.dp_code).to eq(RequestSizeSensor::DP_UNUSUAL_REQUEST_SIZE)
+          end
+        end
+
+        context "setting limit on sensor" do
+          it "should have properties set to defaults" do
+            sensor = RequestSizeSensor.new({"limit" => 1})
+            expect(sensor.enabled).to eq(false)
+            expect(sensor.limit).to eq(1)
+            expect(sensor.exclude_routes).to eq({})
+            expect(sensor.dp_code).to eq(RequestSizeSensor::DP_UNUSUAL_REQUEST_SIZE)
+          end
+        end
+
+        context "setting excluded routes on sensor" do
+          it "should have properties set to defaults" do
+            sensor = RequestSizeSensor.new({"exclude_routes" => ["1", "10", "20"]})
+            expect(sensor.enabled).to eq(false)
+            expect(sensor.limit).to eq(524288)
+            expect(sensor.exclude_routes).to eq({"1"=>true, "10"=>true, "20"=>true})
+            expect(sensor.dp_code).to eq(RequestSizeSensor::DP_UNUSUAL_REQUEST_SIZE)
+          end
+        end
+      end
+
+      context "#check" do
+        context "with disabled sensor" do
+          it "should not send event" do
+            sensor = RequestSizeSensor.new({"enabled" => false})
+
+            expect(sensor).to_not receive(:send_event)
+            sensor.check({}, 10)
+          end
+        end
+
+        context "with enabled sensor" do
+          context "size is nil" do
+            it "should not send event" do
+              sensor = RequestSizeSensor.new({
+                "enabled" => true,
+                "limit" => 2048,
+                "exclude_routes" => []
+              })
+
+              meta = TCellAgent::SensorEvents::AppSensorMetaEvent.new
+              meta.remote_address = "remote_address"
+              meta.method = "get"
+              meta.location = "location"
+              meta.route_id = "route_id"
+              meta.session_id = "session_id"
+              meta.user_id = "user_id"
+              meta.transaction_id = "transaction_id"
+
+              expect(sensor).to_not receive(:send_event)
+              sensor.check(meta, nil)
+            end
+          end
+
+          context "size is ok" do
+            it "should not send event" do
+              sensor = RequestSizeSensor.new({
+                "enabled" => true,
+                "limit" => 2048,
+                "exclude_routes" => []
+              })
+
+              meta = TCellAgent::SensorEvents::AppSensorMetaEvent.new
+              meta.remote_address = "remote_address"
+              meta.method = "get"
+              meta.location = "location"
+              meta.route_id = "route_id"
+              meta.session_id = "session_id"
+              meta.user_id = "user_id"
+              meta.transaction_id = "transaction_id"
+
+              expect(sensor).to_not receive(:send_event)
+              sensor.check(meta, 1024)
+            end
+          end
+
+          context "size is too big" do
+            context "route_id is excluded" do
+              it "should not send event" do
+                sensor = RequestSizeSensor.new({
+                  "enabled" => true,
+                  "limit" => 1024,
+                  "exclude_routes" => ["excluded_route_id"]
+                })
+
+                meta = TCellAgent::SensorEvents::AppSensorMetaEvent.new
+                meta.remote_address = "remote_address"
+                meta.method = "get"
+                meta.location = "location"
+                meta.route_id = "excluded_route_id"
+                meta.session_id = "session_id"
+                meta.user_id = "user_id"
+                meta.transaction_id = "transaction_id"
+
+                expect(sensor).to_not receive(:send_event)
+                sensor.check(meta, 2048)
+              end
+            end
+
+            context "route is not excluded" do
+              it "should send event" do
+                sensor = RequestSizeSensor.new({
+                  "enabled" => true,
+                  "limit" => 1024,
+                  "exclude_routes" => []
+                })
+
+                meta = TCellAgent::SensorEvents::AppSensorMetaEvent.new
+                meta.remote_address = "remote_address"
+                meta.method = "get"
+                meta.location = "location"
+                meta.route_id = "excluded_route_id"
+                meta.session_id = "session_id"
+                meta.user_id = "user_id"
+                meta.transaction_id = "transaction_id"
+
+                expect(TCellAgent).to receive(:send_event).with(
+                  {
+                    "event_type" => "as",
+                    "dp" => RequestSizeSensor::DP_UNUSUAL_REQUEST_SIZE,
+                    "param" => "2048",
+                    "remote_addr" => "remote_address",
+                    "rou" => "excluded_route_id",
+                    "m" => "get"
+                  }
+                )
+                sensor.check(meta, 2048)
+              end
+            end
+          end
+
+        end
+      end
+    end
+
+  end
+end

data/spec/lib/tcell_agent/policies/appsensor/response_codes_sensor_spec.rb

@@ -0,0 +1,194 @@
+require 'spec_helper'
+
+module TCellAgent
+  module Policies
+
+    describe ResponseCodesSensor do
+      context "#initialize" do
+        context "default sensor" do
+          it "should have properties set to defaults" do
+            sensor = ResponseCodesSensor.new
+            expect(sensor.enabled).to eq(false)
+            expect(sensor.series_400_enabled).to eq(false)
+            expect(sensor.series_500_enabled).to eq(false)
+          end
+        end
+
+        context "setting enabled on sensor" do
+          it "should have properties set to defaults" do
+            sensor = ResponseCodesSensor.new({"enabled" => true})
+            expect(sensor.enabled).to eq(true)
+            expect(sensor.series_400_enabled).to eq(false)
+            expect(sensor.series_500_enabled).to eq(false)
+          end
+        end
+
+        context "setting series_400_enabled on sensor" do
+          it "should have properties set to defaults" do
+            sensor = ResponseCodesSensor.new({"series_400_enabled" => true})
+            expect(sensor.enabled).to eq(false)
+            expect(sensor.series_400_enabled).to eq(true)
+            expect(sensor.series_500_enabled).to eq(false)
+          end
+        end
+
+        context "setting series_500_enabled on sensor" do
+          it "should have properties set to defaults" do
+            sensor = ResponseCodesSensor.new({"series_500_enabled" => true})
+            expect(sensor.enabled).to eq(false)
+            expect(sensor.series_400_enabled).to eq(false)
+            expect(sensor.series_500_enabled).to eq(true)
+          end
+        end
+      end
+
+      context "#check" do
+        context "with disabled sensor" do
+          it "should not send event" do
+            sensor = ResponseCodesSensor.new({"enabled" => false})
+            meta = TCellAgent::SensorEvents::AppSensorMetaEvent.new
+            meta.remote_address = "remote_address"
+            meta.method = "get"
+            meta.location = "location"
+            meta.route_id = "route_id"
+            meta.session_id = "session_id"
+            meta.user_id = "user_id"
+            meta.transaction_id = "transaction_id"
+
+            expect(sensor).to_not receive(:send_event)
+            sensor.check(meta, 200)
+          end
+        end
+
+        context "with enabled sensor" do
+          context "with 200 response code" do
+            it "should not send an event" do
+              sensor = ResponseCodesSensor.new({"enabled" => true})
+              meta = TCellAgent::SensorEvents::AppSensorMetaEvent.new
+              meta.remote_address = "remote_address"
+              meta.method = "get"
+              meta.location = "location"
+              meta.route_id = "route_id"
+              meta.session_id = "session_id"
+              meta.user_id = "user_id"
+              meta.transaction_id = "transaction_id"
+
+              expect(sensor).to_not receive(:send_event)
+              sensor.check(meta, 200)
+            end
+          end
+
+          context "with 300 response code" do
+            it "should not send an event" do
+              sensor = ResponseCodesSensor.new({"enabled" => true})
+              meta = TCellAgent::SensorEvents::AppSensorMetaEvent.new
+              meta.remote_address = "remote_address"
+              meta.method = "get"
+              meta.location = "location"
+              meta.route_id = "route_id"
+              meta.session_id = "session_id"
+              meta.user_id = "user_id"
+              meta.transaction_id = "transaction_id"
+
+              expect(sensor).to_not receive(:send_event)
+              sensor.check(meta, 300)
+            end
+          end
+
+          context "with disabled series_400_enabled" do
+            context "with 400 response code" do
+              it "should not send an event" do
+                sensor = ResponseCodesSensor.new({
+                  "enabled" => true,
+                  "series_400_enabled" => false
+                })
+                meta = TCellAgent::SensorEvents::AppSensorMetaEvent.new
+                meta.remote_address = "remote_address"
+                meta.method = "get"
+                meta.location = "location"
+                meta.route_id = "route_id"
+                meta.session_id = "session_id"
+                meta.user_id = "user_id"
+                meta.transaction_id = "transaction_id"
+
+                expect(sensor).to_not receive(:send_event)
+                sensor.check(meta, 400)
+              end
+            end
+          end
+
+          context "with enabled series_400_enabled" do
+            context "with 400 response code" do
+              it "should send an event" do
+                sensor = ResponseCodesSensor.new({
+                  "enabled" => true,
+                  "series_400_enabled" => true
+                })
+                meta = TCellAgent::SensorEvents::AppSensorMetaEvent.new
+                meta.remote_address = "remote_address"
+                meta.method = "get"
+                meta.location = "location"
+                meta.route_id = "route_id"
+                meta.session_id = "session_id"
+                meta.user_id = "user_id"
+                meta.transaction_id = "transaction_id"
+
+                expect(sensor).to receive(:send_event).with(
+                  meta, ResponseCodesSensor::RESPONSE_CODE_DP_DICT[4], "400", nil
+                )
+                sensor.check(meta, 400)
+              end
+            end
+          end
+
+          context "with disabled series_500_enabled" do
+            context "with 500 response code" do
+              it "should not send an event" do
+                sensor = ResponseCodesSensor.new({
+                  "enabled" => true,
+                  "series_500_enabled" => false
+                })
+                meta = TCellAgent::SensorEvents::AppSensorMetaEvent.new
+                meta.remote_address = "remote_address"
+                meta.method = "get"
+                meta.location = "location"
+                meta.route_id = "route_id"
+                meta.session_id = "session_id"
+                meta.user_id = "user_id"
+                meta.transaction_id = "transaction_id"
+
+                expect(sensor).to_not receive(:send_event)
+                sensor.check(meta, 500)
+              end
+            end
+          end
+
+          context "with enabled series_500_enabled" do
+            context "with 500 response code" do
+              it "should send an event" do
+                sensor = ResponseCodesSensor.new({
+                  "enabled" => true,
+                  "series_500_enabled" => true
+                })
+                meta = TCellAgent::SensorEvents::AppSensorMetaEvent.new
+                meta.remote_address = "remote_address"
+                meta.method = "get"
+                meta.location = "location"
+                meta.route_id = "route_id"
+                meta.session_id = "session_id"
+                meta.user_id = "user_id"
+                meta.transaction_id = "transaction_id"
+
+                expect(sensor).to receive(:send_event).with(
+                  meta, ResponseCodesSensor::RESPONSE_CODE_DP_DICT[500], "500", nil
+                )
+                sensor.check(meta, 500)
+              end
+            end
+          end
+        end
+      end
+    end
+
+  end
+end

data/spec/lib/tcell_agent/policies/appsensor/response_size_sensor_spec.rb

@@ -0,0 +1,157 @@
+require 'spec_helper'
+
+module TCellAgent
+  module Policies
+
+    describe ResponseSizeSensor do
+      context "#initialize" do
+        context "default sensor" do
+          it "should have properties set to defaults" do
+            sensor = ResponseSizeSensor.new
+            expect(sensor.enabled).to eq(false)
+            expect(sensor.limit).to eq(2097152)
+            expect(sensor.exclude_routes).to eq({})
+            expect(sensor.dp_code).to eq(ResponseSizeSensor::DP_UNUSUAL_RESPONSE_SIZE)
+          end
+        end
+
+        context "setting enabled on sensor" do
+          it "should have properties set to defaults" do
+            sensor = ResponseSizeSensor.new({"enabled" => true})
+            expect(sensor.enabled).to eq(true)
+            expect(sensor.limit).to eq(2097152)
+            expect(sensor.exclude_routes).to eq({})
+            expect(sensor.dp_code).to eq(ResponseSizeSensor::DP_UNUSUAL_RESPONSE_SIZE)
+          end
+        end
+
+        context "setting limit on sensor" do
+          it "should have properties set to defaults" do
+            sensor = ResponseSizeSensor.new({"limit" => 1})
+            expect(sensor.enabled).to eq(false)
+            expect(sensor.limit).to eq(1)
+            expect(sensor.exclude_routes).to eq({})
+            expect(sensor.dp_code).to eq(ResponseSizeSensor::DP_UNUSUAL_RESPONSE_SIZE)
+          end
+        end
+
+        context "setting excluded routes on sensor" do
+          it "should have properties set to defaults" do
+            sensor = ResponseSizeSensor.new({"exclude_routes" => ["1", "10", "20"]})
+            expect(sensor.enabled).to eq(false)
+            expect(sensor.limit).to eq(2097152)
+            expect(sensor.exclude_routes).to eq({"1"=>true, "10"=>true, "20"=>true})
+            expect(sensor.dp_code).to eq(ResponseSizeSensor::DP_UNUSUAL_RESPONSE_SIZE)
+          end
+        end
+      end
+
+      context "#check" do
+        context "with disabled sensor" do
+          it "should not send event" do
+            sensor = ResponseSizeSensor.new({"enabled" => false})
+
+            expect(sensor).to_not receive(:send_event)
+            sensor.check({}, 10)
+          end
+        end
+
+        context "with enabled sensor" do
+          context "size is nil" do
+            it "should not send event" do
+              sensor = ResponseSizeSensor.new({
+                "enabled" => true,
+                "limit" => 2048,
+                "exclude_routes" => []
+              })
+
+              meta = TCellAgent::SensorEvents::AppSensorMetaEvent.new
+              meta.remote_address = "remote_address"
+              meta.method = "get"
+              meta.location = "location"
+              meta.route_id = "route_id"
+              meta.session_id = "session_id"
+              meta.user_id = "user_id"
+              meta.transaction_id = "transaction_id"
+
+              expect(sensor).to_not receive(:send_event)
+              sensor.check(meta, nil)
+            end
+          end
+
+          context "size is ok" do
+            it "should not send event" do
+              sensor = ResponseSizeSensor.new({
+                "enabled" => true,
+                "limit" => 2048,
+                "exclude_routes" => []
+              })
+
+              meta = TCellAgent::SensorEvents::AppSensorMetaEvent.new
+              meta.remote_address = "remote_address"
+              meta.method = "get"
+              meta.location = "location"
+              meta.route_id = "route_id"
+              meta.session_id = "session_id"
+              meta.user_id = "user_id"
+              meta.transaction_id = "transaction_id"
+
+              expect(sensor).to_not receive(:send_event)
+              sensor.check(meta, 1024)
+            end
+          end
+
+          context "size is too big" do
+            context "route_id is excluded" do
+              it "should not send event" do
+                sensor = ResponseSizeSensor.new({
+                  "enabled" => true,
+                  "limit" => 1024,
+                  "exclude_routes" => ["excluded_route_id"]
+                })
+
+                meta = TCellAgent::SensorEvents::AppSensorMetaEvent.new
+                meta.remote_address = "remote_address"
+                meta.method = "get"
+                meta.location = "location"
+                meta.route_id = "excluded_route_id"
+                meta.session_id = "session_id"
+                meta.user_id = "user_id"
+                meta.transaction_id = "transaction_id"
+
+                expect(sensor).to_not receive(:send_event)
+                sensor.check(meta, 2048)
+              end
+            end
+
+            context "route is not excluded" do
+              it "should send event" do
+                sensor = ResponseSizeSensor.new({
+                  "enabled" => true,
+                  "limit" => 1024,
+                  "exclude_routes" => []
+                })
+
+                meta = TCellAgent::SensorEvents::AppSensorMetaEvent.new
+                meta.remote_address = "remote_address"
+                meta.method = "get"
+                meta.location = "location"
+                meta.route_id = "excluded_route_id"
+                meta.session_id = "session_id"
+                meta.user_id = "user_id"
+                meta.transaction_id = "transaction_id"
+
+                expect(sensor).to receive(:send_event).with(
+                  meta, ResponseSizeSensor::DP_UNUSUAL_RESPONSE_SIZE, "2048", nil
+                )
+                sensor.check(meta, 2048)
+              end
+            end
+          end
+        end
+
+      end
+    end
+
+  end
+end