FilesExpand file tree

 master

/

test_connect.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

124 lines (101 loc) · 4.26 KB

 master

/

test_connect.py

Copy path

Top

File metadata and controls

• Code
• Blame

124 lines (101 loc) · 4.26 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

82

83

84

85

86

87

88

89

90

91

92

93

94

95

96

97

98

99

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

"""Tests for kernel connection utilities

Authors

-------

* MinRK

"""

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

# Copyright (c) 2013, the IPython Development Team.

#

# Distributed under the terms of the Modified BSD License.

#

# The full license is in the file COPYING.txt, distributed with this software.

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

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

# Imports

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

import json

import os

import nose.tools as nt

from IPython.config import Config

from IPython.consoleapp import IPythonConsoleApp

from IPython.core.application import BaseIPythonApplication

from IPython.utils.tempdir import TemporaryDirectory, TemporaryWorkingDirectory

from IPython.utils.py3compat import str_to_bytes

from IPython.kernel import connect

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

# Classes and functions

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

class DummyConsoleApp(BaseIPythonApplication, IPythonConsoleApp):

def initialize(self, argv=[]):

BaseIPythonApplication.initialize(self, argv=argv)

self.init_connection_file()

sample_info = dict(ip='1.2.3.4', transport='ipc',

shell_port=1, hb_port=2, iopub_port=3, stdin_port=4, control_port=5,

key=b'abc123', signature_scheme='hmac-md5',

)

def test_write_connection_file():

with TemporaryDirectory() as d:

cf = os.path.join(d, 'kernel.json')

connect.write_connection_file(cf, **sample_info)

nt.assert_true(os.path.exists(cf))

with open(cf, 'r') as f:

info = json.load(f)

info['key'] = str_to_bytes(info['key'])

nt.assert_equal(info, sample_info)

def test_app_load_connection_file():

"""test `ipython console --existing` loads a connection file"""

with TemporaryDirectory() as d:

cf = os.path.join(d, 'kernel.json')

connect.write_connection_file(cf, **sample_info)

app = DummyConsoleApp(connection_file=cf)

app.initialize(argv=[])

for attr, expected in sample_info.items():

if attr in ('key', 'signature_scheme'):

continue

value = getattr(app, attr)

nt.assert_equal(value, expected, "app.%s = %s != %s" % (attr, value, expected))

def test_get_connection_file():

cfg = Config()

with TemporaryWorkingDirectory() as d:

cfg.ProfileDir.location = d

cf = 'kernel.json'

app = DummyConsoleApp(config=cfg, connection_file=cf)

app.initialize(argv=[])

profile_cf = os.path.join(app.profile_dir.location, 'security', cf)

nt.assert_equal(profile_cf, app.connection_file)

with open(profile_cf, 'w') as f:

f.write("{}")

nt.assert_true(os.path.exists(profile_cf))

nt.assert_equal(connect.get_connection_file(app), profile_cf)

app.connection_file = cf

nt.assert_equal(connect.get_connection_file(app), profile_cf)

def test_find_connection_file():

cfg = Config()

with TemporaryDirectory() as d:

cfg.ProfileDir.location = d

cf = 'kernel.json'

app = DummyConsoleApp(config=cfg, connection_file=cf)

app.initialize(argv=[])

BaseIPythonApplication._instance = app

profile_cf = os.path.join(app.profile_dir.location, 'security', cf)

with open(profile_cf, 'w') as f:

f.write("{}")

for query in (

'kernel.json',

'kern*',

'*ernel*',

'k*',

):

nt.assert_equal(connect.find_connection_file(query), profile_cf)

BaseIPythonApplication._instance = None

def test_get_connection_info():

with TemporaryDirectory() as d:

cf = os.path.join(d, 'kernel.json')

connect.write_connection_file(cf, **sample_info)

json_info = connect.get_connection_info(cf)

info = connect.get_connection_info(cf, unpack=True)

nt.assert_equal(type(json_info), type(""))

nt.assert_equal(info, sample_info)

info2 = json.loads(json_info)

info2['key'] = str_to_bytes(info2['key'])

nt.assert_equal(info2, sample_info)