dimgigov
6bfc5b3a3c
CI / test (push) Has been cancelled
CI / verify (push) Has been cancelled
Clients CI / build-server (push) Has been cancelled
Clients CI / test-python (push) Has been cancelled
Clients CI / test-javascript (push) Has been cancelled
Clients CI / test-nim (push) Has been cancelled
Clients CI / test-rust (push) Has been cancelled
- Pure Nim wire protocol client (async + sync, no C FFI) - Query Builder: fluent API, SQL generator, execution, transactions - Schema Builder: create/alter/drop tables, column operations - Connection pool with aging and timeout handling - ORM integration via compile-time DB_BARADB switch - Tests: test_open, test_query - Documentation: README.md, PLAN_BARADB.md
132 lines
3.1 KiB
Nim
132 lines
3.1 KiB
Nim
discard """
|
|
cmd: "nim c --mm:orc -d:reset -r $file"
|
|
"""
|
|
|
|
# nim c -r -d:reset tests/v2/postgres/test_transaction.nim
|
|
|
|
import std/unittest
|
|
import std/json
|
|
import std/strformat
|
|
import std/strutils
|
|
import std/options
|
|
import std/asyncdispatch
|
|
import std/macros
|
|
import ../../src/allographer/schema_builder
|
|
import ../../src/allographer/query_builder
|
|
import ./connections
|
|
import ../clear_tables
|
|
|
|
|
|
let rdb = postgres
|
|
|
|
proc setUp(rdb:PostgresConnections) =
|
|
rdb.create([
|
|
table("auth",[
|
|
Column.increments("id"),
|
|
Column.string("auth")
|
|
]),
|
|
table("user",[
|
|
Column.increments("id"),
|
|
Column.string("name").nullable(),
|
|
Column.string("email").nullable(),
|
|
Column.string("address").nullable(),
|
|
Column.foreign("auth_id").reference("id").onTable("auth").onDelete(SET_NULL)
|
|
])
|
|
])
|
|
|
|
# seeder
|
|
seeder(rdb, "auth"):
|
|
rdb.table("auth").insert(@[
|
|
%*{"auth": "admin"},
|
|
%*{"auth": "user"}
|
|
])
|
|
.waitFor()
|
|
|
|
seeder(rdb, "user"):
|
|
var users: seq[JsonNode]
|
|
for i in 1..10:
|
|
let authId = if i mod 2 == 0: 2 else: 1
|
|
users.add(
|
|
%*{
|
|
"name": &"user{i}",
|
|
"email": &"user{i}@example.com",
|
|
"auth_id": authId
|
|
}
|
|
)
|
|
|
|
rdb.table("user").insert(users).waitFor()
|
|
|
|
|
|
suite("raw code transaction"):
|
|
setup:
|
|
setUp(rdb)
|
|
|
|
test("commit"):
|
|
(proc() {.async.} =
|
|
var user1 = rdb.table("user").find(1).await.get()
|
|
check user1["name"].getStr() == "user1"
|
|
|
|
block:
|
|
rdb.begin().await
|
|
try:
|
|
rdb.table("user").where("id", "=", 1).update(%*{"name": "updated"}).await
|
|
rdb.commit().await
|
|
except CatchableError:
|
|
rdb.rollback().await
|
|
|
|
user1 = rdb.table("user").find(1).await.get()
|
|
check user1["name"].getStr() == "updated"
|
|
)().waitFor
|
|
|
|
|
|
test("rollback"):
|
|
(proc() {.async.} =
|
|
var user1 = rdb.table("user").find(1).await.get()
|
|
check user1["name"].getStr() == "user1"
|
|
|
|
block:
|
|
rdb.begin().await
|
|
try:
|
|
rdb.table("user").where("id", "=", 1).update(%*{"name": "updated"}).await
|
|
raise newException(CatchableError, "")
|
|
except:
|
|
rdb.rollback().await
|
|
|
|
user1 = rdb.table("user").find(1).await.get()
|
|
check user1["name"].getStr() == "user1"
|
|
)().waitFor
|
|
|
|
|
|
suite("transaction template"):
|
|
setup:
|
|
setUp(rdb)
|
|
|
|
test("commit"):
|
|
(proc(){.async.} =
|
|
var user1 = rdb.table("user").find(1).await.get()
|
|
check user1["name"].getStr() == "user1"
|
|
|
|
transaction(rdb):
|
|
rdb.table("user").where("id", "=", 1).update(%*{"name": "updated"}).await
|
|
|
|
user1 = rdb.table("user").find(1).await.get()
|
|
check user1["name"].getStr() == "updated"
|
|
)().waitFor()
|
|
|
|
|
|
test("rollback"):
|
|
(proc(){.async.} =
|
|
var user1 = rdb.table("user").find(1).await.get()
|
|
check user1["name"].getStr() == "user1"
|
|
|
|
transaction(rdb):
|
|
rdb.table("user").where("id", "=", 1).update(%*{"name": "updated"}).await
|
|
raise newException(DbError, "")
|
|
|
|
user1 = rdb.table("user").find(1).await.get()
|
|
check user1["name"].getStr() == "user1"
|
|
)().waitFor()
|
|
|
|
|
|
clearTables(rdb).waitFor()
|