tcell_agent 0.4.0 → 1.0.0

data/spec/lib/tcell_agent/rails/routes/routes_spec.rb

@@ -2,7 +2,6 @@ require 'spec_helper'
 
 module TCellAgent
   module Instrumentation
-
     # Fake out grape gem
     module Grape
       class API
@@ -10,19 +9,17 @@ module TCellAgent
     end
 
     class GrapeRoute < Grape::API
-
       def self.routes
-        ["/api/grape/route"]
+        ['/api/grape/route']
       end
-
     end
 
-    describe ".instrument_route" do
-      context "with a nil route" do
+    describe '.instrument_route' do
+      context 'with a nil route' do
         it "shouldn't be reported" do
-          configuration = double("configuration", enabled: true, should_instrument?: true)
+          configuration = double('configuration', :enabled => true, :should_instrument? => true)
 
-          expect(TCellAgent).to receive(:configuration).and_return(configuration).twice()
+          expect(TCellAgent).to receive(:configuration).and_return(configuration).twice
           expect(TCellAgent::Instrumentation).to_not receive(:instrument_grape_api)
           expect(TCellAgent::SensorEvents::Util).to_not receive(:calculateRouteId)
 
@@ -30,49 +27,51 @@ module TCellAgent
         end
       end
 
-      context "with rails 5.0" do
-        context "with a regular route" do
-          it "should report it" do
-            stub_const("::Rails::VERSION::MAJOR", 5)
-            stub_const("::Rails::VERSION::MINOR", 0)
+      context 'with rails 5.0' do
+        context 'with a regular route' do
+          it 'should report it' do
+            stub_const('::Rails::VERSION::MAJOR', 5)
+            stub_const('::Rails::VERSION::MINOR', 0)
 
-            configuration = double("configuration", enabled: true, should_instrument?: true)
+            configuration = double('configuration', :enabled => true, :should_instrument? => true)
             route = double(
-              "route",
-              path: double("path", spec: "/regular/route"),
-              defaults: {},
-              app: double("app", app: "some_app"))
+              'route',
+              :path => double('path', :spec => '/regular/route'),
+              :defaults => {},
+              :app => double('app', :app => 'some_app')
+            )
 
-            expect(TCellAgent).to receive(:configuration).and_return(configuration).twice()
-            expect(route).to receive(:verb).and_return("GET|POST").twice()
+            expect(TCellAgent).to receive(:configuration).and_return(configuration).twice
+            expect(route).to receive(:verb).and_return('GET|POST').twice
             expect(TCellAgent::SensorEvents::AppRoutesSensorEvent).to receive(:new).with(
-              "/regular/route", "GET", "-1187510355", nil, "{}"
+              '/regular/route', 'GET', '-1187510355', nil, '{}'
             ).and_call_original
             expect(TCellAgent::SensorEvents::AppRoutesSensorEvent).to receive(:new).with(
-              "/regular/route", "POST", "1100742947", nil, "{}"
+              '/regular/route', 'POST', '1100742947', nil, '{}'
             ).and_call_original
 
             Rails.instrument_route(route)
           end
         end
 
-        context "with a grape route" do
-          it "should instrument the grape endpoint" do
-            stub_const("::Rails::VERSION::MAJOR", 5)
-            stub_const("::Rails::VERSION::MINOR", 0)
+        context 'with a grape route' do
+          it 'should instrument the grape endpoint' do
+            stub_const('::Rails::VERSION::MAJOR', 5)
+            stub_const('::Rails::VERSION::MINOR', 0)
 
-            configuration = double("configuration", enabled: true, should_instrument?: true)
+            configuration = double('configuration', :enabled => true, :should_instrument? => true)
             route = double(
-              "route",
-              path: double("path", spec: "/grape"),
-              defaults: {},
-              app: double("app", app: GrapeRoute))
+              'route',
+              :path => double('path', :spec => '/grape'),
+              :defaults => {},
+              :app => double('app', :app => GrapeRoute)
+            )
 
