blob: 9a843c2c6fa4d964fcaa0c7b5156ff1b4270f9e5 (
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
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
|
require 'json'
#
#
# Base class for custodian protocol-tests
#
# Each subclass will register themselves, via the call
# to 'register_test_type'.
#
# This class is a factory that will return the correct
# derived class for a given line from our configuration
# file.
#
# TODO: We also wish to create from json.
#
class ProtocolTest
#
# The subclasses we have.
#
@@subclasses = { }
#
# Create a test-type object given a line of text from our parser.
#
# The line will be like "target must run tcp|ssh|ftp|smtp .."
#
#
def self.create( line )
if ( line =~ /must\s+run\s+(\S+)(\s+|\.|$)/ )
test_type = $1.dup
test_type.chomp!( "." )
c = @@subclasses[test_type]
if c
c.new( line )
else
raise ArgumentError, "Bad test type: #{test_type}"
end
else
raise "Unknown line given - Failed to instantiate a suitable protocol-test."
end
end
#
# Register a new test type - this must be called by our derived classes
#
def self.register_test_type name
@@subclasses[name] = self
end
#
# Return an array of test-types we know about
#
# i.e. Derived classes that have registered themselves.
#
#
def types
@@subclasses
end
end
|
