stanford-oval / thingpedia-common-devices / 992

pending completion

Build # 992

Build Type

Pull #308

travis-ci-com

Committed by

web-flow

Commit Message

Merge 9a1e45cab into 78d38911b

Pull Request Pull Request #308: Add covid-vaccine skill

Run Details

409 of 783 branches covered (52.23%)

Branch coverage included in aggregate %.

11 of 33 new or added lines in 1 file covered. (33.33%)

1 existing line in 1 file now uncovered.

759 of 1236 relevant lines covered (61.41%)

659.93 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

39.02

/main/org.thingpedia.covid-vaccine/index.js

// -*- mode: js; indent-tabs-mode: nil; js-basic-offset: 4 -*-
//
// Copyright 2021 Chaofei Fan <stfan@stanford.edu>
//
// See LICENSE for details
"use strict";

const Tp = require('thingpedia');
const MongoClient = require('mongodb');

const DB_URL = process.env.MONGODB_URL || "mongodb://localhost:27017";
const DB_NAME = "pharmacy_vaccines";
const COLLECTION_NAME = "appointments";

function prettyprintAddress(appointment) {
    return [
        appointment.properties.name,
        appointment.properties.address,
        appointment.properties.city,
        appointment.properties.state,
        appointment.properties.postal_code,
    ].filter((i) => i.length > 0).join(', ');
}

module.exports = class COVIDVaccineAPIDevice extends Tp.BaseDevice {

    async _mongodb_client() {
        console.log(`Connect to ${DB_URL}`);
        return new MongoClient(DB_URL, {
            useNewUrlParser: true,
            useUnifiedTopology: true,
        });
    }

    async get_appointment({ location, distance }) {
        if (distance === undefined || distance === null)
            distance = 10.0;
        console.log(location, distance);

        // TODO: Find a better way to mock.
        if (process.env.CI) {
            return [
                {
                    id: new Tp.Value.Entity("0", "Safeway"),
                    geo: new Tp.Value.Location(37, -122, "Safeway, Palo Alto"),
                    link: 'http://www.safeway.com'
                },
                {
                    id: new Tp.Value.Entity("1", "Walmart"),
                    geo: new Tp.Value.Location(37, -122, "Walmart, Palo Alto"),
                    link: 'http://www.walmart.com'
                }
            ];
        }

        const client = await this._mongodb_client();
        try {
            await client.connect();
            const db = client.db(DB_NAME);
            const coll = db.collection(COLLECTION_NAME);

            const query = {
                "geometry.coordinates": {
                    $near: {
                        $geometry: {
                            type: "Point",
                            coordinates: [location.x, location.y]
                        },
                        $maxDistance: distance * 1610,  // Mile to meter
                    },
                },
                "properties.appointments_available": true
            };

            const cursor = coll.find(query);
            let appointments = await cursor.toArray();
            appointments = appointments.map((appt) => {
                const id = new Tp.Value.Entity(appt._id.toString(), appt.properties.name);
                const geo = new Tp.Value.Location(
                    appt.geometry.coordinates[1],
                    appt.geometry.coordinates[0],
                    prettyprintAddress(appt));
                return {
                    id: id,
                    geo: geo,
                    link: appt.properties.url
                };
            });

            await cursor.close();
            console.log(appointments);
            return appointments;
        } catch (error) {
            console.error(error);
            throw new Error("Failed to get vaccine appointments");
        } finally {
            await client.close();
        }
    }
};

1	// -- mode: js; indent-tabs-mode: nil; js-basic-offset: 4 --
2	//
3	// Copyright 2021 Chaofei Fan <stfan@stanford.edu>
4	//
5	// See LICENSE for details
6	"use strict";
7
8	const Tp = require('thingpedia');	2✔
9	const MongoClient = require('mongodb');	2✔
10
11	const DB_URL = process.env.MONGODB_URL \|\| "mongodb://localhost:27017";	2✔
12	const DB_NAME = "pharmacy_vaccines";	2✔
13	const COLLECTION_NAME = "appointments";	2✔
14
15	function prettyprintAddress(appointment) {
NEW 16	return [	×
17	appointment.properties.name,
18	appointment.properties.address,
19	appointment.properties.city,
20	appointment.properties.state,
21	appointment.properties.postal_code,
NEW 22	].filter((i) => i.length > 0).join(', ');	×
23	}
24
25	module.exports = class COVIDVaccineAPIDevice extends Tp.BaseDevice {	2✔
26
27	async _mongodb_client() {
NEW 28	console.log(`Connect to ${DB_URL}`);	×
NEW 29	return new MongoClient(DB_URL, {	×
30	useNewUrlParser: true,
31	useUnifiedTopology: true,
32	});
33	}
34
35	async get_appointment({ location, distance }) {
36	if (distance === undefined \|\| distance === null)	2!
37	distance = 10.0;	2✔
38	console.log(location, distance);	2✔
39
40	// TODO: Find a better way to mock.
41	if (process.env.CI) {	2!
42	return [	2✔
43	{
44	id: new Tp.Value.Entity("0", "Safeway"),
45	geo: new Tp.Value.Location(37, -122, "Safeway, Palo Alto"),
46	link: 'http://www.safeway.com'
47	},
48	{
49	id: new Tp.Value.Entity("1", "Walmart"),
50	geo: new Tp.Value.Location(37, -122, "Walmart, Palo Alto"),
51	link: 'http://www.walmart.com'
52	}
53	];
54	}
55
NEW 56	const client = await this._mongodb_client();	×
NEW 57	try {	×
NEW 58	await client.connect();	×
NEW 59	const db = client.db(DB_NAME);	×
NEW 60	const coll = db.collection(COLLECTION_NAME);	×
61
NEW 62	const query = {	×
63	"geometry.coordinates": {
64	$near: {
65	$geometry: {
66	type: "Point",
67	coordinates: [location.x, location.y]
68	},
69	$maxDistance: distance * 1610, // Mile to meter
70	},
71	},
72	"properties.appointments_available": true
73	};
74
NEW 75	const cursor = coll.find(query);	×
NEW 76	let appointments = await cursor.toArray();	×
NEW 77	appointments = appointments.map((appt) => {	×
NEW 78	const id = new Tp.Value.Entity(appt._id.toString(), appt.properties.name);	×
NEW 79	const geo = new Tp.Value.Location(	×
80	appt.geometry.coordinates[1],
81	appt.geometry.coordinates[0],
82	prettyprintAddress(appt));
NEW 83	return {	×
84	id: id,
85	geo: geo,
86	link: appt.properties.url
87	};
88	});
89
NEW 90	await cursor.close();	×
NEW 91	console.log(appointments);	×
NEW 92	return appointments;	×
93	} catch (error) {
NEW 94	console.error(error);	×
NEW 95	throw new Error("Failed to get vaccine appointments");	×
96	} finally {
NEW 97	await client.close();	×
98	}
99	}
100	};

stanford-oval / thingpedia-common-devices / 992

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous