mirror of
https://github.com/bluxmit/alnoda-workspaces.git
synced 2024-04-29 19:52:19 +12:00
72 lines
2.8 KiB
Python
72 lines
2.8 KiB
Python
# encoding: utf-8
|
|
import os
|
|
from unittest import TestCase
|
|
|
|
from ....utils import transform_ipynb_uri
|
|
from ..handlers import uri_rewrites
|
|
|
|
uri_rewrite_list = uri_rewrites()
|
|
|
|
|
|
class TestRewrite(TestCase):
|
|
def assert_rewrite(self, uri, rewrite):
|
|
new = transform_ipynb_uri(uri, uri_rewrite_list)
|
|
self.assertEqual(new, rewrite)
|
|
|
|
def assert_rewrite_ghe(self, uri, rewrite):
|
|
os.environ["GITHUB_API_URL"] = "https://example.com/api/v3/"
|
|
uri_rewrite_ghe_list = uri_rewrites()
|
|
os.environ.pop("GITHUB_API_URL", None)
|
|
new = transform_ipynb_uri(uri, uri_rewrite_ghe_list)
|
|
self.assertEqual(new, rewrite)
|
|
|
|
def test_githubusercontent(self):
|
|
uri = u"https://raw.githubusercontent.com/user/reopname/deadbeef/a mřížka.ipynb"
|
|
rewrite = u"/github/user/reopname/blob/deadbeef/a mřížka.ipynb"
|
|
self.assert_rewrite(uri, rewrite)
|
|
|
|
def test_blob(self):
|
|
uri = u"https://github.com/user/reopname/blob/deadbeef/a mřížka.ipynb"
|
|
rewrite = u"/github/user/reopname/blob/deadbeef/a mřížka.ipynb"
|
|
self.assert_rewrite(uri, rewrite)
|
|
|
|
def test_raw_uri(self):
|
|
uri = u"https://github.com/user/reopname/raw/deadbeef/a mřížka.ipynb"
|
|
rewrite = u"/github/user/reopname/blob/deadbeef/a mřížka.ipynb"
|
|
self.assert_rewrite(uri, rewrite)
|
|
|
|
def test_raw_subdomain(self):
|
|
uri = u"https://raw.github.com/user/reopname/deadbeef/a mřížka.ipynb"
|
|
rewrite = u"/github/user/reopname/blob/deadbeef/a mřížka.ipynb"
|
|
self.assert_rewrite(uri, rewrite)
|
|
|
|
def test_tree(self):
|
|
uri = u"https://github.com/user/reopname/tree/deadbeef/a mřížka.ipynb"
|
|
rewrite = u"/github/user/reopname/tree/deadbeef/a mřížka.ipynb"
|
|
self.assert_rewrite(uri, rewrite)
|
|
|
|
def test_userrepo(self):
|
|
uri = u"username/reponame"
|
|
rewrite = u"/github/username/reponame/tree/master/"
|
|
self.assert_rewrite(uri, rewrite)
|
|
|
|
def test_user(self):
|
|
uri = u"username"
|
|
rewrite = u"/github/username/"
|
|
self.assert_rewrite(uri, rewrite)
|
|
|
|
def test_ghe_blob(self):
|
|
uri = u"https://example.com/user/reopname/blob/deadbeef/a mřížka.ipynb"
|
|
rewrite = u"/github/user/reopname/blob/deadbeef/a mřížka.ipynb"
|
|
self.assert_rewrite_ghe(uri, rewrite)
|
|
|
|
def test_ghe_raw_uri(self):
|
|
uri = u"https://example.com/user/reopname/raw/deadbeef/a mřížka.ipynb"
|
|
rewrite = u"/github/user/reopname/blob/deadbeef/a mřížka.ipynb"
|
|
self.assert_rewrite_ghe(uri, rewrite)
|
|
|
|
def test_ghe_tree(self):
|
|
uri = u"https://example.com/user/reopname/tree/deadbeef/a mřížka.ipynb"
|
|
rewrite = u"/github/user/reopname/tree/deadbeef/a mřížka.ipynb"
|
|
self.assert_rewrite_ghe(uri, rewrite)
|