import db.sqlite

struct Child {
    id        int @[primary; serial]
    parent_id int
}

struct PlainNoArg {
    id       int @[primary; serial; sql: 'custom_id']
    name     string
    children []Child @[fkey: 'parent_id']
}

fn test_plain_no_arg() {
    mut db := sqlite.connect(':memory:')!
    defer { db.close() or {} }

    sql db {
        create table PlainNoArg
    }!

    first := PlainNoArg{
        name: 'first'
    }
    second := PlainNoArg{
        name: 'second'
    }

    sql db {
        insert first into PlainNoArg
        insert second into PlainNoArg
    }!

    rows := sql db {
        select from PlainNoArg order by id desc
    }!

    assert rows[0].id == 2
    assert rows[0].name == 'second'
}

fn test_plain_no_arg_with_children() {
    mut db := sqlite.connect(':memory:')!

    parent := PlainNoArg{
        children: [Child{}, Child{}]
    }

    sql db {
        create table Child
        create table PlainNoArg
        insert parent into PlainNoArg
    }!

    parents := sql db {
        select from PlainNoArg limit 1
    }!

    assert parent.children.len == parents[0].children.len
}

struct SqlSerial {
    id   int @[primary; sql: serial]
    name string
}

fn test_sql_serial() {
    mut db := sqlite.connect(':memory:')!
    defer { db.close() or {} }

    sql db {
        create table SqlSerial
    }!

    first := SqlSerial{
        name: 'first'
    }
    second := SqlSerial{
        name: 'second'
    }

    sql db {
        insert first into SqlSerial
        insert second into SqlSerial
    }!

    rows := sql db {
        select from SqlSerial order by id desc
    }!

    assert rows[0].id == 2
    assert rows[0].name == 'second'
}