Skip to content

Commit 6be702f

Browse files
committed
add test for iterator
1 parent 0f80fe9 commit 6be702f

File tree

1 file changed

+12
-5
lines changed

1 file changed

+12
-5
lines changed

tests/test_type.nim

Lines changed: 12 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -8,9 +8,16 @@ type
88
proc createUsersTable() {.importdb: "create table users(id INTEGER PRIMARY KEY, username TEXT NOT NULL, email TEXT NOT NULL)".}
99
proc insertUser(username, email: string) {.importdb: "insert into users(username, email) values ($username, $email)".}
1010
proc selectUserById(id: int): User {.importdb: "select * from users where id = $id".}
11+
iterator iterateUsers(): User {.importdb: "select * from users".} = discard
1112

12-
test "type decl":
13-
var db = initDatabase(":memory:")
14-
db.createUsersTable()
15-
db.insertUser("user", "[email protected]")
16-
check db.selectUserById(1) == (id: 1, username: "user", email: "[email protected]")
13+
suite "type decl":
14+
setup:
15+
var db = initDatabase(":memory:")
16+
db.createUsersTable()
17+
db.insertUser("user", "[email protected]")
18+
db.insertUser("abc", "[email protected]")
19+
test "typed query":
20+
check db.selectUserById(1) == (id: 1, username: "user", email: "[email protected]")
21+
test "typed iterator":
22+
for id, username, email in db.iterateUsers():
23+
check email == (username & "@example.com")

0 commit comments

Comments
 (0)