sjdonado
87 lines
2.3 KiB
Crystal
87 lines
2.3 KiB
Crystal
require "file_utils"
|
|
|
|
require "spec-kemal"
|
|
require "micrate"
|
|
|
|
require "dotenv"
|
|
Dotenv.load ".env.#{ENV["ENV"]}"
|
|
|
|
require "../bit"
|
|
|
|
Spec.before_suite do
|
|
# Delete the SQLite database file if it exists
|
|
db_file_path = ENV["DATABASE_URL"].split("sqlite3://").last.split("?").first
|
|
if File.exists?(db_file_path)
|
|
File.delete(db_file_path)
|
|
end
|
|
|
|
Micrate::DB.connection_url = ENV["DATABASE_URL"]
|
|
Micrate::Cli.run_up
|
|
|
|
Kemal.config.logging = false
|
|
end
|
|
|
|
def create_test_user
|
|
user = App::Models::User.new
|
|
user.name = "Tester"
|
|
user.api_key = Random::Secure.urlsafe_base64()
|
|
|
|
changeset = App::Lib::Database.insert(user)
|
|
if !changeset.valid?
|
|
error_messages = changeset.errors.map { |error| "#{error}" }.join(", ")
|
|
raise "Test user creation failed #{error_messages}"
|
|
end
|
|
|
|
changeset.instance
|
|
end
|
|
|
|
def create_test_link(user, url)
|
|
link = App::Models::Link.new
|
|
link.slug = App::Services::SlugService.shorten_url(url, user.id.not_nil!)
|
|
link.url = url
|
|
link.user = user
|
|
|
|
changeset = App::Lib::Database.insert(link)
|
|
unless changeset.valid?
|
|
error_messages = changeset.errors.map { |error| "#{error}" }.join(", ")
|
|
raise "Test link creation failed: #{error_messages}"
|
|
end
|
|
|
|
inserted_link = changeset.instance
|
|
inserted_link.clicks = [] of App::Models::Click
|
|
|
|
inserted_link
|
|
end
|
|
|
|
def create_test_click(link)
|
|
click = App::Models::Click.new
|
|
click.user_agent = "Mozilla/5.0 (Macintosh; Intel Mac OS X 10.15; rv:127.0) Gecko/20100101 Firefox/127.0"
|
|
click.browser = "Firefox"
|
|
click.os = "Mac OS X"
|
|
click.referer = "example.com"
|
|
click.country = "US"
|
|
click.created_at = Time.utc
|
|
click.link = link
|
|
click.link_id = link.id.not_nil!
|
|
|
|
changeset = App::Lib::Database.insert(click)
|
|
unless changeset.valid?
|
|
error_messages = changeset.errors.map { |error| "#{error}" }.join(", ")
|
|
raise "Test click creation failed: #{error_messages}"
|
|
end
|
|
changeset.instance
|
|
end
|
|
|
|
def get_test_link(link_id : Int64)
|
|
query = App::Lib::Database::Query.where(id: link_id).limit(1)
|
|
link = App::Lib::Database.all(App::Models::Link, query, preload: [:clicks]).first?
|
|
|
|
raise "Link not found" if link.nil?
|
|
|
|
link
|
|
end
|
|
|
|
def delete_test_link(link_id : Int64)
|
|
App::Lib::Database.raw_exec("DELETE FROM links WHERE id = (?)", link_id) # tempfix: Database.delete does not work
|
|
end
|