Skip to content

Commit c0a7ac3

Browse files
committed
fix previous commit; improve node/ test case a little
1 parent 253c9c7 commit c0a7ac3

File tree

2 files changed

+15
-8
lines changed

2 files changed

+15
-8
lines changed

intro/discovery/discovery/db/db.go

+1-1
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@ func createTables() (err error) {
4343

4444
const enableForeignKeys = `PRAGMA foreign_keys = ON;`
4545

46-
if _, err = sql.Exec(enableForeignKeys); err != nil {
46+
if _, err = engine.Exec(enableForeignKeys); err != nil {
4747
return
4848
}
4949

node/node_test.go

+14-7
Original file line numberDiff line numberDiff line change
@@ -278,22 +278,29 @@ func TestNode_Publish(t *testing.T) {
278278

279279
// check out channels
280280

281-
select {
282-
case <-gr1:
283-
case <-time.After(TM):
284-
t.Fatal("not received")
285-
}
281+
var after = time.After(TM)
286282

287283
select {
284+
case <-gr1:
285+
select {
286+
case <-gr2:
287+
case <-after:
288+
t.Fatal("not received")
289+
}
288290
case <-gr2:
289-
case <-time.After(TM):
291+
select {
292+
case <-gr1:
293+
case <-after:
294+
t.Fatal("not received")
295+
}
296+
case <-after:
290297
t.Fatal("not received")
291298
}
292299

293300
select {
294301
case <-gn1:
295302
t.Fatal("received")
296-
case <-time.After(TM):
303+
case <-after:
297304
}
298305

299306
// ok, now let's disconnect one by one and check

0 commit comments

Comments
 (0)