#!/usr/bin/env ruby

require 'TestSetup'
require 'test/unit'
require 'rubygems'
require_gem 'fireruby'

include FireRuby

class BackupRestoreTest < Test::Unit::TestCase
   CURDIR      = "#{Dir.getwd}"
   DB_FILE     = "#{CURDIR}#{File::SEPARATOR}backup_restore_unit_test.fdb"
   BACKUP_FILE = "#{CURDIR}#{File::SEPARATOR}database.bak"

   def setup
      # Remove existing database files.
      if File.exist?(DB_FILE)
         db = Database.new(DB_FILE).drop(DB_USER_NAME, DB_PASSWORD)
      end

      if File.exist?(BACKUP_FILE)
         File.delete(BACKUP_FILE)
      end

      # Create and populate the database files.
      @database = Database.create(DB_FILE, DB_USER_NAME, DB_PASSWORD)
      @database.connect(DB_USER_NAME, DB_PASSWORD) do |cxn|
         cxn.execute_immediate('create table test(id integer)')
         cxn.execute_immediate('insert into test values (1000)')
         cxn.execute_immediate('insert into test values (2000)')
         cxn.execute_immediate('insert into test values (NULL)')
         cxn.execute_immediate('insert into test values (3000)')
         cxn.execute_immediate('insert into test values (4000)')
      end
   end

   def teardown
      # Remove existing database files.
      if File.exist?(DB_FILE)
         db = Database.new(DB_FILE).drop(DB_USER_NAME, DB_PASSWORD)
      end

      if File.exist?(BACKUP_FILE)
         File.delete(BACKUP_FILE)
      end
   end

   def test01
      sm = ServiceManager.new('localhost')
      sm.connect(DB_USER_NAME, DB_PASSWORD)

      b = Backup.new(DB_FILE, BACKUP_FILE)
      b.execute(sm)

      assert(File.exist?(BACKUP_FILE))
      @database.drop(DB_USER_NAME, DB_PASSWORD)
      assert(File.exists?(DB_FILE) == false)

      r = Restore.new(BACKUP_FILE, DB_FILE)
      r.execute(sm)
      sm.disconnect

      assert(File.exist?(DB_FILE))
      @database.connect(DB_USER_NAME, DB_PASSWORD) do |cxn|
         total = 0
         cxn.execute_immediate('select * from test') do |row|
            assert([1000, 2000, 3000, 4000, nil].include?(row[0]))
            total += 1
         end
         assert(total == 5)
      end
   end

   def test02
      sm = ServiceManager.new('localhost')
      sm.connect(DB_USER_NAME, DB_PASSWORD)

      b = Backup.new(DB_FILE, BACKUP_FILE)
      b.metadata_only = true
      b.execute(sm)

      assert(File.exist?(BACKUP_FILE))
      @database.drop(DB_USER_NAME, DB_PASSWORD)
      assert(File.exists?(DB_FILE) == false)

      r = Restore.new(BACKUP_FILE, DB_FILE)
      r.execute(sm)
      sm.disconnect

      assert(File.exist?(DB_FILE))
      @database.connect(DB_USER_NAME, DB_PASSWORD) do |cxn|
         cxn.execute_immediate('select count(*) from test') do |row|
            assert(row[0] == 0)
         end
      end
   end
end