composite_primary_keys 0.1.3

data/lib/composite_primary_keys/version.rb

@@ -0,0 +1,9 @@
+module CompositePrimayKeys
+  module VERSION #:nodoc:
+    MAJOR = 0
+    MINOR = 1
+    TINY  = 3
+
+    STRING = [MAJOR, MINOR, TINY].join('.')
+  end
+end

data/test/abstract_unit.rb

@@ -0,0 +1,69 @@
+$:.unshift(File.dirname(__FILE__) + '/../lib')
+$:.unshift(File.dirname(__FILE__) + '/../../activesupport/lib')
+
+require 'test/unit'
+require 'active_record'
+require 'active_record/fixtures'
+require 'active_support/binding_of_caller'
+require 'active_support/breakpoint'
+require 'connection'
+require 'composite_primary_keys'
+require 'composite_primary_keys/fixtures'
+
+QUOTED_TYPE = ActiveRecord::Base.connection.quote_column_name('type') unless Object.const_defined?(:QUOTED_TYPE)
+
+class Test::Unit::TestCase #:nodoc:
+  self.fixture_path = File.dirname(__FILE__) + "/fixtures/"
+  self.use_instantiated_fixtures = false
+  self.use_transactional_fixtures = (ENV['AR_NO_TX_FIXTURES'] != "yes")
+
+  def create_fixtures(*table_names, &block)
+    Fixtures.create_fixtures(File.dirname(__FILE__) + "/fixtures/", table_names, {}, &block)
+  end
+  
+  def assert_date_from_db(expected, actual, message = nil)
+    # SQL Server doesn't have a separate column type just for dates, 
+    # so the time is in the string and incorrectly formatted
+    if current_adapter?(:SQLServerAdapter)
+      assert_equal expected.strftime("%Y/%m/%d 00:00:00"), actual.strftime("%Y/%m/%d 00:00:00")
+    elsif current_adapter?(:SybaseAdapter)
+      assert_equal expected.to_s, actual.to_date.to_s, message
+    else
+      assert_equal expected.to_s, actual.to_s, message
+    end
+  end
+
+  def assert_queries(num = 1)
+    ActiveRecord::Base.connection.class.class_eval do
+      self.query_count = 0
+      alias_method :execute, :execute_with_query_counting
+    end
+    yield
+  ensure
+    ActiveRecord::Base.connection.class.class_eval do
+      alias_method :execute, :execute_without_query_counting
+    end
+    assert_equal num, ActiveRecord::Base.connection.query_count, "#{ActiveRecord::Base.connection.query_count} instead of #{num} queries were executed."
+  end
+
+  def assert_no_queries(&block)
+    assert_queries(0, &block)
+  end
+end
+
+def current_adapter?(type)
+  ActiveRecord::ConnectionAdapters.const_defined?(type) &&
+    ActiveRecord::Base.connection.instance_of?(ActiveRecord::ConnectionAdapters.const_get(type))
+end
+
+ActiveRecord::Base.connection.class.class_eval do
+  cattr_accessor :query_count
+  alias_method :execute_without_query_counting, :execute
+  def execute_with_query_counting(sql, name = nil)
+    self.query_count += 1
+    execute_without_query_counting(sql, name)
+  end
+end
+
+#ActiveRecord::Base.logger = Logger.new(STDOUT)
+#ActiveRecord::Base.colorize_logging = false

data/test/connections/native_mysql/connection.rb

@@ -0,0 +1,13 @@
+print "Using native MySQL\n"
+require 'logger'
+
+ActiveRecord::Base.logger = Logger.new("debug.log")
+
+db1 = 'composite_primary_keys_unittest'
+
+ActiveRecord::Base.establish_connection(
+  :adapter  => "mysql",
+  :username => "root",
+  :encoding => "utf8",
+  :database => db1
+)

data/test/dummy_test.rb

@@ -0,0 +1,8 @@
+require 'abstract_unit'
+
+class DummyTest < Test::Unit::TestCase
+
+  def test_truth
+    assert true
+  end
+end

data/test/find_test.rb

