foodsoft/db/migrate/002_create_groups.rb

class CreateGroups < ActiveRecord::Migration[4.2]
  GROUP_ADMIN = 'Administrators'
  GROUP_ORDER = 'Sample Order Group'

  def self.up
    create_table :groups do |t|
      t.column :type, :string, null: false # inheritance, types: Group, OrderGroup
      t.column :name, :string, null: false
      t.column :description, :string
      t.column :actual_size, :integer # OrderGroup column
      t.column :account_balance, :decimal, precision: 8, scale: 2, null: false, default: 0 # OrderGroup column
      t.column :account_updated, :timestamp # OrderGroup column
      t.column :created_on, :timestamp, null: false
      t.column :role_admin, :boolean, default: false, null: false
    end
    add_index(:groups, :name, unique: true)

    create_table :memberships do |t|
      t.column :group_id, :integer, null: false
      t.column :user_id, :integer, null: false
    end
    add_index(:memberships, %i[user_id group_id], unique: true)

    # Create the default "Administrators" group...
    puts "Creating group #{GROUP_ADMIN}..."
    Group.create(name: GROUP_ADMIN, description: 'System administrators.', role_admin: true)
    raise 'Failed!' unless administrators = Group.find_by_name(GROUP_ADMIN)

    # Create a sample order group...
    puts "Creating order group #{GROUP_ORDER}..."
    ordergroup = OrderGroup.create!(name: GROUP_ORDER,
                                    description: 'A sample order group created by the migration.', actual_size: 1, account_updated: Time.now)
    raise 'Wrong type created!' unless ordergroup.is_a?(OrderGroup)

    # Get the admin user and join the admin group...
    raise "User #{CreateUsers::USER_ADMIN} not found, cannot join group '#{administrators.name}'!" unless admin = User.find_by_nick(CreateUsers::USER_ADMIN)

    puts "Joining #{CreateUsers::USER_ADMIN} user to new '#{administrators.name}' group as a group admin..."
    membership = Membership.create(group: administrators, user: admin)
    raise 'Failed!' unless admin.memberships.first == membership
    raise "User #{CreateUsers::USER_ADMIN} has no admin_roles" unless admin.role_admin?

    # Get the test user and join the order group...
    raise "User #{CreateUsers::USER_TEST} not found, cannot join group '#{ordergroup.name}'!" unless test = User.find_by_nick(CreateUsers::USER_TEST)

    puts "Joining #{CreateUsers::USER_TEST} user to new '#{ordergroup.name}' group as a group admin..."
    membership = Membership.create(group: ordergroup, user: test)
    raise 'Failed!' unless test.memberships.first == membership
  end

  def self.down
    drop_table :groups
    drop_table :memberships
  end
end