From 1260d03e638e71309f69bfd8be6f23c189fdc8d0 Mon Sep 17 00:00:00 2001 From: AntonioTerceiro Date: Mon, 24 Sep 2007 18:17:17 +0000 Subject: [PATCH] ActionItem70: creating an actual relationship from environment to its categories --- app/models/environment.rb | 2 ++ test/unit/environment_test.rb | 14 ++++++++++++++ 2 files changed, 16 insertions(+), 0 deletions(-) diff --git a/app/models/environment.rb b/app/models/environment.rb index 237d89a..39b2d0d 100644 --- a/app/models/environment.rb +++ b/app/models/environment.rb @@ -22,6 +22,8 @@ class Environment < ActiveRecord::Base has_many :domains, :as => :owner has_many :profiles + has_many :categories + # ################################################# # Attributes # ################################################# diff --git a/test/unit/environment_test.rb b/test/unit/environment_test.rb index 3ac9a1b..f5cdd4f 100644 --- a/test/unit/environment_test.rb +++ b/test/unit/environment_test.rb @@ -101,12 +101,26 @@ class EnvironmentTest < Test::Unit::TestCase env = Environment.create!(:name => 'a test environment') cat1 = Category.create!(:name => 'first category', :environment_id => env.id) cat2 = Category.create!(:name => 'second category', :environment_id => env.id) + subcat = Category.create!(:name => 'child category', :environment_id => env.id, :parent_id => cat2.id) cats = env.top_level_categories assert_equal 2, cats.size assert cats.include?(cat1) assert cats.include?(cat2) + assert !cats.include?(subcat) + end + def test_should_list_all_categories + env = Environment.create!(:name => 'a test environment') + cat1 = Category.create!(:name => 'first category', :environment_id => env.id) + cat2 = Category.create!(:name => 'second category', :environment_id => env.id) + subcat = Category.create!(:name => 'child category', :environment_id => env.id, :parent_id => cat2.id) + + cats = env.categories + assert_equal 3, cats.size + assert cats.include?(cat1) + assert cats.include?(cat2) + assert cats.include?(subcat) end end -- libgit2 0.21.2