blob: 61900baf8c599647be4a1fba70b44d3025555377 (
plain)
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
|
import pytest
import unittest
#import trapd_runtime_pid
import sys
#from /home/ubuntu/HB_Nov5/miss_htbt_service/mod/ import trapd_io
import trapd_runtime_pid
import trapd_io
class test_save_pid(unittest.TestCase):
"""
Test the save_pid mod
"""
def test_correct_usage(self):
"""
Test that attempt to create pid file in standard location works
"""
result = trapd_runtime_pid.save_pid('/tmp/snmptrap_test_pid_file')
self.assertEqual(result, True)
def test_missing_directory(self):
"""
Test that attempt to create pid file in missing dir fails
"""
result = trapd_runtime_pid.save_pid('/bogus/directory/for/snmptrap_test_pid_file')
self.assertEqual(result, False)
class test_rm_pid(unittest.TestCase):
"""
Test the rm_pid mod
"""
def test_correct_usage(self):
"""
Test that attempt to remove pid file in standard location works
"""
# must create it before removing it
result = trapd_runtime_pid.save_pid('/tmp/snmptrap_test_pid_file')
self.assertEqual(result, True)
result = trapd_runtime_pid.rm_pid('/tmp/snmptrap_test_pid_file')
self.assertEqual(result, True)
def test_missing_file(self):
"""
Test that attempt to rm non-existent pid file fails
"""
result = trapd_runtime_pid.rm_pid('/tmp/snmptrap_test_pid_file_9999')
self.assertEqual(result, False)
#if __name__ == '__main__':
# unittest.main()
|
