| #!/usr/bin/env python |
| # Copyright 2015 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 io |
| import os |
| import sys |
| import unittest |
| |
| sys.path.insert(0, os.path.join(os.path.dirname(os.path.abspath(__file__)), |
| "..", "..", "..", "public", "tools", |
| "mojom_fetcher")) |
| sys.path.insert(0, os.path.join(os.path.dirname(os.path.abspath(__file__)), |
| "..", "..", "..", "public", "tools", |
| "mojom_fetcher", "pylib")) |
| sys.path.insert(0, os.path.join(os.path.dirname(os.path.abspath(__file__)), |
| "..", "..", "..", "public", "tools", |
| "bindings", "pylib")) |
| |
| from mojom_fetcher import UrlRewriter, UrlRewriterException, MojomFetcher |
| |
| # Fake repository for testing |
| from fakes import FakeRepository |
| |
| |
| class TestUrlRewriter(unittest.TestCase): |
| def test_no_transitive(self): |
| rules = {"foo.com": "bar.com/foo", "bar.com": "baz.com"} |
| try: |
| UrlRewriter(rules) |
| self.fail() |
| except UrlRewriterException: |
| # This is expected |
| pass |
| |
| def test_rewrite(self): |
| rules = {"foo.com": "bar.com/foo", "baz.com": "bar.com/baz"} |
| rewriter = UrlRewriter(rules) |
| self.assertEquals("bar.com/foo/foo_file", |
| rewriter.rewrite("foo.com/foo_file")) |
| self.assertEquals("bar.com/baz/foo_file", |
| rewriter.rewrite("baz.com/foo_file")) |
| self.assertEquals("example.com/file", |
| rewriter.rewrite("example.com/file")) |
| |
| |
| class FakeRequest(object): |
| def __init__(self, content, ok): |
| self.content = content |
| self.ok = ok |
| |
| |
| class FakeMojomFetcher(MojomFetcher): |
| data = """module test; |
| interface Fiz {};""" |
| |
| def __init__(self, repository, rewriter): |
| self.count = 1 |
| self.opened_files = {} |
| self.downloaded_urls = [] |
| MojomFetcher.__init__(self, repository, rewriter) |
| |
| def _requests_get(self, url): |
| self.downloaded_urls.append(url) |
| return FakeRequest(self.data, True) |
| |
| def _os_makedirs(self, _): |
| return |
| |
| def _open(self, f, _): |
| fake_file = io.BytesIO() |
| self.opened_files[f] = fake_file |
| if "services.fiz.org/foo/bar.mojom" in f: |
| self._repository.all_files_available = True |
| return fake_file |
| |
| |
| class TestMojomFetcher(unittest.TestCase): |
| def setUp(self): |
| self.rules = {"foo.com": "bar.com/foo", "baz.com": "bar.com/baz"} |
| self.rewriter = UrlRewriter(self.rules) |
| self.repository = FakeRepository("/path/to/repo", "third_party/external") |
| self.fetcher = FakeMojomFetcher(self.repository, self.rewriter) |
| |
| def test_get(self): |
| self.fetcher.get("foo.com/bar.mojom") |
| self.assertEquals(["https://bar.com/foo/bar.mojom", |
| "https://services.fiz.org/foo/bar.mojom"], |
| self.fetcher.downloaded_urls) |
| |
| def test_update(self): |
| self.fetcher.update() |
| self.assertEquals(["https://services.domokit.org/foo/fiz.mojom", |
| "https://services.fiz.org/foo/bar.mojom"], |
| self.fetcher.downloaded_urls) |
| |
| def test_discover(self): |
| self.fetcher.update() |
| self.assertEquals(["https://services.domokit.org/foo/fiz.mojom", |
| "https://services.fiz.org/foo/bar.mojom"], |
| self.fetcher.downloaded_urls) |
| |
| if __name__ == '__main__': |
| loader = unittest.defaultTestLoader |
| runner = unittest.TextTestRunner() |
| directory = os.path.dirname(os.path.abspath(__file__)) |
| suite = loader.discover(directory, '*_tests.py') |
| runner.run(suite) |
