aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorTeddy Wing2017-06-15 12:11:55 +0200
committerTeddy Wing2017-06-15 12:11:55 +0200
commit34ee3f25dd9e903e27bbbbcdf85ae2803a869c76 (patch)
treea7c1aec6bac68fbbe85b1ba32b3d3b1789cd9542
parent9a2be7854d2149f812f90dd9f42f680199e573d4 (diff)
downloadchouette-core-34ee3f25dd9e903e27bbbbcdf85ae2803a869c76.tar.bz2
TableBuilder spec: Decorate companies collection
In order to enable access to `CompanyDecorator#action_links`, wrap the companies collection passed to the table builder in the test in the decorator. Specify the class that `CompanyDecorator` decorates explicitly. Since `Chouette::Company` is namespaced, Draper can't infer the model from the decorator class name. Refs #3479
-rw-r--r--app/decorators/company_decorator.rb2
-rw-r--r--spec/helpers/table_builder_helper_spec.rb5
2 files changed, 7 insertions, 0 deletions
diff --git a/app/decorators/company_decorator.rb b/app/decorators/company_decorator.rb
index 7d14f8c8b..5a2fdd1c0 100644
--- a/app/decorators/company_decorator.rb
+++ b/app/decorators/company_decorator.rb
@@ -1,4 +1,6 @@
class CompanyDecorator < Draper::Decorator
+ decorates Chouette::Company
+
delegate_all
def self.collection_decorator_class
diff --git a/spec/helpers/table_builder_helper_spec.rb b/spec/helpers/table_builder_helper_spec.rb
index c5cbd972d..89660dd56 100644
--- a/spec/helpers/table_builder_helper_spec.rb
+++ b/spec/helpers/table_builder_helper_spec.rb
@@ -200,6 +200,11 @@ describe TableBuilderHelper, type: :helper do
referential_id: referential.id
})
+ companies = ModelDecorator.decorate(
+ companies,
+ with: CompanyDecorator
+ )
+
expected = <<-HTML
<table class="table has-search">
<thead>