31 lines
492 B
Ruby
31 lines
492 B
Ruby
class MockFactory
|
|
attr_reader :declarations, :traits, :callbacks
|
|
|
|
def initialize
|
|
@declarations = []
|
|
@traits = []
|
|
@callbacks = []
|
|
@to_create = nil
|
|
end
|
|
|
|
def to_create(&block)
|
|
if block_given?
|
|
@to_create = block
|
|
else
|
|
@to_create
|
|
end
|
|
end
|
|
|
|
def declare_attribute(declaration)
|
|
@declarations << declaration
|
|
end
|
|
|
|
def add_callback(callback)
|
|
@callbacks << callback
|
|
end
|
|
|
|
def define_trait(trait)
|
|
@traits << trait
|
|
end
|
|
end
|