rack-tracker/spec/tracker/tracker_spec.rb at master · viecode09/rack-tracker · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
# frozen_string_literal: true
class DummyHandler < Rack::Tracker::Handler
  def render
    Tilt.new( File.join( File.dirname(__FILE__), '../fixtures/dummy.erb') ).render(self)
  end

  def dummy_alert
    env['tracker']['dummy']
  end
end

class BodyHandler < DummyHandler
  self.position = :body
end

RSpec.describe Rack::Tracker do
  def app
    Rack::Builder.new do
      use Rack::Session::Cookie, secret: "FOO"

      use Rack::Tracker do
        handler DummyHandler, { foo: 'head' }
        handler BodyHandler, { foo: 'body' }
        handler DummyHandler, { foo: 'I am evil', DO_NOT_RESPECT_DNT_HEADER: true }
      end

      run lambda {|env|
        request = Rack::Request.new(env)
        case request.path
          when '/' then
            [200, {'Content-Type' => 'application/html'}, ['<head>Hello world</head>']]
          when '/body' then
            [200, {'Content-Type' => 'application/html'}, ['<body>bob here</body>']]
          when '/body-head' then
            [200, {'Content-Type' => 'application/html'}, ['<head></head><body></body>']]
          when '/test.xml' then
            [200, {'Content-Type' => 'application/xml'}, ['Xml here']]
          when '/redirect' then
            [302, {'Content-Type' => 'application/html', 'Location' => '/'}, ['<body>redirection</body>']]
          when '/moved' then
            [301, {'Content-Type' => 'application/html', 'Location' => '/redirect'}, ['<body>redirection</body>']]
          else
            [404, 'Nothing here']
        end
      }
    end
  end
  subject { app }

  describe 'when head is present' do
    it 'injects the handler code' do
      get '/'
      expect(last_response.body).to include("alert('this is a dummy class');")
    end

    it 'will pass options to the Handler' do
      get '/'
      expect(last_response.body).to include("console.log('head');")
      expect(last_response.body).to_not include("console.log('body');")
    end

    it 'injects custom variables that was directly assigned' do
      get '/', {}, {'tracker' => { 'dummy' => 'foo bar'}}
      expect(last_response.body).to include("alert('foo bar');")
    end

    it 'injects custom variables that lives in the session' do
      get '/', {}, {'rack.session' => {'tracker' => { 'dummy' => 'bar foo'}}}
      expect(last_response.body).to include("alert('bar foo');")
    end
  end

  describe 'when body is present' do
    it 'will not inject the body handler code' do
      get '/body'
      expect(last_response.body).to include("console.log('body');")
      expect(last_response.body).to_not include("console.log('head');")
    end
  end

  describe 'when head and body is present' do
    it 'will pass options to the Handler' do
      get '/body-head'
      expect(last_response.body).to include("console.log('head');")
      expect(last_response.body).to include("console.log('body');")
    end
  end

  describe 'when a redirect' do
    it 'will keep the tracker attributes and show them on the new location' do
      get '/redirect', {}, { 'tracker' => { 'dummy' => 'Keep this!' } }
      follow_redirect!
      expect(last_response.body).to include("alert('Keep this!');")
    end

    it 'will keep the tracker attributes over multiple redirects' do
      get '/moved', {}, { 'tracker' => { 'dummy' => 'Keep this twice!' } }
      follow_redirect!
      expect(last_response.body).to include("alert('Keep this twice!');")
    end
  end

  describe 'when not html' do
    it 'will not inject the handler code' do
      get '/test.xml'

      expect(last_response.body).to_not include("alert('this is a dummy class');")
    end
  end

  describe 'do not track' do
    context 'DNT header set to 1' do
      it 'will not inject any tracker' do
        get '/', {}, {'HTTP_DNT' => 1 }

        # the DummyHandler respects the DNT
        expect(last_response.body).to_not include("console.log('head');")
      end

      it 'will allow the DO_NOT_RESPECT_DNT_HEADER overwrite' do
        get '/', {}, {'HTTP_DNT' => 1 }

        # the EvilHandler respects the DNT
        expect(last_response.body).to include("console.log('I am evil');")
      end
    end

    context 'DNT header set to 0' do
      it 'injects all trackers' do
        get '/', {}, {'HTTP_DNT' => 0 }
        expect(last_response.body).to include("console.log('head');")
        expect(last_response.body).to include("console.log('I am evil');")
      end
    end

    context 'DNT header is not set' do
      it 'injects all trackers' do
        get '/'
        expect(last_response.body).to include("console.log('head');")
        expect(last_response.body).to include("console.log('I am evil');")
      end
    end

  end
end