From 34ee3f25dd9e903e27bbbbcdf85ae2803a869c76 Mon Sep 17 00:00:00 2001
From: Teddy Wing
Date: Thu, 15 Jun 2017 12:11:55 +0200
Subject: 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
---
 app/decorators/company_decorator.rb | 2 ++
 1 file changed, 2 insertions(+)

(limited to 'app')

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
-- 
cgit v1.2.3