sqlite_seed.rb

module Support
  module SqliteSeed
    def seed_db
      newspaper1 = Newspaper.create(:name => "First Newspaper")
      newspaper2 = Newspaper.create(:name => "Second Newspaper")

      writer1 = Writer.create(:name => 'first', :newspaper => newspaper1)
      writer2 = Writer.create(:name => 'second', :newspaper => newspaper2)
      user1 = BaseUser.create(:name => 'third', :newspaper => newspaper1)
      user2 = BaseUser.create(:name => 'fourth', :newspaper => newspaper2)


      category1 = Category.create(:name => 'first')
      category2 = Category.create(:name => 'second')

      post1 = category1.posts.create(:name => 'first', :writer => writer1)
      post1a = category1.posts.create(:name => 'like first', :writer => writer2)
      post2 = category2.posts.create(:name => 'second', :writer => writer2)

      comment1 = post1.comments.create(:name => 'first', :author => writer1)
      comment2 = post1.comments.create(:name => 'first2', :author => writer1)
      comment3 = post1.comments.create(:name => 'first3', :author => writer1)
      comment4 = post1.comments.create(:name => 'second', :author => writer2)
      comment8 = post1a.comments.create(:name => "like first 1", :author => writer1)
      comment9 = post1a.comments.create(:name => "like first 2", :author => writer2)
      comment5 = post2.comments.create(:name => 'third', :author => user1)
      comment6 = post2.comments.create(:name => 'fourth', :author => user2)
      comment7 = post2.comments.create(:name => 'fourth', :author => writer1)

      entry1 = category1.entries.create(:name => 'first')
      entry2 = category1.entries.create(:name => 'second')

      student1 = Student.create(:name => 'first')
      student2 = Student.create(:name => 'second')
      teacher1 = Teacher.create(:name => 'first')
      teacher2 = Teacher.create(:name => 'second')
      student1.teachers = [teacher1, teacher2]
      student2.teachers = [teacher1, teacher2]
      teacher1.students << student1
      teacher2.students << student2

      firm1 = Firm.create(:name => 'first')
      firm2 = Firm.create(:name => 'second')
      client1 = Client.create(:name => 'first')
      client2 = Client.create(:name => 'second')
      firm1.clients = [client1, client2]
      firm2.clients = [client1, client2]
      client1.firms << firm1
      client2.firms << firm2

      company1 = Company.create(:name => 'first')
      company2 = Company.create(:name => 'second')

      Address.create(:name => 'first', :company => company1)
      Address.create(:name => 'second', :company => company2)

      country1 = Country.create(:name => 'first')
      country2 = Country.create(:name => 'second')

      country1.cities.create(:name => 'first')
      country1.cities.create(:name => 'second')
      country2.cities.create(:name => 'third')
      country2.cities.create(:name => 'fourth')

      person1 = Person.create(:name => 'first')
      person2 = Person.create(:name => 'second')

      person1.pets.create(:name => 'first')
      person1.pets.create(:name => 'second')
      person2.pets.create(:name => 'third')
      person2.pets.create(:name => 'fourth')

      author1 = Author.create(:name => 'author1')
      author2 = Author.create(:name => 'author2')
      folder1 = Folder.create(:name => 'folder1', :author_id => author1.id)
      folder2 = Folder.create(:name => 'folder2', :author_id => author2.id)
      page1 = Page.create(:name => 'page1', :parent_id => folder1.id, :author_id => author1.id)
      page2 = Page.create(:name => 'page2', :parent_id => folder1.id, :author_id => author1.id)
      page3 = Page.create(:name => 'page3', :parent_id => folder2.id, :author_id => author2.id)
      page4 = Page.create(:name => 'page4', :parent_id => folder2.id, :author_id => author2.id)

      user1 = User.create(:name => 'user1', :category => category1)
      user2 = User.create(:name => 'user2', :category => category1)

      submission1 = category1.submissions.create(:name => "submission1", :user => user1)
      submission2 = category1.submissions.create(:name => "submission2", :user => user2)
      submission3 = category2.submissions.create(:name => "submission3", :user => user1)
      submission4 = category2.submissions.create(:name => "submission4", :user => user2)
    end

    def setup_db
      ActiveRecord::Base.establish_connection(:adapter => 'sqlite3', :database => ':memory:')

      ActiveRecord::Schema.define(:version => 1) do
        create_table :addresses do |t|
          t.column :name, :string
          t.column :company_id, :integer
        end

        create_table :authors do |t|
          t.string :name
        end

        create_table :base_users do |t|
          t.column :name, :string
          t.column :type, :string
          t.column :newspaper_id, :integer
        end

        create_table :categories do |t|
          t.column :name, :string
        end

        create_table :cities do |t|
          t.string :name
          t.integer :country_id
        end

        create_table :clients do |t|
          t.column :name, :string
        end

        create_table :comments do |t|
          t.column :name, :string
          t.column :post_id, :integer
          t.column :author_id, :integer
        end

        create_table :companies do |t|
          t.column :name, :string
        end

        create_table :contacts do |t|
          t.column :name, :string
        end

        create_table :countries do |t|
          t.string :name
        end

        create_table :deals do |t|
          t.column :name, :string
          t.column :hotel_id, :integer
        end

        create_table :documents do |t|
          t.string :name
          t.string :type
          t.integer :parent_id
          t.integer :author_id
        end

        create_table :emails do |t|
          t.column :name, :string
          t.column :contact_id, :integer
        end

        create_table :entries do |t|
          t.column :name, :string
          t.column :category_id, :integer
        end

        create_table :firms do |t|
          t.column :name, :string
        end

        create_table :hotels do |t|
          t.column :name, :string
          t.column :location_id, :integer
        end

        create_table :locations do |t|
          t.column :name, :string
        end

        create_table :newspapers do |t|
          t.column :name, :string
        end

        create_table :people do |t|
          t.string :name
          t.integer :pets_count
        end

        create_table :pets do |t|
          t.string :name
          t.integer :person_id
        end

        create_table :posts do |t|
          t.column :name, :string
          t.column :category_id, :integer
          t.column :writer_id, :integer
        end

        create_table :relationships do |t|
          t.column :firm_id, :integer
          t.column :client_id, :integer
        end

        create_table :students do |t|
          t.column :name, :string
        end

        create_table :students_teachers, :id => false do |t|
          t.column :student_id, :integer
          t.column :teacher_id, :integer
        end

        create_table :teachers do |t|
          t.column :name, :string
        end

        create_table :submissions do |t|
          t.column :name, :string
          t.column :category_id, :integer
          t.column :user_id, :integer
        end

        create_table :users do |t|
          t.column :name, :string
          t.column :category_id, :integer
        end
      end
    end

    def teardown_db
      ActiveRecord::Base.connection.tables.each do |table|
        ActiveRecord::Base.connection.drop_table(table)
      end
    end

    extend self
  end
end