fix: update to use postgres for test (#105)

* fix: update to use postgres * fix: flaky tests * fix: review * fix: lint
xmtp · May 17, 2022 · 0ec6b34 · 0ec6b34
1 parent cb9603b
commit 0ec6b34
Show file tree

Hide file tree

Showing 5 changed files with 13 additions and 3 deletions.
diff --git a/dev/docker-compose.yml b/dev/docker-compose.yml
@@ -1,12 +1,20 @@
 services:
   waku-node:
-    image: xmtp/node-go@sha256:489f72a4049272989b42032f6fe23aac91cfaf13c93605017d4dce93d21f4993
+    image: xmtp/node-go:latest
     environment:
       - GOWAKU-NODEKEY=8a30dcb604b0b53627a5adc054dbf434b446628d4bd1eccc681d223f0550ce67
     command:
       - --ws
       - --store
+      - --message-db-connection-string=postgres://postgres:xmtp@db:5432/postgres?sslmode=disable
       - --lightpush
       - --ws-port=9001
+      - --wait-for-db=30s
     ports:
       - 9001:9001
+    depends_on:
+      - db
+  db:
+    image: postgres:13
+    environment:
+      POSTGRES_PASSWORD: xmtp
diff --git a/src/store/PrivateTopicStore.ts b/src/store/PrivateTopicStore.ts
@@ -28,7 +28,7 @@ export default class NetworkStore implements Store {
 
   async set(key: string, value: Buffer): Promise<void> {
     const keys = Uint8Array.from(value)
-    await this.waku.relay.send(
+    await this.waku.lightPush.push(
       await WakuMessage.fromBytes(keys, this.buildTopic(key))
     )
   }

diff --git a/src/utils.ts b/src/utils.ts
@@ -52,7 +52,7 @@ export async function publishUserContact(
   keys: PublicKeyBundle,
   address: string
 ): Promise<void> {
-  return waku.relay.send(
+  await waku.lightPush.push(
     await WakuMessage.fromBytes(keys.toBytes(), buildUserContactTopic(address))
   )
 }
diff --git a/test/conversations/Conversations.test.ts b/test/conversations/Conversations.test.ts
@@ -17,6 +17,7 @@ describe('conversations', () => {
   beforeEach(async () => {
     alice = await Client.create(Wallet.createRandom(), opts)
     bob = await Client.create(Wallet.createRandom(), opts)
+    await sleep(100)
   })
 
   afterEach(async () => {

diff --git a/test/store/NetworkStore.test.ts b/test/store/NetworkStore.test.ts
@@ -56,6 +56,7 @@ describe('PrivateTopicStore', () => {
         expect(empty).toBeNull()
 
         await store.set(key, Buffer.from(value))
+        await sleep(100)
         const full = await store.get(key)
 
         expect(full).toBeDefined()