mirror of https://github.com/bazelbuild/rules_pkg
141 lines
4.5 KiB
Python
141 lines
4.5 KiB
Python
# Copyright 2016 The Bazel Authors. All rights reserved.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
"""Testing for helper functions."""
|
|
|
|
import collections
|
|
import imp
|
|
import os
|
|
import unittest
|
|
|
|
|
|
pkg_bzl = imp.load_source('pkg_bzl', 'pkg/path.bzl')
|
|
|
|
Owner = collections.namedtuple('Owner', ['workspace_name', 'workspace_root'])
|
|
Root = collections.namedtuple('Root', ['path'])
|
|
|
|
|
|
class File(object):
|
|
"""Mock Skylark File class for testing."""
|
|
|
|
def __init__(self, short_path, is_generated=False, is_external=False):
|
|
self.is_source = not is_generated
|
|
self.root = Root('bazel-out/k8-fastbuild/bin' if is_generated else '')
|
|
if is_external:
|
|
self.owner = Owner('repo', 'external/repo')
|
|
self.short_path = '../repo/' + short_path
|
|
else:
|
|
self.owner = Owner('', '')
|
|
self.short_path = short_path
|
|
self.path = os.path.join(
|
|
self.root.path, self.owner.workspace_root, short_path)
|
|
|
|
|
|
class SafeShortPathTest(unittest.TestCase):
|
|
"""Testing for safe_short_path."""
|
|
|
|
def testSafeShortPath(self):
|
|
path = pkg_bzl.safe_short_path(File('foo/bar/baz'))
|
|
self.assertEqual('foo/bar/baz', path)
|
|
|
|
def testSafeShortPathGen(self):
|
|
path = pkg_bzl.safe_short_path(File('foo/bar/baz', is_generated=True))
|
|
self.assertEqual('foo/bar/baz', path)
|
|
|
|
def testSafeShortPathExt(self):
|
|
path = pkg_bzl.safe_short_path(File('foo/bar/baz', is_external=True))
|
|
self.assertEqual('external/repo/foo/bar/baz', path)
|
|
|
|
def testSafeShortPathGenExt(self):
|
|
path = pkg_bzl.safe_short_path(
|
|
File('foo/bar/baz', is_generated=True, is_external=True))
|
|
self.assertEqual('external/repo/foo/bar/baz', path)
|
|
|
|
|
|
class ShortPathDirnameTest(unittest.TestCase):
|
|
"""Testing for _short_path_dirname."""
|
|
|
|
def testShortPathDirname(self):
|
|
path = pkg_bzl._short_path_dirname(File('foo/bar/baz'))
|
|
self.assertEqual('foo/bar', path)
|
|
|
|
def testShortPathDirnameGen(self):
|
|
path = pkg_bzl._short_path_dirname(File('foo/bar/baz', is_generated=True))
|
|
self.assertEqual('foo/bar', path)
|
|
|
|
def testShortPathDirnameExt(self):
|
|
path = pkg_bzl._short_path_dirname(File('foo/bar/baz', is_external=True))
|
|
self.assertEqual('external/repo/foo/bar', path)
|
|
|
|
def testShortPathDirnameGenExt(self):
|
|
path = pkg_bzl._short_path_dirname(
|
|
File('foo/bar/baz', is_generated=True, is_external=True))
|
|
self.assertEqual('external/repo/foo/bar', path)
|
|
|
|
def testTopLevel(self):
|
|
path = pkg_bzl._short_path_dirname(File('baz'))
|
|
self.assertEqual('', path)
|
|
|
|
def testTopLevelGen(self):
|
|
path = pkg_bzl._short_path_dirname(File('baz', is_generated=True))
|
|
self.assertEqual('', path)
|
|
|
|
def testTopLevelExt(self):
|
|
path = pkg_bzl._short_path_dirname(File('baz', is_external=True))
|
|
self.assertEqual('external/repo', path)
|
|
|
|
|
|
class DestPathTest(unittest.TestCase):
|
|
"""Testing for _dest_path."""
|
|
|
|
def testDestPath(self):
|
|
path = pkg_bzl.dest_path(File('foo/bar/baz'), 'foo')
|
|
self.assertEqual('/bar/baz', path)
|
|
|
|
def testDestPathGen(self):
|
|
path = pkg_bzl.dest_path(File('foo/bar/baz', is_generated=True), 'foo')
|
|
self.assertEqual('/bar/baz', path)
|
|
|
|
def testDestPathExt(self):
|
|
path = pkg_bzl.dest_path(
|
|
File('foo/bar/baz', is_external=True), 'external/repo/foo')
|
|
self.assertEqual('/bar/baz', path)
|
|
|
|
def testDestPathExtWrong(self):
|
|
path = pkg_bzl.dest_path(File('foo/bar/baz', is_external=True), 'foo')
|
|
self.assertEqual('external/repo/foo/bar/baz', path)
|
|
|
|
def testNoMatch(self):
|
|
path = pkg_bzl.dest_path(File('foo/bar/baz'), 'qux')
|
|
self.assertEqual('foo/bar/baz', path)
|
|
|
|
def testNoStrip(self):
|
|
path = pkg_bzl.dest_path(File('foo/bar/baz'), None)
|
|
self.assertEqual('/baz', path)
|
|
|
|
def testTopLevel(self):
|
|
path = pkg_bzl.dest_path(File('baz'), None)
|
|
self.assertEqual('baz', path)
|
|
|
|
def testPartialDirectoryMatch(self):
|
|
path = pkg_bzl.dest_path(File('foo/bar/baz'), 'fo')
|
|
self.assertEqual('foo/bar/baz', path)
|
|
|
|
def testPartialDirectoryMatchWithDataPath(self):
|
|
path = pkg_bzl.dest_path(File('foo/bar/baz'), 'foo/ba', 'foo')
|
|
self.assertEqual('/bar/baz', path)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|