@@ -0,0 +1,88 @@
+require 'abstract_unit'
+require 'fixtures/reference_type'
+require 'fixtures/reference_code'
+
+# Testing the find action on composite ActiveRecords with two primary keys
+class FindTest < Test::Unit::TestCase
+  fixtures :reference_types, :reference_codes
+  
+  CLASSES = {
+    :single => {
+      :class => ReferenceType,
+      :primary_keys => [:reference_type_id],
+    },
+    :dual   => { 
+      :class => ReferenceCode,
+      :primary_keys => [:reference_type_id, :reference_code],
+    }
+  }
+  
+  def test_primary_keys
+    testing_with do
+      if composite?
+        assert_equal @primary_keys, @klass.primary_keys
+      else
+        assert_equal @primary_keys, [@klass.primary_key.to_sym]
+      end
+    end
+  end
+  
+  def test_find_first
+    testing_with do
+      obj = @klass.find_first
+      assert obj
+      assert_equal @klass, obj.class
+    end
+  end
+  
+  def test_ids
+    testing_with do
+      assert_equal @first.id, @first.ids if composite?
+    end
+  end
+  
+  def test_find
+    testing_with do
+      found = @klass.find(*first_id) # e.g. find(1,1) or find 1,1
+      assert found
+      assert_equal @klass, found.class
+      #breakpoint
+      assert_equal found, @klass.find(found.id)
+      assert_equal found, @klass.find(found.to_param)
+    end
+  end
+  
+  def test_to_param
+    testing_with do
+      assert_equal first_id_str, @first.to_param.to_s
+    end
+  end
+  
+  def things_to_look_at
+    testing_with do
+      assert_equal found, @klass.find(found.id.to_s) # fails for 2+ keys
+    end
+  end
+  
+protected
+  
+  def testing_with(&block)
+    CLASSES.keys.each do |@key_test|
+      @klass, @primary_keys = CLASSES[@key_test][:class], CLASSES[@key_test][:primary_keys]
+      @first = @klass.find_first
+      yield
+    end
+  end
+  
+  def first_id
+    (1..@primary_keys.length).map {|num| 1}
+  end
+  
+  def first_id_str
+    first_id.join(',')
+  end
+  
+  def composite?
+    @key_test != :single
+  end  
+end

data/test/fixtures/db_definitions/mysql.drop.sql

@@ -0,0 +1,30 @@
+DROP TABLE accounts;
+DROP TABLE funny_jokes;
+DROP TABLE companies;
+DROP TABLE topics;
+DROP TABLE developers;
+DROP TABLE projects;
+DROP TABLE developers_projects;
+DROP TABLE customers;
+DROP TABLE orders;
+DROP TABLE movies;
+DROP TABLE subscribers;
+DROP TABLE booleantests;
+DROP TABLE auto_id_tests;
+DROP TABLE entrants;
+DROP TABLE colnametests;
+DROP TABLE mixins;
+DROP TABLE people;
+DROP TABLE readers;
+DROP TABLE binaries;
+DROP TABLE computers;
+DROP TABLE tasks;
+DROP TABLE posts;
+DROP TABLE comments;
+DROP TABLE authors;
+DROP TABLE categories;
+DROP TABLE categories_posts;
+DROP TABLE fk_test_has_fk;
+DROP TABLE fk_test_has_pk;
+DROP TABLE keyboards;
+DROP TABLE legacy_things;

data/test/fixtures/db_definitions/mysql.sql

@@ -0,0 +1,16 @@
+CREATE TABLE `reference_types` (
+  `reference_type_id` int(11) NOT NULL auto_increment,
+  `type_label` varchar(50) default NULL,
+  `abbreviation` varchar(50) default NULL,
+  `description` varchar(50) default NULL,
+  PRIMARY KEY  (`reference_type_id`)
+) TYPE=InnoDB;
+
+CREATE TABLE `reference_codes` (
+  `reference_type_id` int(11) NOT NULL,
+  `reference_code` int(11) NOT NULL,
+  `code_label` varchar(50) default NULL,
+  `abbreviation` varchar(50) default NULL,
+  `description` varchar(50) default NULL,
+  PRIMARY KEY  (`reference_type_id`,`reference_code`)
+) TYPE=InnoDB;

data/test/fixtures/db_definitions/mysql2.drop.sql

@@ -0,0 +1,2 @@
+DROP TABLE courses;
+

data/test/fixtures/db_definitions/mysql2.sql

@@ -0,0 +1,5 @@
+CREATE TABLE `courses` (
+ `id` INTEGER NOT NULL PRIMARY KEY,
+ `name` VARCHAR(255) NOT NULL
+) TYPE=InnoDB;
+