-            expect(TCellAgent).to receive(:configuration).and_return(configuration).twice()
-            expect(route).to receive(:verb).and_return("GET|POST")
+            expect(TCellAgent).to receive(:configuration).and_return(configuration).twice
+            expect(route).to receive(:verb).and_return('GET|POST')
             expect(TCellAgent::SensorEvents::Util).to_not receive(:calculateRouteId)
             expect(TCellAgent::Instrumentation).to receive(:instrument_grape_api).with(
-              "/grape", ["/api/grape/route"]
+              '/grape', ['/api/grape/route']
             )
 
             Rails.instrument_route(route)
@@ -80,47 +79,49 @@ module TCellAgent
         end
       end
 
-      context "with rails 4.2" do
-        context "with a regular route" do
-          it "should report it" do
-            stub_const("::Rails::VERSION::MAJOR", 4)
-            stub_const("::Rails::VERSION::MINOR", 2)
+      context 'with rails 4.2' do
+        context 'with a regular route' do
+          it 'should report it' do
+            stub_const('::Rails::VERSION::MAJOR', 4)
+            stub_const('::Rails::VERSION::MINOR', 2)
 
-            configuration = double("configuration", enabled: true, should_instrument?: true)
+            configuration = double('configuration', :enabled => true, :should_instrument? => true)
             route = double(
-              "route",
-              path: double("path", spec: "/regular/route"),
-              defaults: {},
-              constraints: {request_method: "DELETE"},
-              app: double("app", app: "some_app"))
+              'route',
+              :path => double('path', :spec => '/regular/route'),
+              :defaults => {},
+              :constraints => { :request_method => 'DELETE' },
+              :app => double('app', :app => 'some_app')
+            )
 
-            expect(TCellAgent).to receive(:configuration).and_return(configuration).twice()
+            expect(TCellAgent).to receive(:configuration).and_return(configuration).twice
             expect(TCellAgent::Instrumentation).to_not receive(:instrument_grape_api)
             expect(route).to receive(:verb).and_return(/DELETE/).at_least(9)
             expect(TCellAgent::SensorEvents::AppRoutesSensorEvent).to receive(:new).with(
-              "/regular/route", "DELETE", "-990446408", nil, "{}"
+              '/regular/route', 'DELETE', '-990446408', nil, '{}'
             ).and_call_original
 
             Rails.instrument_route(route)
           end
         end
 
-        context "with a grape route" do
-          it "should instrument the grape endpoint" do
-            stub_const("::Rails::VERSION::MAJOR", 4)
-            stub_const("::Rails::VERSION::MINOR", 2)
+        context 'with a grape route' do
+          it 'should instrument the grape endpoint' do
+            stub_const('::Rails::VERSION::MAJOR', 4)
+            stub_const('::Rails::VERSION::MINOR', 2)
 
-            configuration = double("configuration", enabled: true, should_instrument?: true)
+            configuration = double('configuration', :enabled => true, :should_instrument? => true)
             route = double(
-              "route",
-              path: double("path", spec: "/grape"),
-              defaults: {},
-              constraints: {request_method: "PUT"},
-              app: double("app", app: GrapeRoute))
+              'route',
+              :path => double('path', :spec => '/grape'),
+              :defaults => {},
+              :constraints => { :request_method => 'PUT' },
+              :app => double('app', :app => GrapeRoute)
+            )
 
-            expect(TCellAgent).to receive(:configuration).and_return(configuration).twice()
+            expect(TCellAgent).to receive(:configuration).and_return(configuration).twice
             expect(TCellAgent::Instrumentation).to receive(:instrument_grape_api).with(
-              "/grape", ["/api/grape/route"]
+              '/grape', ['/api/grape/route']
             )
             expect(TCellAgent::SensorEvents::Util).to_not receive(:calculateRouteId)
 
@@ -129,46 +130,48 @@ module TCellAgent
         end
       end
 
-      context "with rails 4.1" do
-        context "with a regular route" do
-          it "should report it" do
-            stub_const("::Rails::VERSION::MAJOR", 4)
-            stub_const("::Rails::VERSION::MINOR", 1)
+      context 'with rails 4.1' do
+        context 'with a regular route' do
+          it 'should report it' do
+            stub_const('::Rails::VERSION::MAJOR', 4)
+            stub_const('::Rails::VERSION::MINOR', 1)
 
