Class: ViewModel::TestHelpers::ARVMBuilder

Inherits:
Object
  • Object
show all
Defined in:
lib/view_model/test_helpers/arvm_builder.rb

Defined Under Namespace

Classes: Spec

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(name, model_base: ApplicationRecord, viewmodel_base: ViewModelBase, namespace: Object, spec: nil, &block) ⇒ ARVMBuilder 



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
 
# File 'lib/view_model/test_helpers/arvm_builder.rb', line 36

def initialize(name, model_base: ApplicationRecord, viewmodel_base: ViewModelBase, namespace: Object, spec: nil, &block)
  @model_base = model_base
  @viewmodel_base = viewmodel_base
  @namespace = namespace
  @name = name.to_s.camelize
  @no_viewmodel = false

  if spec
    define_schema(&spec.schema)
    define_model(&spec.model)
    define_viewmodel(&spec.viewmodel)
  else
    instance_eval(&block)
  end

  raise 'Model not created in ARVMBuilder'     unless model
  raise 'Schema not created in ARVMBuilder'    unless model.table_exists?
  raise 'ViewModel not created in ARVMBuilder' unless viewmodel || @no_viewmodel

  # Force the realization of the view model into the library's lookup
  # table. If this doesn't happen the library may have conflicting entries in
  # the deferred table, and will allow viewmodels to leak between tests.
  unless @no_viewmodel || !(@viewmodel < ViewModel::Record)
    resolved = ViewModel::Registry.for_view_name(viewmodel.view_name)
    raise 'Failed to register expected new class!' unless resolved == @viewmodel
  end
end

Instance Attribute Details

#model ⇒ Object (readonly)

Returns the value of attribute model.



4
5
6
 
# File 'lib/view_model/test_helpers/arvm_builder.rb', line 4

def model
  @model
end

#name ⇒ Object (readonly)

Returns the value of attribute name.



4
5
6
 
# File 'lib/view_model/test_helpers/arvm_builder.rb', line 4

def name
  @name
end

#namespace ⇒ Object (readonly)

Returns the value of attribute namespace.



4
5
6
 
# File 'lib/view_model/test_helpers/arvm_builder.rb', line 4

def namespace
  @namespace
end

#viewmodel ⇒ Object (readonly)

Returns the value of attribute viewmodel.



4
5
6
 
# File 'lib/view_model/test_helpers/arvm_builder.rb', line 4

def viewmodel
  @viewmodel
end

Instance Method Details

#teardown ⇒ Object 



64
65
66
67
68
69
70
71
72
73
 
# File 'lib/view_model/test_helpers/arvm_builder.rb', line 64

def teardown
  ActiveRecord::Base.connection.execute("DROP TABLE IF EXISTS #{name.underscore.pluralize} CASCADE")
  namespace.send(:remove_const, name)
  namespace.send(:remove_const, viewmodel_name) if viewmodel

  # prevent cached old class from being used to resolve associations
  if ActiveSupport::VERSION::MAJOR < 7
    ActiveSupport::Dependencies::Reference.clear!
  end
end