# frozen_string_literal: true

Bundler.require :cassandra

CassandraObject::Base.config = {
  keyspace: 'cassandra_object_test',
  hosts: ['127.0.0.1'],
  compression: :lz4,
  connect_timeout: 1,
  timeout: 30,
  consistency: :quorum,
  protocol_version: 3,
  page_size: 10000,
  trace: true,
  connections_per_local_node: 4,
  schema_refresh_delay: 0.1,
  schema_refresh_timeout: 0.1,
  load_balancing_policy: Cassandra::LoadBalancing::Policies::RoundRobin.new,
  reconnection_policy: { policy: 'Constant', params: [5] },
  retry_policy: 'Default',
  # connections_per_remote_node: nil,
  # logger: Logger.new($stderr)
}

begin
  CassandraObject::Schema.drop_keyspace 'cassandra_object_test', true
  CassandraObject::Schema.drop_keyspace 'Blah', true
rescue Exception => e
  puts e.message
end

sleep 1
CassandraObject::Schema.create_keyspace 'cassandra_object_test'
CassandraObject::Schemaless.create_column_family 'Issues'
CassandraObject::Schema.create_column_family 'IssueSchemas', { attributes: 'id text, title text, description text, field float, intero int, created_at timestamp, updated_at timestamp, PRIMARY KEY (id)', options: {} }
CassandraObject::Schema.create_column_family 'IssueSchemaCks', { attributes: 'id text, type text, date timestamp, value float, PRIMARY KEY (id, type, date)', options: {} }
CassandraObject::Schemaless.create_column_family 'IssueDynamics'
CassandraObject::Schemaless.create_column_family 'IssuesCustomConfig'
CassandraObject::Schema.create_column_family 'IssueSchemaFathers', { attributes: 'id text, title text, field float, created_at timestamp, updated_at timestamp, PRIMARY KEY (id)', options: {} }
CassandraObject::Schema.create_column_family 'IssueSchemaChildren', { attributes: 'id text, title text, description text, field float, created_at timestamp, updated_at timestamp, issue_schema_father_id text, PRIMARY KEY (id)', options: {} }

CassandraObject::Base.class_eval do
  class_attribute :created_records
  self.created_records = []

  after_create do
    created_records << self
  end

  def self.delete_after_test
    # created_records.reject(&:destroyed?).each(&:destroy)
    Issue.delete_all
    IssueSchema.delete_all
    IssueSchemaCk.delete_all
    IssueDynamic.delete_all
    created_records.clear
  end
end

module ActiveSupport
  class TestCase
    self.test_order = :random

    def after_setup
      CassandraObject::Base.created_records.clear if CassandraObject::Base.created_records.any?
    end

    teardown do
      if CassandraObject::Base.created_records.any?
        CassandraObject::Base.delete_after_test
      end
    end
  end
end