# Copyright 2014 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

import unittest

from telemetry.internal import forwarders
from telemetry.internal.forwarders import cros_forwarder

# pylint: disable=protected-access
class ForwardingArgsTest(unittest.TestCase):
  port_pairs = forwarders.PortPairs(
      http=forwarders.PortPair(111, 222),
      https=forwarders.PortPair(333, 444),
      dns=None)

  def testForwardingArgsReverse(self):
    forwarding_args = cros_forwarder.CrOsSshForwarder._ForwardingArgs(
        use_remote_port_forwarding=True, host_ip='5.5.5.5',
        port_pairs=self.port_pairs)
    self.assertEqual(
        ['-R222:5.5.5.5:111', '-R444:5.5.5.5:333'],
        forwarding_args)

  def testForwardingArgs(self):
    forwarding_args = cros_forwarder.CrOsSshForwarder._ForwardingArgs(
        use_remote_port_forwarding=False, host_ip='2.2.2.2',
        port_pairs=self.port_pairs)
    self.assertEqual(
        ['-L111:2.2.2.2:222', '-L333:2.2.2.2:444'],
        forwarding_args)