FilesExpand file tree

 master

/

test_hooks.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

81 lines (66 loc) · 2.43 KB

 master

/

test_hooks.py

Copy path

Top

File metadata and controls

• Code
• Blame

81 lines (66 loc) · 2.43 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

# -*- coding: utf-8 -*-

"""Tests for CommandChainDispatcher."""

from __future__ import absolute_import

#-----------------------------------------------------------------------------

# Imports

#-----------------------------------------------------------------------------

import nose.tools as nt

from IPython.core.error import TryNext

from IPython.core.hooks import CommandChainDispatcher

#-----------------------------------------------------------------------------

# Local utilities

#-----------------------------------------------------------------------------

# Define two classes, one which succeeds and one which raises TryNext. Each

# sets the attribute `called` to True when it is called.

class Okay(object):

def __init__(self, message):

self.message = message

self.called = False

def __call__(self):

self.called = True

return self.message

class Fail(object):

def __init__(self, message):

self.message = message

self.called = False

def __call__(self):

self.called = True

raise TryNext(self.message)

#-----------------------------------------------------------------------------

# Test functions

#-----------------------------------------------------------------------------

def test_command_chain_dispatcher_ff():

"""Test two failing hooks"""

fail1 = Fail(u'fail1')

fail2 = Fail(u'fail2')

dp = CommandChainDispatcher([(0, fail1),

(10, fail2)])

try:

dp()

except TryNext as e:

nt.assert_equal(str(e), u'fail2')

else:

assert False, "Expected exception was not raised."

nt.assert_true(fail1.called)

nt.assert_true(fail2.called)

def test_command_chain_dispatcher_fofo():

"""Test a mixture of failing and succeeding hooks."""

fail1 = Fail(u'fail1')

fail2 = Fail(u'fail2')

okay1 = Okay(u'okay1')

okay2 = Okay(u'okay2')

dp = CommandChainDispatcher([(0, fail1),

# (5, okay1), # add this later

(10, fail2),

(15, okay2)])

dp.add(okay1, 5)

nt.assert_equal(dp(), u'okay1')

nt.assert_true(fail1.called)

nt.assert_true(okay1.called)

nt.assert_false(fail2.called)

nt.assert_false(okay2.called)

def test_command_chain_dispatcher_eq_priority():

okay1 = Okay(u'okay1')

okay2 = Okay(u'okay2')

dp = CommandChainDispatcher([(1, okay1)])

dp.add(okay2, 1)