tcell_agent 0.4.0 → 1.0.0

data/spec/lib/tcell_agent/sensor_events/appsensor_meta_event_spec.rb

@@ -2,39 +2,41 @@ require 'spec_helper'
 
 module TCellAgent
   module SensorEvents
-
     describe AppSensorMetaEvent do
-
-      describe "#body_params" do
+      describe '#body_params' do
         before(:each) do
           @appsensor_meta = AppSensorMetaEvent.new(
-            "get",
-            "remote_address",
-            "route_id",
-            "session_id",
-            "user_id",
-            "transaction_id")
+            'get',
+            'remote_address',
+            'route_id',
+            'session_id',
+            'user_id',
+            'transaction_id',
+            'http://test.com'
+          )
         end
 
-        context "with text/html content type" do
-          it "should set the body params to empty" do
+        context 'with text/html content type' do
+          it 'should set the body params to empty' do
             @appsensor_meta.set_body_dict(
               67,
-              "text/html",
-              {username:"tester",password:"pass"}.to_json
+              'text/html',
+              {
+                :username => 'tester',
+                :password => 'pass'
+              }.to_json
             )
 
             expect(@appsensor_meta.body_dict).to eq({})
           end
         end
 
-        context "with application/json content type" do
-
-          context "with empty request body" do
-            it "should set the body params to empty" do
+        context 'with application/json content type' do
+          context 'with empty request body' do
+            it 'should set the body params to empty' do
               @appsensor_meta.set_body_dict(
                 67,
-                "application/json",
+                'application/json',
                 nil
               )
 
@@ -42,11 +44,11 @@ module TCellAgent
             end
           end
 
-          context "with bad json in the body" do
-            it "should set the body params to empty" do
+          context 'with bad json in the body' do
+            it 'should set the body params to empty' do
               @appsensor_meta.set_body_dict(
                 67,
-                "application/json",
+                'application/json',
                 '{"username":"tester""password":"pass"}'
               )
 
@@ -54,32 +56,41 @@ module TCellAgent
             end
           end
 
-          context "with valid json in the body" do
-            it "should set the body params" do
+          context 'with valid json in the body' do
+            it 'should set the body params' do
               @appsensor_meta.set_body_dict(
                 67,
-                "application/json",
-                {username:"tester",password:"pass"}.to_json
+                'application/json',
+                {
+                  :username => 'tester',
+                  :password => 'pass'
+                }.to_json
               )
 
-              expect(@appsensor_meta.body_dict).to eq({["username"]=>"tester",["password"]=>"pass"})
+              expect(@appsensor_meta.body_dict).to eq(
+                {
+                  ['username'] => 'tester',
+                  ['password'] => 'pass'
+                }
+              )
             end
           end
 
-          context "with a json body that's too big" do
-            it "should set the body params to empty" do
+          context 'with a json body that is too big' do
+            it 'should set the body params to empty' do
               @appsensor_meta.set_body_dict(
-                20000000,
-                "application/json",
-                {username:"tester",password:"pass"}.to_json
+                20_000_000,
+                'application/json',
+                {
+                  :username => 'tester',
+                  :password => 'pass'
+                }.to_json
               )
               expect(@appsensor_meta.body_dict).to eq({})
             end
           end
         end
       end
-
     end
-
   end
 end

data/spec/lib/tcell_agent/sensor_events/dlp_spec.rb

@@ -3,10 +3,10 @@ require 'spec_helper'
 module TCellAgent
   module SensorEvents
     describe DlpEvent do
-      context "Domain from Url" do
-        it "Test Simple Domain" do
-          dlp_event = DlpEvent.new("a","b","c").for_request("x","y")
-          expect(dlp_event["context"]).to eq("x")
+      context 'Domain from Url' do
+        it 'Test Simple Domain' do
+          dlp_event = DlpEvent.new('a', 'b', 'c').for_request('x', 'y')
+          expect(dlp_event['context']).to eq('x')
         end
       end
     end

data/spec/lib/tcell_agent/sensor_events/sessions_metric_spec.rb

