require 'faker'
require 'uuidtools'

module Bench
  module TestData
    def get_test_data(num=1000,generate=false)
      file = File.join(File.dirname(__FILE__),'..',"testdata","#{num}-data.txt")
      data = nil
      if File.exists?(file) and not generate
        data = open(file, 'r') {|f| Marshal.load(f)}
      else
        data = generate_fake_data(num)
        f = File.new(file, 'w')
        f.write Marshal.dump(data)
        f.close
      end
      data
    end

    private

    PREFIX = ["Account", "Administrative", "Advertising", "Assistant", "Banking", "Business Systems", 
      "Computer", "Distribution", "IT", "Electronics", "Environmental", "Financial", "General", "Head", 
      "Laboratory", "Maintenance", "Medical", "Production", "Quality Assurance", "Software", "Technical", 
      "Chief", "Senior"] unless defined? PREFIX
    SUFFIX = ["Clerk", "Analyst", "Manager", "Supervisor", "Plant Manager", "Mechanic", "Technician", "Engineer", 
      "Director", "Superintendent", "Specialist", "Technologist", "Estimator", "Scientist", "Foreman", "Nurse", 
      "Worker", "Helper", "Intern", "Sales", "Mechanic", "Planner", "Recruiter", "Officer", "Superintendent",
      "Vice President", "Buyer", "Production Supervisor", "Chef", "Accountant", "Executive"] unless defined? SUFFIX

    def title
      prefix = PREFIX[rand(PREFIX.length)]
      suffix = SUFFIX[rand(SUFFIX.length)]

      "#{prefix} #{suffix}"
    end

    def generate_fake_data(num=1000)
      res = {}
      num.times do |n|
        mock_id = UUIDTools::UUID.random_create.to_s.gsub(/\-/,'')
        res[mock_id] = {
          "mock_id" => mock_id,
          "FirstName" => Faker::Name.first_name,
          "LastName" => Faker::Name.last_name,
          "Email" =>  Faker::Internet.free_email,
          "Company" => Faker::Company.name,
          "JobTitle" => title,
          "Phone1" => Faker::PhoneNumber.phone_number
        }
      end
      res
    end
  end
end