tests/testirker.py

   1 # -*- coding: utf-8 -*-
   2 # vim:fileencoding=utf-8 ai ts=4 sts=4 et sw=4
   3 # Copyright 2012 Neil Muller
   4 # GPL 2+ - see COPYING for details
   5 from unittest import TestLoader
   6
   7 from bzrlib import (
   8     config,
   9     tests,
  10     )
  11 from bzrlib.bzrdir import BzrDir
  12 from bzrlib.tests import TestCaseInTempDir
  13 from bzrlib.plugins.bzrirker.irkerhook import IrkerSender
  14
  15
  16 def test_suite():
  17     return TestLoader().loadTestsFromName(__name__)
  18
  19 sample_config=("[DEFAULT]\n"
  20         "irker_project=bzrirker\n"
  21         "irker_channels=irc://chat.example.test/commits\n")
  22
  23 colour_config=("[DEFAULT]\n"
  24         "irker_project=bzrirker\n"
  25         "irker_colours=mIRC\n"
  26         "irker_channels=irc://chat.example.test/commits\n")
  27
  28
  29 class TestConfigured(TestCaseInTempDir):
  30
  31     def test_no_colours(self):
  32         irker = self.get_irker()
  33         self.assertEqual({
  34             'bold': '', 'green': '', 'blue': '', 'red': '',
  35             'yellow': '', 'brown': '', 'magenta': '', 'cyan': '',
  36             'reset': ''}, irker.colours())
  37
  38     def test_colours(self):
  39         irker = self.get_irker(colour_config)
  40         self.assertEqual({
  41             'bold': '\x02', 'green': '\x0303', 'blue': '\x0302',
  42             'red': '\x0305', 'yellow': '\x0307', 'brown': '\x0305',
  43             'magenta': '\x0306', 'cyan': '\x0310', 'reset': '\x0F'},
  44             irker.colours())
  45
  46     def test_message(self):
  47         irker = self.get_irker()
  48         self.assertEqual(
  49                 'bzrirker: Sample <john@example.test> work * 1 /  : foo bar ',
  50                 irker._format())
  51
  52     def test_message_colour(self):
  53         irker = self.get_irker(colour_config)
  54         self.assertEqual(
  55                 '\x02bzrirker:\x0f \x0303Sample <john@example.test>\x0f'
  56                 ' work * \x021\x0f /  \x02\x0f: foo bar ', irker._format())
  57
  58
  59     def get_irker(self, text=sample_config):
  60         my_config = config.MemoryStack(text)
  61         self.branch = BzrDir.create_branch_convenience('.')
  62         tree = self.branch.bzrdir.open_workingtree()
  63         tree.commit('foo bar\nfuzzy\nwuzzy', rev_id='A',
  64                 allow_pointless=True,
  65                 timestamp=1,
  66                 timezone=0,
  67                 committer="Sample <john@example.test>",
  68                 )
  69         irker = IrkerSender(self.branch, 'A', my_config)
  70         # We're not calling send, so do the setup stuff here
  71         irker._setup_revision_and_revno()
  72         return irker
  73
  74