forked from iovisor/bcc
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathtest_queuestack.py
More file actions
executable file
·81 lines (62 loc) · 1.9 KB
/
Copy pathtest_queuestack.py
File metadata and controls
executable file
·81 lines (62 loc) · 1.9 KB
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
#!/usr/bin/env python
# Copyright (c) PLUMgrid, Inc.
# Licensed under the Apache License, Version 2.0 (the "License")
from bcc import BPF
import os
import distutils.version
import ctypes as ct
import random
import time
import subprocess
from unittest import main, TestCase, skipUnless
def kernel_version_ge(major, minor):
# True if running kernel is >= X.Y
version = distutils.version.LooseVersion(os.uname()[2]).version
if version[0] > major:
return True
if version[0] < major:
return False
if minor and version[1] < minor:
return False
return True
class TestQueueStack(TestCase):
@skipUnless(kernel_version_ge(4,20), "requires kernel >= 4.20")
def test_stack(self):
text = """
BPF_STACK(stack, u64, 10);
"""
b = BPF(text=text)
stack = b['stack']
for i in range(10):
stack.push(ct.c_uint64(i))
with self.assertRaises(Exception):
stack.push(ct.c_uint(10))
assert stack.peek().value == 9
for i in reversed(range(10)):
assert stack.pop().value == i
with self.assertRaises(KeyError):
stack.peek()
with self.assertRaises(KeyError):
stack.pop()
b.cleanup()
@skipUnless(kernel_version_ge(4,20), "requires kernel >= 4.20")
def test_queue(self):
text = """
BPF_QUEUE(queue, u64, 10);
"""
b = BPF(text=text)
queue = b['queue']
for i in range(10):
queue.push(ct.c_uint64(i))
with self.assertRaises(Exception):
queue.push(ct.c_uint(10))
assert queue.peek().value == 0
for i in range(10):
assert queue.pop().value == i
with self.assertRaises(KeyError):
queue.peek()
with self.assertRaises(KeyError):
queue.pop()
b.cleanup()
if __name__ == "__main__":
main()