Source Code for Module ioids.testioids

 1  """ 
 2  Test procedures for IOIDS. 
 3   
 4  Inter-Organisational Intrusion Detection System (IOIDS) 
 5   
 6  @author: Michael Pilgermann 
 7  @contact: mailto:mpilgerm@glam.ac.uk 
 8  @license: GPL (General Public License) 
 9  """ 
10   
11 -def test(): 
12  ##    testWrapper() 
13  ##    testSelect() 
14  ##    testDicts() 
15      testWrapper2() 
16       
17 -def testWrapper(): 
18      from messagewrapper import getXMLDBWrapper 
19      from dbconnector import OPERATOR_GREATER_THEN 
20       
21  ##    print getXMLDBWrapper().wrapSelect('event', 'all', [['oid',OPERATOR_GREATER_THEN,'30608']]) 
22      xml = "<RELATIONS command='SELECT_RESULTS' >" + \ 
23          "<REL RESULTS_ID='1'>" + \ 
24          "<REL name='table1'>" + \ 
25          "<ATT name='x1'>128</ATT><ATT name='table1_id'>1</ATT><ATT name='table2_id'>1</ATT>" + \ 
26          "</REL>" + \ 
27          "<REL name='table1'>" + \ 
28          "<ATT name='x1'>12</ATT><ATT name='table1_id'>2</ATT><ATT name='table2_id'>1</ATT>" + \ 
29          "</REL>" + \ 
30          "<REL name='TOTAL_RECORDS'>2</REL>" + \ 
31          "</REL>" + \ 
32          "<REL name='TOTAL_RESULTS'>1</REL>" + \ 
33          "</RELATIONS>" 
34       
35      print "Number of sets: %d\n%s" %(getXMLDBWrapper().parseSelectReply(xml)) 
36   
37 -def testSelect(): 
38      from dbconnector import getDBConnector 
39      getDBConnector().connect() 
40       
41      print getDBConnector().getEvents() 
42      print "\n", getDBConnector().getEventsFromEventID(3) 
43       
44      print "\n", getDBConnector().getIoidsEvents() 
45      print "\n", getDBConnector().getIoidsEventsFromEventID(3) 
46   
47 -def testDicts(): 
48      from dbconnector import getDBConnector 
49      getDBConnector().connect() 
50      from messagewrapper import getXMLDBWrapper 
51  ##    lists = getDBConnector().getEvent('5') 
52      lists = getDBConnector().getIoidsEvent('2') 
53  ##    print lists 
54      xml = getXMLDBWrapper().wrapInsert(lists[0], lists[1], lists[2]) 
55      print xml 
56      getDBConnector().disconnect() 
57       
58 -def testWrapper2(): 
59      from messagewrapper import getIoidsMessageWrapper 
60      xml =  getIoidsMessageWrapper().wrapKnowledgeRequestMessage([['timestamp', 'greater_than','2005-06-05'], ['source', 'equals', 'M003']]) 
61      print xml 
62      print getIoidsMessageWrapper().parseKnowledgeRequestMessage(xml) 
63       
64  if __name__ == "__main__": 
65      test() 
66