define relationship between company and licenses and return active licenses.

2 files changed, 9 insertions, 1 deletions

diff --git a/app/models/company.rb b/app/models/company.rb
index bff819b..0959d4d 100644
--- a/app/models/company.rb
+++ b/app/models/company.rb
@@ -1,2 +1,8 @@
-class Company
+class Company < ActiveRecord::Base
+  has_many :licenses
+
+  def status(status)
+    today = DateTime.now
+    licenses.where('issued_at < ? AND expired_at > ?', today, today)
+  end
 end
diff --git a/app/models/license.rb b/app/models/license.rb
index 8eb7612..d1ce61d 100644
--- a/app/models/license.rb
+++ b/app/models/license.rb
@@ -1,4 +1,6 @@
 class License < ActiveRecord::Base
+  belongs_to :company
+
   def self.most_recent(page: 1, per_page: 10)
     offset = (page - 1) * per_page
     offset = offset >= 0 ? offset : 0