Skip to content
This repository has been archived by the owner on Jul 1, 2023. It is now read-only.

add ownership verify test #87

Open
wants to merge 3 commits into
base: main
Choose a base branch
from
Open

add ownership verify test #87

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
165d9b1
add ownership verify test
jingchen2222 Jun 23, 2023
12b1fd3
add doc id not found ut
jingchen2222 Jun 24, 2023
39319ff
fix ut
jingchen2222 Jun 25, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions src/store/document_v2.ts
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -112,7 +112,7 @@ export async function deleteDoc(col: Collection, ids: string[]) {
col.db.client.nonce += 1
return [response.id, response.block, response.order]
} else {
throw new Error('fail to create collection')
throw new Error('fail to delete doc')
}
}

Expand Down Expand Up @@ -151,7 +151,7 @@ export async function updateDoc(
col.db.client.nonce += 1
return [response.id, response.block, response.order]
} else {
throw new Error('fail to create collection')
throw new Error('fail to update doc')
}
}
export async function addDoc(col: Collection, doc: DocumentData) {
Expand Down
290 changes: 288 additions & 2 deletions tests/client_v2.test.ts
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -39,8 +39,8 @@ import {
createDocumentDatabase,
createEventDatabase,
showDatabase,
createCollection,
} from '../src/store/database_v2'
createCollection, getDatabase, getCollection
} from '../src/store/database_v2';
import { Index, IndexType } from '../src/proto/db3_database_v2'

