require 'spec_helper'
require 'rtsp/response'

describe RTSP::Response do
  describe "#initialize" do
    it "expects a non-nil string on" do
      expect { RTSP::Response.new(nil) }.to raise_exception RTSP::Error
    end

    it "expects a non-empty string on" do
      expect { RTSP::Response.new("") }.to raise_exception RTSP::Error
    end
  end

  describe "#extract_status_line" do
    before do
      RTSP::Response.any_instance.stub(:split_head_and_body_from)
      RTSP::Response.any_instance.stub(:parse_head)
      RTSP::Response.any_instance.stub(:parse_body)
    end

    subject { RTSP::Response.new " " }

    context "RTSP response" do
      let(:status_line) { "RTSP/1.0 200 OK\r\n" }
      before { subject.extract_status_line(status_line) }
      specify {
        subject.rtsp_version.should == "1.0"
        subject.code.should == 200
        subject.message.should == "OK"
      }
    end

    context "HTTP response" do
      let(:status_line) { "HTTP/1.1 200 OK\r\n" }
      before { subject.extract_status_line(status_line) }
      specify {
        subject.rtsp_version.should == "1.1"
        subject.code.should == 200
        subject.message.should == "OK"
      }
    end
  end

  describe "#parse_head" do
    let(:head) do
      head = double "head"
      head.stub(:split).and_return(['', session_line])
      head.stub(:each_with_index).and_yield

      head
    end

    context "Session header contains session-id and timeout" do
      let(:session_line) { "Session: 118;timeout=49" }
      subject { RTSP::Response.new SETUP_RESPONSE_WITH_SESSION_TIMEOUT }

      it "creates a :session reader with value being a Hash with key/value" do
        subject.stub(:extract_status_line)
        subject.should_receive(:create_reader).with("session",
          { session_id: "118", timeout: 49 })
        subject.parse_head(head)
      end
    end

    context "Session header contains just session-id" do
      let(:session_line) { "Session: 118" }
      subject { RTSP::Response.new SETUP_RESPONSE_WITH_SESSION_TIMEOUT }

      it "creates a :session reader with value being a Hash with key/value" do
        subject.stub(:extract_status_line)
        subject.should_receive(:create_reader).with("session",
          { session_id: "118" })
        subject.parse_head(head)
      end
    end

    subject { RTSP::Response.new OPTIONS_RESPONSE }

    it "raises when RTSP version is corrupted" do
      expect { subject.parse_head "RTSP/ 200 OK\r\n" }.to raise_error RTSP::Error
    end

    it "raises when the response code is corrupted" do
      expect { subject.parse_head "RTSP/1.0 2 OK\r\n" }.to raise_error RTSP::Error
    end

    it "raises when the response message is corrupted" do
      expect { subject.parse_head "RTSP/1.0 200 \r\n" }.to raise_error RTSP::Error
    end
  end

  describe "#parse_body" do
    it "returns an SDP::Description when @content_type is 'application/sdp" do
      response = RTSP::Response.new DESCRIBE_RESPONSE
      sdp = SDP::Description.new
      sdp.username = "me"
      sdp.id = 12345
      sdp.version = 12345
      sdp.network_type = "IN"
      sdp.address_type = "IP4"
      body = response.parse_body sdp.to_s
      body.class.should == SDP::Description
    end

    it "returns the text that was passed to it but with line feeds removed" do
      response = RTSP::Response.new OPTIONS_RESPONSE
      string = "hi\r\nguys\r\n\r\n"
      body = response.parse_body string
      body.class.should == String
      body.should == string
    end
  end

  describe "#to_s" do
    it "returns the text that was passed in" do
      response = RTSP::Response.new OPTIONS_RESPONSE
      response.to_s.should == OPTIONS_RESPONSE
    end
  end

  describe "#inspect" do
    before do
      @response = RTSP::Response.new OPTIONS_RESPONSE
    end

    it "contains the class name and object ID first" do
      @response.inspect.should match(/^#<RTSP::Response:\d+/)
    end

    it "begins with <# and ends with >" do
      @response.inspect.should match(/^#<.*>$/)
    end
  end

  context "options" do
    before do
      @response = RTSP::Response.new OPTIONS_RESPONSE
    end

    it "returns a 200 code" do
      @response.code.should == 200
    end

    it "returns 'OK' message" do
      @response.message.should == 'OK'
    end

    it "returns the date header" do
      @response.date.should == 'Fri, Jan 28 2011 01:14:42 GMT'
    end

    it "returns the supported methods in the Public header" do
      @response.public.should == 'OPTIONS, DESCRIBE, SETUP, TEARDOWN, PLAY, PAUSE'
    end
  end

  context "describe" do
    before do
      @response = RTSP::Response.new DESCRIBE_RESPONSE
    end

    it "returns a 200 code" do
      @response.code.should == 200
    end

    it "returns 'OK' message" do
      @response.message.should == "OK"
    end

    it "returns all header fields" do
      @response.server.should == "DSS/5.5 (Build/489.7; Platform/Linux; Release/Darwin; )"
      @response.cseq.should == 1
      @response.cache_control.should == "no-cache"
      @response.content_length.should == 406
      @response.date.should == "Sun, 23 Jan 2011 00:36:45 GMT"
      @response.expires.should == "Sun, 23 Jan 2011 00:36:45 GMT"
      @response.content_type.should == "application/sdp"
      @response.x_accept_retransmit.should == "our-retransmit"
      @response.x_accept_dynamic_rate.should == 1
      @response.content_base.should == "rtsp://64.202.98.91:554/gs.sdp/"
    end

    it "body is a parsed SDP::Description" do
      @response.body.should be_kind_of SDP::Description
      sdp_info = @response.body
      sdp_info.protocol_version.should == "0"
      sdp_info.name.should == "Groove Salad from SomaFM [aacPlus]"
    end
  end

  context "setup" do
    before do
      @response = RTSP::Response.new SETUP_RESPONSE
    end

    it "returns a 200 code" do
      @response.code.should == 200
    end

    it "returns 'OK' message" do
      @response.message.should == 'OK'
    end

    it "returns the date header" do
      @response.date.should == 'Fri, Jan 28 2011 01:14:42 GMT'
    end

    it "returns the supported transport" do
      @response.transport.should == 'RTP/AVP;unicast;destination=10.221.222.186;source=10.221.222.235;client_port=9000-9001;server_port=6700-6701'
    end

    it "returns the session" do
      @response.session[:session_id].should == "118"
    end
  end

  context "play" do
    before do
      @response = RTSP::Response.new PLAY_RESPONSE
    end

    it "returns a 200 code" do
      @response.code.should == 200
    end

    it "returns 'OK' message" do
      @response.message.should == 'OK'
    end

    it "returns the date header" do
      @response.date.should == 'Fri, Jan 28 2011 01:14:42 GMT'
    end

    it "returns the supported range" do
      @response.range.should == 'npt=0.000-'
    end

    it "returns the session" do
      @response.session[:session_id].should == "118"
    end

    it "returns the rtp_info" do
      @response.rtp_info.should == 'url=rtsp://10.221.222.235/stream1/track1;seq=17320;rtptime=400880602'
    end
  end

  context "teardown" do
    before do
      @response = RTSP::Response.new TEARDOWN_RESPONSE
    end

    it "returns a 200 code" do
      @response.code.should == 200
    end

    it "returns 'OK' message" do
      @response.message.should == 'OK'
    end

    it "returns the date header" do
      @response.date.should == 'Fri, Jan 28 2011 01:14:47 GMT'
    end
  end

  context "#parse_head" do
    before do
      @response = RTSP::Response.new OPTIONS_RESPONSE
    end

    it "extracts the RTSP version from the header" do
      @response.rtsp_version.should == "1.0"
    end

    it "extracts the response code from the header as a Fixnum" do
      @response.code.is_a?(Fixnum).should be_true
      @response.code.should == 200
    end

    it "extracts the response message from the header" do
      @response.message.should == "OK"
    end

    it "returns empty value string when header has no value" do
      response = RTSP::Response.new NO_CSEQ_VALUE_RESPONSE
      response.parse_head NO_CSEQ_VALUE_RESPONSE
      response.instance_variable_get(:@cseq).should == ""
    end
  end

  describe "#split_head_and_body_from" do
    it "splits responses with headers and no body" do
      response = RTSP::Response.new OPTIONS_RESPONSE
      head_and_body = response.split_head_and_body_from OPTIONS_RESPONSE
      head_and_body.first.should == %Q{RTSP/1.0 200 OK\r
CSeq: 1\r
Date: Fri, Jan 28 2011 01:14:42 GMT\r
Public: OPTIONS, DESCRIBE, SETUP, TEARDOWN, PLAY, PAUSE\r
}
    end

    it "splits responses with headers and body" do
      response = RTSP::Response.new DESCRIBE_RESPONSE
      head_and_body = response.split_head_and_body_from DESCRIBE_RESPONSE
      head_and_body.first.should == %{RTSP/1.0 200 OK\r
Server: DSS/5.5 (Build/489.7; Platform/Linux; Release/Darwin; )\r
Cseq: 1\r
Cache-Control: no-cache\r
Content-length: 406\r
Date: Sun, 23 Jan 2011 00:36:45 GMT\r
Expires: Sun, 23 Jan 2011 00:36:45 GMT\r
Content-Type: application/sdp\r
x-Accept-Retransmit: our-retransmit\r
x-Accept-Dynamic-Rate: 1\r
Content-Base: rtsp://64.202.98.91:554/gs.sdp/}

      head_and_body.last.should == %{\r
v=0\r
o=- 545877020 467920391 IN IP4 127.0.0.1\r
s=Groove Salad from SomaFM [aacPlus]\r
i=Downtempo Ambient Groove\r
c=IN IP4 0.0.0.0\r
t=0 0\r
a=x-qt-text-cmt:Orban Opticodec-PC\r
a=x-qt-text-nam:Groove Salad from SomaFM [aacPlus]\r
a=x-qt-text-inf:Downtempo Ambient Groove\r
a=control:*\r
m=audio 0 RTP/AVP 96\r
b=AS:48\r
a=rtpmap:96 MP4A-LATM/44100/2\r
a=fmtp:96 cpresent=0;config=400027200000\r
a=control:trackID=1\r
}
    end
  end
end