vendor/gems/composite_primary_keys-2.2.2/test/test_ids.rb

   1 require 'abstract_unit'
   2 require 'fixtures/reference_type'
   3 require 'fixtures/reference_code'
   4
   5 class TestIds < Test::Unit::TestCase
   6   fixtures :reference_types, :reference_codes
   7
   8   CLASSES = {
   9     :single => {
  10       :class => ReferenceType,
  11       :primary_keys => [:reference_type_id],
  12     },
  13     :dual   => {
  14       :class => ReferenceCode,
  15       :primary_keys => [:reference_type_id, :reference_code],
  16     },
  17     :dual_strs   => {
  18       :class => ReferenceCode,
  19       :primary_keys => ['reference_type_id', 'reference_code'],
  20     },
  21   }
  22
  23   def setup
  24     self.class.classes = CLASSES
  25   end
  26
  27   def test_id
  28     testing_with do
  29       assert_equal @first.id, @first.ids if composite?
  30     end
  31   end
  32
  33   def test_id_to_s
  34     testing_with do
  35       assert_equal first_id_str, @first.id.to_s
  36       assert_equal first_id_str, "#{@first.id}"
  37     end
  38   end
  39
  40   def test_ids_to_s
  41     testing_with do
  42       order = @klass.primary_key.is_a?(String) ? @klass.primary_key : @klass.primary_key.join(',')
  43       to_test = @klass.find(:all, :order => order)[0..1].map(&:id)
  44       assert_equal '(1,1),(1,2)', @klass.ids_to_s(to_test) if @key_test == :dual
  45       assert_equal '1,1;1,2', @klass.ids_to_s(to_test, ',', ';', '', '') if @key_test == :dual
  46     end
  47   end
  48
  49   def test_composite_where_clause
  50     testing_with do
  51       where = 'reference_codes.reference_type_id=1 AND reference_codes.reference_code=2) OR (reference_codes.reference_type_id=2 AND reference_codes.reference_code=2'
  52       assert_equal(where, @klass.composite_where_clause([[1, 2], [2, 2]])) if @key_test == :dual
  53     end
  54   end
  55
  56   def test_set_ids_string
  57     testing_with do
  58       array = @primary_keys.collect {|key| 5}
  59       expected = composite? ? array.to_composite_keys : array.first
  60       @first.id = expected.to_s
  61       assert_equal expected, @first.id
  62     end
  63   end
  64
  65   def test_set_ids_array
  66     testing_with do
  67       array = @primary_keys.collect {|key| 5}
  68       expected = composite? ? array.to_composite_keys : array.first
  69       @first.id = expected
  70       assert_equal expected, @first.id
  71     end
  72   end
  73
  74   def test_set_ids_comp
  75     testing_with do
  76       array = @primary_keys.collect {|key| 5}
  77       expected = composite? ? array.to_composite_keys : array.first
  78       @first.id = expected
  79       assert_equal expected, @first.id
  80     end
  81   end
  82
  83   def test_primary_keys
  84     testing_with do
  85       if composite?
  86         assert_not_nil @klass.primary_keys
  87         assert_equal @primary_keys.map {|key| key.to_sym}, @klass.primary_keys
  88         assert_equal @klass.primary_keys, @klass.primary_key
  89       else
  90         assert_not_nil @klass.primary_key
  91         assert_equal @primary_keys, [@klass.primary_key.to_sym]
  92       end
  93       assert_equal @primary_keys.join(','), @klass.primary_key.to_s
  94       # Need a :primary_keys should be Array with to_s overridden
  95     end
  96   end
  97 end