@@ -3,256 +3,270 @@ require 'spec_helper'
 module TCellAgent
   module SensorEvents
     describe SessionsMetric do
-      describe "#initialize" do
-        context "empty sessions info" do
-          it "should have the event type set" do
+      describe '#initialize' do
+        context 'empty sessions info' do
+          it 'should have the event type set' do
             sessions_metric = SessionsMetric.new
-            expect(sessions_metric).to eq({"event_type" => "metrics", "sessions" => {}})
+            expect(sessions_metric).to eq({ 'event_type' => 'metrics', 'sessions' => {} })
           end
         end
       end
 
-      describe "#add_session_info" do
-        context "empty sessions info" do
-          context "adding session info" do
-            it "should have the session info" do
+      describe '#add_session_info' do
+        context 'empty sessions info' do
+          context 'adding session info' do
+            it 'should have the session info' do
               sessions_metric = SessionsMetric.new
-              sessions_metric.add_session_info("hmac_session_id", "user_id", "127.0.0.1", "user_agent")
+              sessions_metric.add_session_info('hmac_session_id', 'user_id', '127.0.0.1', 'user_agent')
 
               expect(sessions_metric.has_sessions?).to eq(true)
-              expect(sessions_metric).to eq({
-                "event_type"=>"metrics",
-                "sessions"=>{
-                  "hmac_session_id" => [
-                    {
-                      "uid"=>"user_id",
-                      "track"=>[
-                        ["user_agent", ["127.0.0.1"]]
-                      ]
-                    }
-                  ]
+              expect(sessions_metric).to eq(
+                {
+                  'event_type' => 'metrics',
+                  'sessions' => {
+                    'hmac_session_id' =>  [
+                      {
+                        'uid' => 'user_id',
+                        'track' => [
+                          ['user_agent', ['127.0.0.1']]
+                        ]
+                      }
+                    ]
+                  }
                 }
-              })
+              )
             end
           end
 
-          context "adding a really long user agent" do
-            it "should truncate the agent" do
-              long_user_agent = "user_agent_user_agent_user_agent_user_agent_user_agent_user_agent_user_agent_"
-              long_user_agent += "user_agent_user_agent_user_agent_user_agent_user_agent_user_agent_user_agent"
-              long_user_agent += "user_agent_user_agent_user_agent_user_agent_user_agent_user_agent_user_agent"
-              long_user_agent += "user_agent_user_agent_user_agent_user_agent_user_agent_user_agent_user_agent"
+          context 'adding a really long user agent' do
+            it 'should truncate the agent' do
+              long_user_agent = 'user_agent_user_agent_user_agent_user_agent_user_agent_user_agent_user_agent_'
+              long_user_agent += 'user_agent_user_agent_user_agent_user_agent_user_agent_user_agent_user_agent'
+              long_user_agent += 'user_agent_user_agent_user_agent_user_agent_user_agent_user_agent_user_agent'
+              long_user_agent += 'user_agent_user_agent_user_agent_user_agent_user_agent_user_agent_user_agent'
               sessions_metric = SessionsMetric.new
-              sessions_metric.add_session_info("hmac_session_id", "user_id", "127.0.0.1", long_user_agent)
-              expect(sessions_metric).to eq({
-                "event_type"=>"metrics",
-                "sessions"=>{
-                  "hmac_session_id" => [
-                    {
-                      "uid"=>"user_id",
-                      "track"=>[
-                        ["user_agent_user_agent_user_agent_user_agent_user_agent_user_agent_user_agent_user_agent_user_agent_user_agent_user_agent_user_agent_user_agent_user_agentuser_agent_user_agent_user_agent_user_agent_user_agent_user_agent_user_agentuser_agent_user_agent_user_", ["127.0.0.1"]]
-                      ]
-                    }
-                  ]
+              sessions_metric.add_session_info('hmac_session_id', 'user_id', '127.0.0.1', long_user_agent)
+
+              expect(sessions_metric).to eq(
+                {
+                  'event_type' => 'metrics',
+                  'sessions' => {
+                    'hmac_session_id' =>  [
+                      {
+                        'uid' => 'user_id',
+                        'track' => [
+                          ['user_agent_user_agent_user_agent_user_agent_user_agent_user_agent_user_agent_user_agent_user_agent_user_agent_user_agent_user_agent_user_agent_user_agentuser_agent_user_agent_user_agent_user_agent_user_agent_user_agent_user_agentuser_agent_user_agent_user_', ['127.0.0.1']]
+                        ]
+                      }
+                    ]
+                  }
                 }
-              })
+              )
             end
           end
         end
 
-        context "with an existing session info" do
-          context "adding the same info" do
-            it "should not do anything" do
+        context 'with an existing session info' do
+          context 'adding the same info' do
+            it 'should not do anything' do
               sessions_metric = SessionsMetric.new
-              sessions_metric.add_session_info("hmac_session_id", "user_id", "127.0.0.1", "user_agent")
-              sessions_metric.add_session_info("hmac_session_id", "user_id", "127.0.0.1", "user_agent")
+              sessions_metric.add_session_info('hmac_session_id', 'user_id', '127.0.0.1', 'user_agent')
+              sessions_metric.add_session_info('hmac_session_id', 'user_id', '127.0.0.1', 'user_agent')
 
-              expect(sessions_metric).to eq({
-                "event_type"=>"metrics",
-                "sessions"=>{
-                  "hmac_session_id" => [
-                    {
-                      "uid"=>"user_id",
-                      "track"=>[
-                        ["user_agent", ["127.0.0.1"]]
-                      ]
-                    }
-                  ]
+              expect(sessions_metric).to eq(
+                {
+                  'event_type' => 'metrics',
+                  'sessions' => {
+                    'hmac_session_id' =>  [
+                      {
+                        'uid' => 'user_id',
+                        'track' => [
+                          ['user_agent', ['127.0.0.1']]
+                        ]
+                      }
+                    ]
+                  }
                 }
-              })
+              )
             end
           end
 
-
-          context "adding a new session" do
-            it "should add the new info" do
+          context 'adding a new session' do
+            it 'should add the new info' do
               sessions_metric = SessionsMetric.new
-              sessions_metric.add_session_info("hmac_session_id", "user_id", "127.0.0.1", "user_agent")
-
-              sessions_metric.add_session_info("hmac_session_id_dos", "user_id_dos", "127.0.0.1_dos", "user_agent_dos")
-              expect(sessions_metric).to eq({
-                "event_type"=>"metrics",
-                "sessions"=>{
-                  "hmac_session_id" => [
-                    {
-                      "uid"=>"user_id",
-                      "track"=>[
-                        ["user_agent", ["127.0.0.1"]]
-                      ]
-                    }
-                  ],
-                  "hmac_session_id_dos" => [
-                    {
-                      "uid"=>"user_id_dos",
-                      "track"=>[
-                        ["user_agent_dos", ["127.0.0.1_dos"]]
-                      ]
-                    }
-                  ]
-                }
-              })
-            end
-
-            context "with the same user_id" do
-              it "should add the new info" do
-                sessions_metric = SessionsMetric.new
-                sessions_metric.add_session_info("hmac_session_id", "user_id", "127.0.0.1", "user_agent")
+              sessions_metric.add_session_info('hmac_session_id', 'user_id', '127.0.0.1', 'user_agent')
 
-                sessions_metric.add_session_info("hmac_session_id_dos", "user_id", "127.0.0.1", "user_agent")
-                expect(sessions_metric).to eq({
-                  "event_type"=>"metrics",
-                  "sessions"=>{
-                    "hmac_session_id" => [
+              sessions_metric.add_session_info('hmac_session_id_dos', 'user_id_dos', '127.0.0.1_dos', 'user_agent_dos')
+              expect(sessions_metric).to eq(
+                {
+                  'event_type' => 'metrics',
+                  'sessions' => {
+                    'hmac_session_id' =>  [
                       {
-                        "uid"=>"user_id",
-                        "track"=>[
-                          ["user_agent", ["127.0.0.1"]]
+                        'uid' => 'user_id',
+                        'track' => [
+                          ['user_agent', ['127.0.0.1']]
                         ]
                       }
                     ],
-                    "hmac_session_id_dos" => [
+                    'hmac_session_id_dos' =>  [
                       {
-                        "uid"=>"user_id",
-                        "track"=>[
-                          ["user_agent", ["127.0.0.1"]]
+                        'uid' => 'user_id_dos',
+                        'track' => [
+                          ['user_agent_dos', ['127.0.0.1_dos']]
                         ]
                       }
                     ]
                   }
-                })
+                }
+              )
+            end
+
+            context 'with the same user_id' do
+              it 'should add the new info' do
+                sessions_metric = SessionsMetric.new
+                sessions_metric.add_session_info('hmac_session_id', 'user_id', '127.0.0.1', 'user_agent')
+
+                sessions_metric.add_session_info('hmac_session_id_dos', 'user_id', '127.0.0.1', 'user_agent')
+                expect(sessions_metric).to eq(
+                  {
+                    'event_type' => 'metrics',
+                    'sessions' => {
+                      'hmac_session_id' =>  [
+                        {
+                          'uid' => 'user_id',
+                          'track' => [
+                            ['user_agent', ['127.0.0.1']]
+                          ]
+                        }
+                      ],
+                      'hmac_session_id_dos' =>  [
+                        {
+                          'uid' => 'user_id',
+                          'track' => [
+                            ['user_agent', ['127.0.0.1']]
+                          ]
+                        }
+                      ]
+                    }
+                  }
+                )
               end
             end
           end
 
-          context "adding new info for the existing session" do
-            context "adding a new user_id" do
-              it "should append the ip address to the existing info" do
+          context 'adding new info for the existing session' do
+            context 'adding a new user_id' do
+              it 'should append the ip address to the existing info' do
                 sessions_metric = SessionsMetric.new
-                sessions_metric.add_session_info("hmac_session_id", "user_id", "127.0.0.1", "user_agent")
-                sessions_metric.add_session_info("hmac_session_id", "user_id_dos", "127.0.0.1", "user_agent")
+                sessions_metric.add_session_info('hmac_session_id', 'user_id', '127.0.0.1', 'user_agent')
+                sessions_metric.add_session_info('hmac_session_id', 'user_id_dos', '127.0.0.1', 'user_agent')
 
-                expect(sessions_metric).to eq({
-                  "event_type"=>"metrics",
-                  "sessions"=>{
-                    "hmac_session_id" => [
-                      {
-                        "uid"=>"user_id",
-                        "track"=>[
-                          ["user_agent", ["127.0.0.1"]]
-                        ]
-                      },
-                      {
-                        "uid"=>"user_id_dos",
-                        "track"=>[
-                          ["user_agent", ["127.0.0.1"]]
-                        ]
-                      }
-                    ]
+                expect(sessions_metric).to eq(
+                  {
+                    'event_type' => 'metrics',
+                    'sessions' => {
+                      'hmac_session_id' =>  [
+                        {
+                          'uid' => 'user_id',
+                          'track' => [
+                            ['user_agent', ['127.0.0.1']]
+                          ]
+                        },
+                        {
+                          'uid' => 'user_id_dos',
+                          'track' => [
+                            ['user_agent', ['127.0.0.1']]
+                          ]
+                        }
+                      ]
+                    }
                   }
-                })
+                )
               end
             end
 
-            context "adding a new ip address" do
-              it "should append the ip address to the existing info" do
+            context 'adding a new ip address' do
+              it 'should append the ip address to the existing info' do
                 sessions_metric = SessionsMetric.new
-                sessions_metric.add_session_info("hmac_session_id", "user_id", "127.0.0.1", "user_agent")
-                sessions_metric.add_session_info("hmac_session_id", "user_id", "127.0.0.1_dos", "user_agent")
+                sessions_metric.add_session_info('hmac_session_id', 'user_id', '127.0.0.1', 'user_agent')
+                sessions_metric.add_session_info('hmac_session_id', 'user_id', '127.0.0.1_dos', 'user_agent')
 
-                expect(sessions_metric).to eq({
-                  "event_type"=>"metrics",
-                  "sessions"=>{
-                    "hmac_session_id" => [
-                      {
-                        "uid"=>"user_id",
-                        "track"=>[
-                          ["user_agent", ["127.0.0.1", "127.0.0.1_dos"]]
-                        ]
-                      }
-                    ]
+                expect(sessions_metric).to eq(
+                  {
+                    'event_type' => 'metrics',
+                    'sessions' => {
+                      'hmac_session_id' =>  [
+                        {
+                          'uid' => 'user_id',
+                          'track' => [
+                            ['user_agent', ['127.0.0.1', '127.0.0.1_dos']]
+                          ]
+                        }
+                      ]
+                    }
                   }
-                })
+                )
               end
             end
 
-            context "adding a new user agent" do
-              it "should add the new user agent" do
+            context 'adding a new user agent' do
+              it 'should add the new user agent' do
                 sessions_metric = SessionsMetric.new
-                sessions_metric.add_session_info("hmac_session_id", "user_id", "127.0.0.1", "user_agent")
-                sessions_metric.add_session_info("hmac_session_id", "user_id", "127.0.0.1", "user_agent_dos")
+                sessions_metric.add_session_info('hmac_session_id', 'user_id', '127.0.0.1', 'user_agent')
+                sessions_metric.add_session_info('hmac_session_id', 'user_id', '127.0.0.1', 'user_agent_dos')
 
-                expect(sessions_metric).to eq({
-                  "event_type"=>"metrics",
-                  "sessions"=>{
-                    "hmac_session_id" => [
-                      {
-                        "uid"=>"user_id",
-                        "track"=>[
-                          ["user_agent", ["127.0.0.1"]],
-                          ["user_agent_dos", ["127.0.0.1"]]
-                        ]
-                      }
-                    ]
+                expect(sessions_metric).to eq(
+                  {
+                    'event_type' => 'metrics',
+                    'sessions' => {
+                      'hmac_session_id' =>  [
+                        {
+                          'uid' => 'user_id',
+                          'track' => [
+                            ['user_agent', ['127.0.0.1']],
+                            ['user_agent_dos', ['127.0.0.1']]
+                          ]
+                        }
+                      ]
+                    }
                   }
-                })
+                )
               end
             end
           end
-
         end
 
-        context "adding 200 sessions" do
-          it "should update its state to flush" do
+        context 'adding 200 sessions' do
+          it 'should update its state to flush' do
             sessions_metric = SessionsMetric.new
             expect(sessions_metric.flush).to eq(false)
 
             200.times do |x|
-              sessions_metric.add_session_info("hmac_session_id", "user_id", "127.0.0.1_#{x}", "user_agent")
+              sessions_metric.add_session_info('hmac_session_id', 'user_id', "127.0.0.1_#{x}", 'user_agent')
             end
 
             expect(sessions_metric.flush).to eq(true)
           end
         end
 
-        context "adding 251 sessions" do
-          it "should drop session after 250" do
-            logger = double("logger")
+        context 'adding 251 sessions' do
+          it 'should drop session after 250' do
+            logger = double('logger')
             sessions_metric = SessionsMetric.new
 
             expect(sessions_metric.flush).to eq(false)
             expect(TCellAgent).to receive(:logger).and_return(logger)
-            expect(logger).to receive(:warn).with("Sessions Metric is full. Information dropped")
+            expect(logger).to receive(:warn).with('Sessions Metric is full. Information dropped')
 
             250.times do |x|
-              sessions_metric.add_session_info("hmac_session_id", "user_id", "127.0.0.1_#{x}", "user_agent")
+              sessions_metric.add_session_info('hmac_session_id', 'user_id', "127.0.0.1_#{x}", 'user_agent')
             end
 
             expect(sessions_metric.flush).to eq(true)
           end
         end
       end
-
     end
   end
 end