-            configuration = double("configuration", enabled: true, should_instrument?: true)
+            configuration = double('configuration', :enabled => true, :should_instrument? => true)
             route = double(
-              "route",
-              path: double("path", spec: "/regular/route"),
-              defaults: {},
-              constraints: {request_method: "PUT"},
-              app: "some_app")
+              'route',
+              :path => double('path', :spec => '/regular/route'),
+              :defaults => {},
+              :constraints => { :request_method => 'PUT' },
+              :app => 'some_app'
+            )
 
-            expect(TCellAgent).to receive(:configuration).and_return(configuration).twice()
+            expect(TCellAgent).to receive(:configuration).and_return(configuration).twice
             expect(TCellAgent::Instrumentation).to_not receive(:instrument_grape_api)
             expect(route).to receive(:verb).and_return(/PUT/).at_least(9)
             expect(TCellAgent::SensorEvents::AppRoutesSensorEvent).to receive(:new).with(
-              "/regular/route", "PUT", "-1393108268", nil, "{}"
+              '/regular/route', 'PUT', '-1393108268', nil, '{}'
             ).and_call_original
 
             Rails.instrument_route(route)
           end
         end
 
-        context "with a grape route" do
-          it "should instrument the grape endpoint" do
-            stub_const("::Rails::VERSION::MAJOR", 4)
-            stub_const("::Rails::VERSION::MINOR", 1)
-            configuration = double("configuration", enabled: true, should_instrument?: true)
+        context 'with a grape route' do
+          it 'should instrument the grape endpoint' do
+            stub_const('::Rails::VERSION::MAJOR', 4)
+            stub_const('::Rails::VERSION::MINOR', 1)
+            configuration = double('configuration', :enabled => true, :should_instrument? => true)
             route = double(
-              "route",
-              path: double("path", spec: "/grape"),
-              defaults: {},
-              constraints: {request_method: "PUT"},
-              app: GrapeRoute)
+              'route',
+              :path => double('path', :spec => '/grape'),
+              :defaults => {},
+              :constraints => { :request_method => 'PUT' },
+              :app => GrapeRoute
+            )
 
-            expect(TCellAgent).to receive(:configuration).and_return(configuration).twice()
+            expect(TCellAgent).to receive(:configuration).and_return(configuration).twice
             expect(TCellAgent::Instrumentation).to receive(:instrument_grape_api).with(
-              "/grape", ["/api/grape/route"]
+              '/grape', ['/api/grape/route']
             )
             expect(TCellAgent::SensorEvents::Util).to_not receive(:calculateRouteId)
 
@@ -177,6 +180,5 @@ module TCellAgent
         end
       end
     end
-
   end
 end

data/spec/lib/tcell_agent/rails_spec.rb

@@ -1,9 +1,7 @@
 require 'spec_helper'
 
 module TCellAgent
-
   class MockRackApp
-
     attr_reader :request_body
 
     def initialize
@@ -13,16 +11,14 @@ module TCellAgent
     def call(env)
       @env = env
       @request_body = env['rack.input'].read
-      [200, {'Content-Type' => 'text/plain'}, ['OK']]
+      [200, { 'Content-Type' => 'text/plain' }, ['OK']]
     end
 
     def [](key)
       @env[key]
     end
-
   end
 
-
   # describe MyMiddleware do
 
   #   let(:file_like_object) { File.open("spec/resources/normal_config.json") }
@@ -53,5 +49,4 @@ module TCellAgent
   #     end
   #   end
   # end
-
 end

data/spec/lib/tcell_agent/rust/models_spec.rb