data/test/fixtures/reference_code.rb

@@ -0,0 +1,7 @@
+class ReferenceCode < ActiveRecord::Base
+  set_primary_keys :reference_type_id, :reference_code
+  
+  belongs_to :reference_type, :foreign_key => "reference_type_id"
+  
+  validates_presence_of :reference_code, :code_label, :abbreviation
+end

data/test/fixtures/reference_codes.yml

@@ -0,0 +1,28 @@
+name_prefix_mr:
+  reference_type_id: 1
+  reference_code: 1
+  code_label: MR
+  abbreviation: Mr
+name_prefix_mrs:
+  reference_type_id: 1
+  reference_code: 2
+  code_label: MRS
+  abbreviation: Mrs
+name_prefix_ms:
+  reference_type_id: 1
+  reference_code: 3
+  code_label: MS
+  abbreviation: Ms
+  
+gender_male:
+  reference_type_id: 2
+  reference_code: 1
+  code_label: MALE
+  abbreviation: Male
+gender_female:
+  reference_type_id: 2
+  reference_code: 2
+  code_label: FEMALE
+  abbreviation: Female
+
+  

data/test/fixtures/reference_type.rb

@@ -0,0 +1,7 @@
+class ReferenceType < ActiveRecord::Base
+  set_primary_key :reference_type_id
+  has_many :reference_codes, :foreign_key => "reference_type_id"
+  
+  validates_presence_of :type_label, :abbreviation
+  validates_uniqueness_of :type_label
+end

data/test/fixtures/reference_types.yml

@@ -0,0 +1,9 @@
+name_prefix:
+  reference_type_id: 1
+  type_label: NAME_PREFIX
+  abbreviation: Name Prefix
+
+gender:
+  reference_type_id: 2
+  type_label: GENDER
+  abbreviation: Gender

metadata

@@ -0,0 +1,78 @@
+--- !ruby/object:Gem::Specification 
+rubygems_version: 0.8.11
+specification_version: 1
+name: composite_primary_keys
+version: !ruby/object:Gem::Version 
+  version: 0.1.3
+date: 2006-07-22 00:00:00 +02:00
+summary: Support for composite primary keys in ActiveRecords
+require_paths: 
+- lib
+email: drnicwilliams@gmail.com
+homepage: http://composite_primary_keys.rubyforge.org
+rubyforge_project: composite_primary_keys
+description: ActiveRecords only support a single primary key, preventing their use on legacy databases where tables have primary keys over 2+ columns. This solution allows an ActiveRecord to be extended to support multiple keys using the class method set_primary_keys.
+autorequire: composite_primary_keys
+default_executable: 
+bindir: bin
+has_rdoc: true
+required_ruby_version: !ruby/object:Gem::Version::Requirement 
+  requirements: 
+  - - ">"
+    - !ruby/object:Gem::Version 
+      version: 0.0.0
+  version: 
+platform: ruby
+signing_key: 
+cert_chain: 
+authors: 
+- Dr Nic Williams
+files: 
+- Rakefile
+- install.rb
+- README
+- CHANGELOG
+- lib/composite_primary_keys
+- lib/composite_primary_keys.rb
+- lib/composite_primary_keys/version.rb
+- lib/composite_primary_keys/base.rb
+- lib/composite_primary_keys/fixtures.rb
+- test/connections
+- test/abstract_unit.rb
+- test/dummy_test.rb
+- test/fixtures
+- test/find_test.rb
+- test/connections/native_mysql
+- test/connections/native_mysql/connection.rb
+- test/fixtures/reference_type.rb
+- test/fixtures/reference_code.rb
+- test/fixtures/reference_types.yml
+- test/fixtures/reference_codes.yml
+- test/fixtures/db_definitions
+- test/fixtures/db_definitions/mysql.drop.sql
+- test/fixtures/db_definitions/mysql.sql
+- test/fixtures/db_definitions/mysql2.drop.sql
+- test/fixtures/db_definitions/mysql2.sql
+test_files: []
+
+rdoc_options: 
+- --main
+- README
+extra_rdoc_files: 
+- README
+executables: []
+
+extensions: []
+
+requirements: []
+
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: activerecord
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Version::Requirement 
+    requirements: 
+    - - "="
+      - !ruby/object:Gem::Version 
+        version: 1.14.3
+    version: