blob: 0d9084881c9670ced35b88df8699b06bf58ffbfa [file] [log] [blame] [edit]
#!/usr/bin/python2
# Copyright 2016 The Chromium OS 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 six.moves.urllib.parse
import unittest
import common
from autotest_lib.server import afe_urls
class AfeUrlsTestCase(unittest.TestCase):
"""Tests for AfeUrls."""
def assertURLEqual(self, a, b):
"""Assert two URLs are equal.
@param a First URL to compare
@param b First URL to compare
"""
urlsplit = six.moves.urllib.parse.urlsplit
self.assertEqual(urlsplit(a), urlsplit(b))
def test__geturl(self):
"""Test _geturl() happy path."""
urls = afe_urls.AfeUrls('http://localhost/afe/')
got = urls._geturl({'foo': 'bar', 'spam': 'eggs'})
self.assertURLEqual(got, 'http://localhost/afe/#foo=bar&spam=eggs')
def test_get_host_url(self):
"""Test get_host_url() happy path."""
urls = afe_urls.AfeUrls('http://localhost/afe/')
got = urls.get_host_url(42)
self.assertURLEqual(
got,
'http://localhost/afe/#tab_id=view_host&object_id=42')
def test_root_url(self):
"""Test happy path for root_url attribute."""
urls = afe_urls.AfeUrls('http://localhost/afe/')
self.assertEqual(urls.root_url, 'http://localhost/afe/')
def test_equal(self):
"""Test happy path for equality."""
urls1 = afe_urls.AfeUrls('http://localhost/afe/')
urls2 = afe_urls.AfeUrls('http://localhost/afe/')
self.assertEqual(urls1, urls2)
def test_from_hostname(self):
"""Test from_hostname() happy path."""
urls = afe_urls.AfeUrls.from_hostname('sharanohiar')
self.assertEqual(urls.root_url, 'http://sharanohiar/afe/')
if __name__ == '__main__':
unittest.main()