loc-chain-backend/test_clients.js

83 lines
2.3 KiB
JavaScript
Raw Normal View History

const fs = require('fs')
const openpgp = require('openpgp')
const bcrypt = require('bcrypt')
class TestClient {
constructor(keyName, id) {
this.keyName = keyName
this.id = id
this.publicKey = fs.readFileSync(`./${this.keyName}.public.gpg.key`).toString('utf-8')
this.privateKey = fs.readFileSync(`./${this.keyName}.private.gpg.key`).toString('utf-8')
this.salt = fs.readFileSync(`./${this.keyName}.salt`).toString('utf-8')
}
getHash(otherClientId) {
const [lesserId, greaterId] = [otherClientId, this.id].sort()
return bcrypt.hashSync(`${lesserId}-${greaterId}`, this.salt)
}
buildExposure() {
return {
clientID: this.id,
timestamp: (new Date).getTime(),
}
}
async interactWith(otherClient) {
const hash = otherClient.getHash(this.id)
const message = openpgp.Message.fromText(hash)
const signature = (await openpgp.sign({
message,
privateKeys: await openpgp.readKey({
armoredKey: this.privateKey
})
}))
return {
combinedHash: hash,
timestamp: (new Date).getTime(),
encodedGPSLocation: JSON.stringify({ none: true }),
partnerPublicKey: otherClient.publicKey,
validationSignature: signature
}
}
}
;(async () => {
const client_a = new TestClient('test_a', 'a893irwe')
const client_b = new TestClient('test_b', 'bawesfdi')
const a_transact = await client_a.interactWith(client_b)
const b_transact = await client_b.interactWith(client_a)
// Simulate an encounter between A and B
console.log(a_transact, b_transact)
postTo('/api/v1/encounter', a_transact)
postTo('/api/v1/encounter', b_transact)
// Simulate B being exposed
const b_exposure = client_b.buildExposure()
console.log(b_exposure)
postTo('/api/v1/exposure', b_exposure)
})()
function postTo(endpoint, data) {
data = JSON.stringify(data)
const options = {
hostname: 'localhost',
port: 8000,
path: endpoint,
method: 'POST',
headers: {
'content-type': 'application/json',
'content-length': data.length,
}
}
const req = require('http').request(options)
req.write(data)
req.end()
}