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