require 'spec_helper'

describe Mixpanel::Tracker::Middleware do
  include Rack::Test::Methods

  describe "Dummy apps, no text/html" do
    before do
      setup_rack_application(DummyApp, :body => html_document, :headers => {})
      get "/"
    end

    it "should pass through if the document is not text/html content type" do
      last_response.body.should == html_document
    end
  end

  describe "Appending async mixpanel scripts" do
    describe "With ajax requests" do
      before do
        setup_rack_application(DummyApp, {:body => html_document, :headers => {"Content-Type" => "text/html"}}, {:async => true})
        get "/", {}, {"HTTP_X_REQUESTED_WITH" => "XMLHttpRequest"}
      end

      it "should not append mixpanel scripts to head element" do
        Nokogiri::HTML(last_response.body).search('script').should be_empty
      end

      it "should not update Content-Length in headers" do
        last_response.headers["Content-Length"].should == html_document.length.to_s
      end
    end
    
    describe "With large ajax response" do
      before do
        setup_rack_application(DummyApp, {:body => large_script, :headers => {"Content-Type" => "text/html"}}, {:async => true})
        get "/", {}, {"HTTP_X_REQUESTED_WITH" => "XMLHttpRequest"}
      end
      
      it "should not append mixpanel scripts to head element" do
        last_response.body.index('var mp_protocol').should be_nil
      end

      it "should pass through if the document is not text/html content type" do
        last_response.body.should == large_script
      end
    end
    
    describe "With regular requests" do
      before do
        setup_rack_application(DummyApp, {:body => html_document, :headers => {"Content-Type" => "text/html"}}, {:async => true})
        get "/"
      end

      it "should append mixpanel scripts to head element" do
        Nokogiri::HTML(last_response.body).search('head script').should_not be_empty
        Nokogiri::HTML(last_response.body).search('body script').should be_empty
      end

      it "should have 1 included script" do
        Nokogiri::HTML(last_response.body).search('script').size.should == 1
      end

      it "should use the specified token instantiating mixpanel lib" do
        last_response.should =~ /mpq.push\(\["init", "#{MIX_PANEL_TOKEN}"\]\)/
      end

      it "should define Content-Length if not exist" do
        last_response.headers.has_key?("Content-Length").should == true
      end

      it "should update Content-Length in headers" do
        last_response.headers["Content-Length"].should_not == html_document.length.to_s
      end
    end
  end

  describe "Appending mixpanel scripts" do
    describe "With ajax requests" do
      before do
        setup_rack_application(DummyApp, :body => html_document, :headers => {"Content-Type" => "text/html"})
        get "/", {}, {"HTTP_X_REQUESTED_WITH" => "XMLHttpRequest"}
      end

      it "should not append mixpanel scripts to head element" do
        Nokogiri::HTML(last_response.body).search('script').should be_empty
      end

      it "should not update Content-Length in headers" do
        last_response.headers["Content-Length"].should == html_document.length.to_s
      end
    end
    
    describe "With large ajax response" do
      before do
        setup_rack_application(DummyApp, :body => large_script, :headers => {"Content-Type" => "text/html"})
        get "/", {}, {"HTTP_X_REQUESTED_WITH" => "XMLHttpRequest"}
      end
      
      it "should not append mixpanel scripts to head element" do
        last_response.body.index('var mp_protocol').should be_nil
      end

      it "should pass through if the document is not text/html content type" do
        last_response.body.should == large_script
      end
    end
    
    describe "With regular requests" do
      before do
        setup_rack_application(DummyApp, :body => html_document, :headers => {"Content-Type" => "text/html"})
        get "/"
      end

      it "should append mixpanel scripts to head element" do
        Nokogiri::HTML(last_response.body).search('head script').should_not be_empty
        Nokogiri::HTML(last_response.body).search('body script').should be_empty
      end

      it "should have 2 included scripts" do
        Nokogiri::HTML(last_response.body).search('script').size.should == 2
      end

      it "should use the specified token instantiating mixpanel lib" do
        last_response.should =~ /new MixpanelLib\('#{MIX_PANEL_TOKEN}'\)/
      end

      it "should define Content-Length if not exist" do
        last_response.headers.has_key?("Content-Length").should == true
      end

      it "should update Content-Length in headers" do
        last_response.headers["Content-Length"].should_not == html_document.length.to_s
      end
    end
  end

  describe "Tracking async appended events" do
    before do
      @mixpanel = Mixpanel::Tracker.new(MIX_PANEL_TOKEN, {})
      @mixpanel.append_event("Visit", {:article => 1})
      @mixpanel.append_event("Sign in")
    end

    describe "With ajax requests and text/html response" do
      before do
        setup_rack_application(DummyApp, {:body => "<p>response</p>", :headers => {"Content-Type" => "text/html"}}, {:async => true})

        get "/", {}, {"mixpanel_events" => @mixpanel.queue, "HTTP_X_REQUESTED_WITH" => "XMLHttpRequest"}
      end

      it "should render only one script tag" do
        Nokogiri::HTML(last_response.body).search('script').size.should == 1
      end

      it "should be tracking the correct events inside a script tag" do
        script = Nokogiri::HTML(last_response.body).search('script')
        script.inner_html.should =~ /try\s?\{(.*)\}\s?catch/m
        script.inner_html.should =~ /mpq\.push\(\["track",\s?"Visit",\s?\{"article":1\}\]\)/
        script.inner_html.should =~ /mpq\.push\(\["track",\s?"Sign in",\s?\{\}\]\)/
      end

      it "should delete events queue after use it" do
        last_request.env.has_key?("mixpanel_events").should == false
      end
    end

    describe "With ajax requests and text/javascript response" do
      before do
        setup_rack_application(DummyApp, {:body => "alert('response')", :headers => {"Content-Type" => "text/javascript"}}, {:async => true})
        get "/", {}, {"mixpanel_events" => @mixpanel.queue, "HTTP_X_REQUESTED_WITH" => "XMLHttpRequest"}
      end

      it "should not render a script tag" do
        Nokogiri::HTML(last_response.body).search('script').size.should == 0
      end

      it "should be tracking the correct events inside a try/catch" do
        script = last_response.body.match(/try\s?\{(.*)\}\s?catch/m)[1]
        script.should =~ /mpq\.push\(\["track",\s?"Visit",\s?\{"article":1\}\]\)/
        script.should =~ /mpq\.push\(\["track",\s?"Sign in",\s?\{\}\]\)/
      end

      it "should delete events queue after use it" do
        last_request.env.has_key?("mixpanel_events").should == false
      end
    end

    describe "With regular requests" do
      before do
        setup_rack_application(DummyApp, {:body => html_document, :headers => {"Content-Type" => "text/html"}}, {:async => true})

        get "/", {}, {"mixpanel_events" => @mixpanel.queue}
      end

      it "should render 2 script tags" do
        Nokogiri::HTML(last_response.body).search('script').size.should == 2
      end

      it "should be tracking the correct events" do
        last_response.body.should =~ /mpq\.push\(\["track",\s?"Visit",\s?\{"article":1\}\]\)/
        last_response.body.should =~ /mpq\.push\(\["track",\s?"Sign in",\s?\{\}\]\)/
      end

      it "should delete events queue after use it" do
        last_request.env.has_key?("mixpanel_events").should == false
      end
    end
  end

  describe "Tracking appended events" do
    before do
      @mixpanel = Mixpanel::Tracker.new(MIX_PANEL_TOKEN, {})
      @mixpanel.append_event("Visit", {:article => 1})
      @mixpanel.append_event("Sign in")
    end

    describe "With ajax requests and text/html response" do
      before do
        setup_rack_application(DummyApp, :body => "<p>response</p>", :headers => {"Content-Type" => "text/html"})

        get "/", {}, {"mixpanel_events" => @mixpanel.queue, "HTTP_X_REQUESTED_WITH" => "XMLHttpRequest"}
      end

      it "should render only one script tag" do
        Nokogiri::HTML(last_response.body).search('script').size.should == 1
      end

      it "should be tracking the correct events inside a script tag" do
        script = Nokogiri::HTML(last_response.body).search('script')
        script.inner_html.should =~ /try\s?\{(.*)\}\s?catch/m
        script.inner_html.should =~ /mpmetrics\.track\("Visit",\s?\{"article":1\}\)/
        script.inner_html.should =~ /mpmetrics\.track\("Sign in",\s?\{\}\)/
      end

      it "should delete events queue after use it" do
        last_request.env.has_key?("mixpanel_events").should == false
      end
    end

    describe "With ajax requests and text/javascript response" do
      before do
        setup_rack_application(DummyApp, :body => "alert('response')", :headers => {"Content-Type" => "text/javascript"})
        get "/", {}, {"mixpanel_events" => @mixpanel.queue, "HTTP_X_REQUESTED_WITH" => "XMLHttpRequest"}
      end

      it "should not render a script tag" do
        Nokogiri::HTML(last_response.body).search('script').size.should == 0
      end

      it "should be tracking the correct events inside a try/catch" do
        script = last_response.body.match(/try\s?\{(.*)\}\s?catch/m)[1]
        script.should =~ /mpmetrics\.track\("Visit",\s?\{"article":1\}\)/
        script.should =~ /mpmetrics\.track\("Sign in",\s?\{\}\)/
      end

      it "should delete events queue after use it" do
        last_request.env.has_key?("mixpanel_events").should == false
      end
    end

    describe "With regular requests" do
      before do
        setup_rack_application(DummyApp, :body => html_document, :headers => {"Content-Type" => "text/html"})

        get "/", {}, {"mixpanel_events" => @mixpanel.queue}
      end

      it "should render 3 script tags" do
        Nokogiri::HTML(last_response.body).search('script').size.should == 3
      end

      it "should be tracking the correct events" do
        last_response.body.should =~ /mpmetrics\.track\("Visit",\s?\{"article":1\}\)/
        last_response.body.should =~ /mpmetrics\.track\("Sign in",\s?\{\}\)/
      end

      it "should delete events queue after use it" do
        last_request.env.has_key?("mixpanel_events").should == false
      end
    end
  end
end