ncsbe-lib
Version:
JavaScript library for working with North Carolina State Board of Elections (NCSBE) historical election data
211 lines (159 loc) • 7.17 kB
JavaScript
import { execSync } from 'child_process';
import fs from 'fs';
import path from 'path';
import { fileURLToPath } from 'url';
const __filename = fileURLToPath(import.meta.url);
const __dirname = path.dirname(__filename);
const TEST_DIR = path.join(__dirname, 'test-package');
console.log(TEST_DIR);
try {
// Run 'npm run build' in the *parent* directory (i.e., the package root)
console.log('🛠️ Building the package...');
execSync('npm run build', {
stdio: 'inherit',
cwd: path.join(__dirname, '..'),
});
// Run `npm pack` in the parent folder so the .tgz file lands there
console.log('📦 Packing the package...');
const packOutput = execSync(
'HUSKY_SKIP_HOOKS=1 npm pack --ignore-scripts',
{
encoding: 'utf-8',
cwd: path.join(__dirname, '..'),
},
);
console.log('npm pack output:', packOutput);
// Find the line that ends with ".tgz"
const tgzLine = packOutput
.split(/\r?\n/)
.map((l) => l.trim())
.find((line) => line.endsWith('.tgz'));
if (!tgzLine) {
throw new Error(
`Could not find .tgz file in npm pack output:\n${packOutput}`,
);
}
// The .tgz is in the parent folder, so join(__dirname, '..', tgzLine)
const tarballPath = path.join(__dirname, '..', tgzLine);
console.log(`✅ Package created: ${tarballPath}`);
// Create a new folder called 'test-package' to store new Node.js folder and test file.
console.log(`📂 Creating test directory: ${TEST_DIR}...`);
if (fs.existsSync(TEST_DIR)) {
fs.rmSync(TEST_DIR, { recursive: true });
}
fs.mkdirSync(TEST_DIR);
// Initialize a Node.js project in the test-package folder
console.log('📜 Initializing new package.json...');
execSync('npm init -y', { cwd: TEST_DIR, stdio: 'inherit' });
// Set the type of the Node.js project to be "module"
console.log('📝 Setting package type to module...');
const packageJsonPath = path.join(TEST_DIR, 'package.json');
const packageJson = JSON.parse(fs.readFileSync(packageJsonPath, 'utf8'));
packageJson.type = 'module';
fs.writeFileSync(packageJsonPath, JSON.stringify(packageJson, null, 2));
// Install the local package
console.log('📥 Installing packed package...');
execSync(`npm install "${tarballPath}"`, {
cwd: TEST_DIR,
stdio: 'inherit',
});
// Create a test.js file that imports the library and runs some tests on it
console.log('📝 Creating test.js file...');
fs.writeFileSync(
path.join(TEST_DIR, 'test.js'),
`
import { NCSBE } from "ncsbe-lib";
async function runTests() {
console.log(" ");
console.log("---------------------");
console.log(" ");
console.log("Running NCSBE Tests...");
console.log(" ");
console.log("---------------------");
console.log(" ");
const ncsbe = new NCSBE("2024-11-05");
console.log("Fetching election data...");
await ncsbe.initialize();
console.log("Data initialized.");
console.log(" ");
console.log("---------------------");
console.log(" ");
console.log("Fetch first 3 entries of entire dataset:");
console.log(ncsbe.getDataset().slice(0, 3));
console.log(" ");
console.log("---------------------");
console.log(" ");
const contests = ncsbe.listContests().slice(0, 5);
console.log("First 5 Contests:", contests);
console.log(" ");
console.log("---------------------");
console.log(" ");
const testContest = contests[4];
console.log("Testing contest:", testContest);
console.log(" ");
console.log("---------------------");
console.log(" ");
const counties = ncsbe.listCounties(testContest);
console.log("Counties participating:", counties);
console.log(" ");
console.log("---------------------");
console.log(" ");
if (counties.length > 0) {
const testCounty = counties[0];
console.log("Testing county:", testCounty);
const precincts = ncsbe.listPrecincts(testContest, testCounty);
console.log("Precincts in", testCounty, ":", precincts);
}
const candidates = ncsbe.listCandidates(testContest);
console.log("Candidates in", testContest, ":", candidates);
console.log(" ");
console.log("---------------------");
console.log(" ");
if (candidates.length > 0) {
const testCandidate = candidates[8];
console.log("Testing candidate:", testCandidate);
console.log(" ");
console.log("---------------------");
console.log(" ");
const voteTotal = ncsbe.getCandidateVoteTotal(testContest, testCandidate);
console.log(testCandidate + "Total Votes:", voteTotal);
console.log(" ");
console.log("---------------------");
console.log(" ");
const votePercentage = ncsbe.getCandidateVotePercentage(testContest, testCandidate);
console.log(testCandidate + "Vote Percentage:", votePercentage.toFixed(2) + "%");
console.log(" ");
console.log("---------------------");
console.log(" ");
const candidateContests = ncsbe.getContestsByCandidate(testCandidate);
console.log("Contests for", testCandidate, ":", candidateContests.map(c => c.contestName));
console.log(" ");
console.log("---------------------");
console.log(" ");
const candidateInfo = ncsbe.getCandidateInfo(testContest, testCandidate);
console.log("Candidate Info for", testCandidate, ":", candidateInfo);
console.log(" ");
console.log("---------------------");
console.log(" ");
}
const winner = ncsbe.getContestWinner(testContest);
console.log("Winner of", testContest, ":", winner);
console.log(" ");
console.log("---------------------");
console.log(" ");
console.log("All tests completed successfully.");
}
runTests().catch(error => console.error("Test failed:", error));
`,
);
console.log('🚀 Running test.js...');
execSync('node test.js', { cwd: TEST_DIR, stdio: 'inherit' });
// Remove all files related to this test
console.log('🧹 Cleaning up...');
fs.rmSync(TEST_DIR, { recursive: true });
fs.unlinkSync(tarballPath);
console.log('✅ Test completed successfully!');
} catch (error) {
console.error('❌ Error during test:', error.message);
process.exit(1);
}