1*387f9dfdSAndroid Build Coastguard Worker#!/usr/bin/env python3 2*387f9dfdSAndroid Build Coastguard Worker# Copyright (c) PLUMgrid, Inc. 3*387f9dfdSAndroid Build Coastguard Worker# Licensed under the Apache License, Version 2.0 (the "License") 4*387f9dfdSAndroid Build Coastguard Worker 5*387f9dfdSAndroid Build Coastguard Worker# test program for the 'dump_func' method 6*387f9dfdSAndroid Build Coastguard Worker 7*387f9dfdSAndroid Build Coastguard Workerfrom bcc import BPF 8*387f9dfdSAndroid Build Coastguard Workerfrom unittest import main, TestCase 9*387f9dfdSAndroid Build Coastguard Worker 10*387f9dfdSAndroid Build Coastguard Workerclass TestDumpFunc(TestCase): 11*387f9dfdSAndroid Build Coastguard Worker def test_return(self): 12*387f9dfdSAndroid Build Coastguard Worker b = BPF(text=b""" 13*387f9dfdSAndroid Build Coastguard Worker int entry(void) 14*387f9dfdSAndroid Build Coastguard Worker { 15*387f9dfdSAndroid Build Coastguard Worker return 1; 16*387f9dfdSAndroid Build Coastguard Worker }""") 17*387f9dfdSAndroid Build Coastguard Worker 18*387f9dfdSAndroid Build Coastguard Worker self.assertEqual( 19*387f9dfdSAndroid Build Coastguard Worker b"\xb7\x00\x00\x00\x01\x00\x00\x00" + 20*387f9dfdSAndroid Build Coastguard Worker b"\x95\x00\x00\x00\x00\x00\x00\x00", 21*387f9dfdSAndroid Build Coastguard Worker b.dump_func(b"entry")) 22*387f9dfdSAndroid Build Coastguard Worker 23*387f9dfdSAndroid Build Coastguard Workerif __name__ == "__main__": 24*387f9dfdSAndroid Build Coastguard Worker main() 25