@@ -0,0 +1,112 @@
+
+require 'spec_helper'
+
+module TCellAgent
+  module Rust
+    describe '.create_request_response' do
+      it 'should create request response' do
+        appsensor_meta = TCellAgent::SensorEvents::AppSensorMetaEvent.new(
+          'GET',
+          '192.168.1.1',
+          '12345',
+          'session_id',
+          'user_id',
+          'transaction_id',
+          'http://192.168.1.1/some/path?xss_param=<script>'
+        )
+        appsensor_meta.response_code = 200
+        appsensor_meta.user_agent = 'Mozilla'
+        appsensor_meta.path = '/some/path'
+        appsensor_meta.request_content_bytes_len = 1024
+        appsensor_meta.response_content_bytes_len = 2048
+        appsensor_meta.get_dict = { 'user' => { 'xss_param' => '<script>' } }
+        appsensor_meta.sql_exceptions = [{ 'exception_name' => 'OperationalError', 'exception_payload' => 'Developer Error' }]
+        appsensor_meta.csrf_exception_name = 'ActionController::InvalidAuthenticityToken'
+        appsensor_meta.path_parameters = { 'xss_param' => '<script>' }
+        appsensor_meta.post_dict = TCellAgent::Utils::Params.flatten({ 'already_flattened' => { 'xss_param' => '<script>' } })
+        appsensor_meta.body_dict = TCellAgent::Utils::Params.flatten({ 'already_flattened' => { 'xss_param' => '<script>' } })
+        appsensor_meta.cookie_dict = { 'xss_param' => '<script>' }
+        appsensor_meta.set_headers_dict({ 'HTTP_XSS_PARAM' => '<script>' })
+
+        result = Models.create_request_response(appsensor_meta)
+
+        expect(result).to eq({
+                               'method' => 'GET',
+                               'status_code' => 200,
+                               'route_id' => '12345',
+                               'path' => '/some/path',
+                               'query_params' => [
+                                 { 'name' => 'xss_param', 'value' => '<script>' }
+                               ],
+                               'post_params' => [
+                                 { 'name' => 'xss_param', 'value' => '<script>' },
+                                 { 'name' => 'xss_param', 'value' => '<script>' }
+                               ],
+                               'headers' => [
+                                 { 'name' => 'xss-param', 'value' => '<script>' }
+                               ],
+                               'cookies' => [
+                                 { 'name' => 'xss_param', 'value' => '<script>' }
+                               ],
+                               'path_params' => [
+                                 { 'name' => 'xss_param', 'value' => '<script>' }
+                               ],
+                               'remote_address' => '192.168.1.1',
+                               'full_uri' => 'http://192.168.1.1/some/path?xss_param=<script>',
+                               'session_id' => 'session_id',
+                               'user_id' => 'user_id',
+                               'user_agent' => 'Mozilla',
+                               'request_bytes_length' => 1024,
+                               'response_bytes_length' => 2048,
+                               'csrf_exception' => {
+                                 'exception_name' => 'ActionController::InvalidAuthenticityToken'
+                               },
+                               'sql_exceptions' => [
+                                 {
+                                   'exception_name' => 'OperationalError',
+                                   'exception_payload' => 'Developer Error'
+                                 }
+                               ]
+                             })
+      end
+    end
+
+    describe '.create_patches_request' do
+      it 'should create patches request' do
+        appsensor_meta = TCellAgent::Patches::MetaData.new(
+          'GET',
+          '192.168.1.1',
+          '12345',
+          'session_id',
+          'user_id',
+          'transaction_id',
+          'http://192.168.1.1/some/path?xss_param=<script>'
+        )
+
+        appsensor_meta.path = '/some/path'
+        appsensor_meta.request_content_bytes_len = 1024
+        appsensor_meta.get_dict = { 'user' => { 'xss_param' => '<script>' } }
+        appsensor_meta.post_dict = { 'user' => { 'xss_param' => '<script>' } }
+        appsensor_meta.body_dict = { 'user' => { 'xss_param' => '<script>' } }
+        appsensor_meta.cookie_dict = { 'xss_param' => '<script>' }
+        appsensor_meta.set_headers_dict({ 'HTTP_XSS_PARAM' => '<script>' })
+
+        result = Models.create_patches_request(appsensor_meta)
+
+        expect(result).to eq({
+                               'method' => 'GET',
+                               'path' => '/some/path',
+                               'remote_address' => '192.168.1.1',
+                               'request_bytes_length' => 1024,
+                               'query_params' => [{ 'name' => 'xss_param', 'value' => '<script>' }],
+                               'post_params' => [
+                                 { 'name' => 'xss_param', 'value' => '<script>' },
+                                 { 'name' => 'r', 'value' => { 'xss_param' => '<script>' } }
+                               ],
+                               'headers' => [{ 'name' => 'xss-param', 'value' => '<script>' }],
+                               'cookies' => [{ 'name' => 'xss_param', 'value' => '<script>' }]
+                             })
+      end
+    end
+  end
+end