Module: Pickle::Session

Defined in:
lib/pickle/session.rb,
lib/pickle/session/parser.rb

Defined Under Namespace

Modules: Parser, PickleParserMethods Classes: ModelNotFoundError, ModelNotKnownError

Class Method Summary collapse

Instance Method Summary collapse

Class Method Details

.extended(world_object) ⇒ Object 



38
39
40
 
# File 'lib/pickle/session.rb', line 38

def extended(world_object)
  proxy_to_pickle_parser(class << world_object; self; end) # metaclass is not 2.1 compatible
end

.included(world_class) ⇒ Object 



34
35
36
 
# File 'lib/pickle/session.rb', line 34

def included(world_class)
  proxy_to_pickle_parser(world_class)
end

Instance Method Details

#build_model(pickle_ref, fields = nil) ⇒ Object 



56
57
58
 
# File 'lib/pickle/session.rb', line 56

def build_model(pickle_ref, fields = nil)
  create_or_build_model(:build, 1, pickle_ref, fields)
end

#build_models(count, pickle_ref, fields = nil) ⇒ Object 



60
61
62
 
# File 'lib/pickle/session.rb', line 60

def build_models(count, pickle_ref, fields = nil)
  create_or_build_model(:build, count, pickle_ref, fields)
end

#create_model(pickle_ref, fields = nil) ⇒ Object 



48
49
50
 
# File 'lib/pickle/session.rb', line 48

def create_model(pickle_ref, fields = nil)
  create_or_build_model(:create, 1, pickle_ref, fields)
end

#create_models(count, pickle_ref, fields = nil) ⇒ Object 



52
53
54
 
# File 'lib/pickle/session.rb', line 52

def create_models(count, pickle_ref, fields = nil)
  create_or_build_model(:create, count, pickle_ref, fields)
end

#create_models_from_table(plural_factory, table) ⇒ Object

if a column exists in the table which matches the singular factory name, this is used as the pickle ref



65
66
67
68
69
70
71
 
# File 'lib/pickle/session.rb', line 65

def create_models_from_table(plural_factory, table)
  factory = plural_factory.singularize
  table.hashes.map do |hash|
    pickle_ref = factory + (hash[factory] ? " \"#{hash.delete(factory)}\"" : "")
    create_model(pickle_ref, hash)
  end
end

#created_model(name) ⇒ Object

return the original model stored by create_model or find_model



114
115
116
117
118
119
120
121
122
123
124
 
# File 'lib/pickle/session.rb', line 114

def created_model(name)
  factory, name_or_index = *parse_model(name)

  if name_or_index.blank?
    models_by_index(factory).last
  elsif name_or_index.is_a?(Integer)
    models_by_index(factory)[name_or_index]
  else
    models_by_name(factory)[name_or_index] or raise ModelNotKnownError, name
  end
end

#created_model!(name) ⇒ Object

like created_model, but raise an error if it can’t be found



149
150
151
 
# File 'lib/pickle/session.rb', line 149

def created_model!(name)
  created_model(name) or raise ModelNotKnownError, name
end

#created_model?(name) ⇒ Boolean

predicate version which raises no errors

Returns:

  • (Boolean) 


127
128
129
 
# File 'lib/pickle/session.rb', line 127

def created_model?(name)
  (created_model(name) rescue nil) ? true : false
end

#created_models(factory) ⇒ Object

return all original models of specified type



154
155
156
 
# File 'lib/pickle/session.rb', line 154

def created_models(factory)
  models_by_index(factory)
end

#find_model(a_model_name, fields = nil) ⇒ Object

Raises:

  • (ArgumentError) 


73
74
75
76
77
78
79
80
81
82
83
84
85
86
 
# File 'lib/pickle/session.rb', line 73

def find_model(a_model_name, fields = nil)
  factory, name = *parse_model(a_model_name)

  raise ArgumentError, "Can't find a model with an ordinal (e.g. 1st user)" if name.is_a?(Integer)

  model_class = pickle_config.factories[factory].klass
  fields      = fields.is_a?(Hash) ? parse_hash(fields) : parse_fields(fields)
  conditions  = convert_models_to_attributes(model_class, fields)
  record      = Pickle::Adapter.find_first_model(model_class, conditions)

  store_model(factory, name, record) if record

  record
end

#find_model!(name, fields = nil) ⇒ Object 



88
89
90
 
# File 'lib/pickle/session.rb', line 88

def find_model!(name, fields = nil)
  find_model(name, fields) or raise ModelNotFoundError, "Can't find #{name}#{" with #{fields}" if fields.present?} from the orm in this scenario"
end

#find_models(factory, fields = nil) ⇒ Object 



92
93
94
95
96
97
98
99
100
101
102
 
# File 'lib/pickle/session.rb', line 92

def find_models(factory, fields = nil)
  factory = pickle_parser.canonical(factory)

  models_by_index(factory).clear

  model_class = pickle_config.factories[factory].klass
  conditions  = convert_models_to_attributes(model_class, parse_fields(fields))
  records     = Pickle::Adapter.find_all_models(model_class, conditions)

  records.each {|record| store_model(factory, nil, record)}
end

#find_models_from_table(plural_factory, table) ⇒ Object

if a column exists in the table which matches the singular factory name, this is used as the pickle ref



105
106
107
108
109
110
111
 
# File 'lib/pickle/session.rb', line 105

def find_models_from_table(plural_factory, table)
  factory = plural_factory.singularize
  table.hashes.map do |hash|
    pickle_ref = factory + (hash[factory] ? " \"#{hash.delete(factory)}\"" : "")
    find_model(pickle_ref, hash)
  end
end

#model(name) ⇒ Object

return a newly selected model



132
133
134
135
136
 
# File 'lib/pickle/session.rb', line 132

def model(name)
  model = created_model(name)
  return nil unless model
  Pickle::Adapter.get_model(model.class, model.id)
end

#model!(name) ⇒ Object

like model, but raise an error if it can’t be found



144
145
146
 
# File 'lib/pickle/session.rb', line 144

def model!(name)
  model(name) or raise ModelNotKnownError, name
end

#model?(name) ⇒ Boolean

predicate version which raises no errors

Returns:

  • (Boolean) 


139
140
141
 
# File 'lib/pickle/session.rb', line 139

def model?(name)
  (model(name) rescue nil) ? true : false
end

#models(factory) ⇒ Object

return all models of specified type (freshly selected from the database)



159
160
161
162
163
 
# File 'lib/pickle/session.rb', line 159

def models(factory)
  created_models(factory).map do |model|
    Pickle::Adapter.get_model(model.class, model.id)
  end
end