#!/usr/bin/env python
#-*- coding: utf-8 -*-
#  Copyright (c) 2012 The WebRTC project authors. All Rights Reserved.
#
#  Use of this source code is governed by a BSD-style license
#  that can be found in the LICENSE file in the root of the source
#  tree. An additional intellectual property rights grant can be found
#  in the file PATENTS.  All contributing project authors may
#  be found in the AUTHORS file in the root of the source tree.

import unittest

import peerconnection_fuzz

SAMPLE_TEMPLATE = """
﻿<html>
<head>
  <title>WebRTC PeerConnection Fuzz Test Template</title>
  <script type="text/javascript">
  var gFirstConnection = null;

// START_OF_POSSIBLE_INJECTED_LOCATION_RELOADS
  function startTest() {
    navigator.webkitGetUserMedia(REQUEST_AUDIO_AND_VIDEO,
                                 getUserMediaOkCallback,
                                 getUserMediaFailedCallback);
  }

  function callUsingStream(localStream) {
    gFirstConnection.createOffer(onOfferCreated);
  }

  function onIceCandidateToFirst(event) {
    if (event.candidate) {
      var candidate = new RTCIceCandidate(event.candidate);
      gSecondConnection.addIceCandidate(candidate);
    }
  }
// END_OF_POSSIBLE_INJECTED_LOCATION_RELOADS
  </script>
</head>"""

RELOADED_TEMPLATE = """
﻿<html>
<head>
  <title>WebRTC PeerConnection Fuzz Test Template</title>
  <script type="text/javascript">
  var gFirstConnection = null;


  function startTest() {
    navigator.webkitGetUserMedia(REQUEST_AUDIO_AND_VIDEO,
                                 getUserMediaOkCallback,
                                 getUserMediaFailedCallback);
    location.reload();
  }

  function callUsingStream(localStream) {
    gFirstConnection.createOffer(onOfferCreated);
    location.reload();
  }

  function onIceCandidateToFirst(event) {
    if (event.candidate) {
      var candidate = new RTCIceCandidate(event.candidate);
    location.reload();
      gSecondConnection.addIceCandidate(candidate);
    location.reload();
    }
  }

  </script>
</head>"""

class PeerConnectionFuzzerTest(unittest.TestCase):
  def testInsertRandomReloadsInsertsAtTheRightPlace(self):
    """Tests we can insert location.reload() in the right places.

    Only tests the case where we replace all since the other case is random.
    """
    result = peerconnection_fuzz._InsertRandomLocationReloadsWithinMarkers(
        SAMPLE_TEMPLATE, replace_all=True)
    self.assertEquals(RELOADED_TEMPLATE, result,
                      'Got "%s", should be "%s"' % (result, RELOADED_TEMPLATE))

if __name__ == '__main__':
  unittest.main()