interface Profile {
Expand Down Expand Up @@ -288,6 +288,292 @@ describe('test db3.js client module', () => {
)
expect(0).toBe(resultSet.docs.length)
}
try {
const [txId4, block4, order4] = await updateDoc(
collection,
docId2,
{
city: 'beijing3',
author: 'imotai3',
age: 3,
},
[]
)
fail('should not reach here')
} catch (e) {
console.log(e)
expect(decodeURI(e.message)).toBe('fail to verify the owner with error doc id is not found')
}

}
} catch (e) {
console.log(e)
expect(1).toBe(0)
}
})
test('test delete doc id not found', async () => {
const client = await createTestClient()
try {
const { db, result } = await createDocumentDatabase(
client,
'db_for_update_delete'
)
const index: Index = {
path: '/city',
indexType: IndexType.StringKey,
}
{
const { collection, result } = await createCollection(
db,
'col',
[index]
)
await new Promise((r) => setTimeout(r, 2000))
const [txId2, block2, order2, docId2] = await addDoc(
collection,
{
city: 'beijing',
author: 'imotai',
age: 10,
}
)
await new Promise((r) => setTimeout(r, 2000))
{
const queryStr = '/[city = beijing]'
const resultSet = await queryDoc<Profile>(
collection,
queryStr
)
expect(1).toBe(resultSet.docs.length)
expect(resultSet.docs[0].doc.city).toBe('beijing')
expect(resultSet.docs[0].doc.author).toBe('imotai')
expect(resultSet.docs[0].doc.age).toBe(10)
expect(resultSet.docs[0].id).toBe(docId2)
}

const [txId, block, order] = await deleteDoc(collection, [
docId2,
])
await new Promise((r) => setTimeout(r, 2000))
{
const queryStr = '/[city = beijing3]'
const resultSet = await queryDoc<Profile>(
collection,
queryStr
)
expect(0).toBe(resultSet.docs.length)
}
try {
const [txId, block, order] = await deleteDoc(collection, [
docId2,
])
fail('should not reach here')
} catch (e) {
expect(decodeURI(e.message)).toBe('fail to verify the owner with error doc id is not found')
}

}
} catch (e) {
console.log(e)
expect(1).toBe(0)
}
})
test('test update doc id not found', async () => {
const client = await createTestClient()
try {
const { db, result } = await createDocumentDatabase(
client,
'db_for_update_delete'
)
const index: Index = {
path: '/city',
indexType: IndexType.StringKey,
}
{
const { collection, result } = await createCollection(
db,
'col',
[index]
)
await new Promise((r) => setTimeout(r, 3000))
const [txId2, block2, order2, docId2] = await addDoc(
collection,
{
city: 'beijing',
author: 'imotai',
age: 10,
}
)
await new Promise((r) => setTimeout(r, 2000))
{
const queryStr = '/[city = beijing]'
const resultSet = await queryDoc<Profile>(
collection,
queryStr
)
expect(1).toBe(resultSet.docs.length)
expect(resultSet.docs[0].doc.city).toBe('beijing')
expect(resultSet.docs[0].doc.author).toBe('imotai')
expect(resultSet.docs[0].doc.age).toBe(10)
expect(resultSet.docs[0].id).toBe(docId2)
}

const [txId, block, order] = await deleteDoc(collection, [
docId2,
])
await new Promise((r) => setTimeout(r, 2000))
{
const queryStr = '/[city = beijing3]'
const resultSet = await queryDoc<Profile>(
collection,
queryStr
)
expect(0).toBe(resultSet.docs.length)
}
try {
const [txId4, block4, order4] = await updateDoc(
collection,
docId2,
{
city: 'beijing3',
author: 'imotai3',
age: 3,
},
[]
)
fail('should not reach here')
} catch (e) {
expect(decodeURI(e.message)).toBe('fail to verify the owner with error doc id is not found')
}

}
} catch (e) {
console.log(e)
expect(1).toBe(0)
}
})
test('test ownership verify for update/delete document', async () => {
const client = await createTestClient()
const client2 = await createTestClient()
try {
const { db, result } = await createDocumentDatabase(
client,
'db_for_update_delete'
)
const index: Index = {
path: '/city',
indexType: IndexType.StringKey,
}
{
const { collection, result } = await createCollection(
db,
'col',
[index]
)
const collection2 = await getCollection(db.addr, 'col', client2)

await new Promise((r) => setTimeout(r, 3000))
const [txId2, block2, order2, docId2] = await addDoc(
collection,
{
city: 'beijing',
author: 'imotai',
age: 10,
}
)

const [txId3, block3, order3, docId3] = await addDoc(
collection2,
{
city: 'beijing2',
author: 'imotai1',
age: 1,
}
)
await new Promise((r) => setTimeout(r, 3000))
{
const queryStr = '/[city = beijing]'
const resultSet = await queryDoc<Profile>(
collection,
queryStr
)
expect(1).toBe(resultSet.docs.length)
expect(resultSet.docs[0].doc.city).toBe('beijing')
expect(resultSet.docs[0].doc.author).toBe('imotai')
expect(resultSet.docs[0].doc.age).toBe(10)
expect(resultSet.docs[0].id).toBe(docId2)
}
{
const queryStr = '/[city = beijing]'
const resultSet = await queryDoc<Profile>(
collection2,
queryStr
)
expect(1).toBe(resultSet.docs.length)
expect(resultSet.docs[0].doc.city).toBe('beijing')
expect(resultSet.docs[0].doc.author).toBe('imotai')
expect(resultSet.docs[0].doc.age).toBe(10)
expect(resultSet.docs[0].id).toBe(docId2)
}

try {
const [txId, block, order] = await deleteDoc(collection, [
docId3,
])
fail('should not be here')
} catch (e) {
expect(decodeURI(e.message)).toBe('fail to verify the owner with error doc owner is not the sender')
}
// verify docId3 is not deleted
await new Promise((r) => setTimeout(r, 3000))
{
const queryStr = '/[city = beijing2]'
const resultSet = await queryDoc<Profile>(
collection,
queryStr
)
expect(1).toBe(resultSet.docs.length)
expect(resultSet.docs[0].id).toBe(docId3)
}
const [txId, block, order] = await deleteDoc(collection2, [
docId3,
])
await new Promise((r) => setTimeout(r, 3000))
{
const queryStr = '/[city = beijing2]'
const resultSet = await queryDoc<Profile>(
collection,
queryStr
)
expect(0).toBe(resultSet.docs.length)
}

// update doc fail with ownership verify error
try {
const [txId, block, order] = await updateDoc(
collection2,
docId2,
{
city: 'beijing_new',
author: 'imotai_new',
age: 3,
},
[]
)
} catch (e) {
expect(decodeURI(e.message)).toBe('fail to verify the owner with error doc owner is not the sender')
}
// verify docId2 is not updated
await new Promise((r) => setTimeout(r, 3000))
{
const queryStr = '/[city = beijing]'
const resultSet = await queryDoc<Profile>(
collection,
queryStr
)
expect(1).toBe(resultSet.docs.length)
expect(resultSet.docs[0].id).toBe(docId2)
}
}
} catch (e) {
console.log(e)
Expand Down