fune/taskcluster/taskgraph/test/test_util_treeherder.py
Tom Prince 7b13402d7e Bug 1456234: [taskgraph]: Add a helper for adding a suffix to a possibly grouped treeherder symbol; r=dustin.
Differential Revision: https://phabricator.services.mozilla.com/D1286

--HG--
extra : rebase_source : 3cb1080e5a94e3e9e5b5a3f6909b0e8d636de09c
extra : source : 4a182c3447fa01c75fa8a231e067bebd83683da9
2018-05-15 13:00:18 -06:00

34 lines
1 KiB
Python

# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
from __future__ import absolute_import, print_function, unicode_literals
import unittest
from taskgraph.util.treeherder import split_symbol, join_symbol, add_suffix
from mozunit import main
class TestSymbols(unittest.TestCase):
def test_split_no_group(self):
self.assertEqual(split_symbol('xy'), ('?', 'xy'))
def test_split_with_group(self):
self.assertEqual(split_symbol('ab(xy)'), ('ab', 'xy'))
def test_join_no_group(self):
self.assertEqual(join_symbol('?', 'xy'), 'xy')
def test_join_with_group(self):
self.assertEqual(join_symbol('ab', 'xy'), 'ab(xy)')
def test_add_suffix_no_group(self):
self.assertEqual(add_suffix('xy', 1), 'xy1')
def test_add_suffix_with_group(self):
self.assertEqual(add_suffix('ab(xy)', 1), 'ab(xy1)')
if __name__ == '__main__':
main()