package Test::SQL::Traits;
use parent qw(IMPL::Test::Unit);

__PACKAGE__->PassThroughArgs;

use IMPL::lang;
use IMPL::Class::Property;
use IMPL::Test qw(test failed shared assert);

use IMPL::SQL::Schema;
use IMPL::SQL::Schema::Traits;
use IMPL::SQL::Types qw(Integer Varchar DateTime);

BEGIN {
    shared public property schema => prop_all; 
}

sub StartUnit {
    return {
        schema => new IMPL::SQL::Schema( name => 'testTraits', version => 1 )
    };
}

test CreateTable => sub {
    my ($this) = @_;
    
    my $table = $this->schema->AddTable(
        new IMPL::SQL::Schema::Traits::Table(
            'user'
        )
    ) or failed "Failed to create table";
    
    $this->schema->GetTable('user') or failed "Can't get a created table";
    
};

test InsertColumn => sub {
    my ($this) = @_;
    
    my $table = $this->schema->GetTable('user');
    
    $table->InsertColumn(
        new IMPL::SQL::Schema::Traits::Column(
            id => Integer, tag => { auto_increment => 1 }
        )
    );
    
    my $column = $table->GetColumn('id') or failed "Column not found";
    
    assert( $column->name eq 'id');
    assert( $column->type->SameValue(Integer()) );
    assert( not $column->isNullable );
    assert( $column->tag->{auto_increment} );
    
    $table->InsertColumn(
        new IMPL::SQL::Schema::Traits::Column(
            name => Varchar(255), isNullable => 1
        )
    );
    
    $column = $table->GetColumn('name');
    
    assert($column);
    assert($column->name eq 'name');
    assert($column->type->SameValue(Varchar(255)));
    assert($column->isNullable);
};

test CreateTableWithColumns => sub {
    my ($this) = @_;
    
    my $table = $this->schema->AddTable(
        new IMPL::SQL::Schema::Traits::Table(
            session => [
                new IMPL::SQL::Schema::Traits::Column( id => Varchar(64)),
                new IMPL::SQL::Schema::Traits::Column( expires => DateTime ),
                new IMPL::SQL::Schema::Traits::Column( role => Varchar(64), defaultValue => 'user' )
            ]
        )
    ) or failed "Failed to create table";
    
    assert( $table->ColumnsCount == 3 );
    
    assert( my $column = $table->GetColumn('id') );
    assert($column->type->SameValue(Varchar(64)));
    assert(not $column->isNullable);
    
    assert( $column = $table->GetColumn('role') );
    assert( $column->defaultValue eq 'user' );
};

sub FinishUnit {
    my ($self,$session) = @_;
    
    $self->supercall::FinishUnit();
    
    $session->{schema}->Dispose();
}

1;