Added CspCensus in fetchElection to avoid requests to the census API

src/services/election.ts

@@ -2,12 +2,13 @@ import { Service, ServiceProperties } from './service';
 import {
   ArchivedElection,
   Census,
+  CspCensus,
   InvalidElection,
   PublishedCensus,
   PublishedElection,
   UnpublishedElection,
 } from '../types';
-import { AccountAPI, ElectionAPI, IElectionCreateResponse, IElectionKeysResponse } from '../api';
+import { AccountAPI, CensusTypeEnum, ElectionAPI, IElectionCreateResponse, IElectionKeysResponse } from '../api';
 import { CensusService } from './census';
 import { allSettled } from '../util/promise';
 import invariant from 'tiny-invariant';
@@ -97,6 +98,15 @@ export class ElectionService extends Service implements ElectionServicePropertie
       );
   }
 
+  private buildCensus(electionInfo): Promise<PublishedCensus | ArchivedCensus> {
+    if (electionInfo.census.censusOrigin === CensusTypeEnum.OFF_CHAIN_CA) {
+      return Promise.resolve(new CspCensus(electionInfo.census.censusRoot, electionInfo.census.censusURL));
+    }
+    return electionInfo.fromArchive
+      ? Promise.resolve(new ArchivedCensus(electionInfo.census.censusRoot, electionInfo.census.censusURL))
+      : this.buildPublishedCensus(electionInfo);
+  }
+
   /**
    * Fetches info about an election.
    *
@@ -122,9 +132,7 @@ export class ElectionService extends Service implements ElectionServicePropertie
       meta: electionInfo.metadata?.meta,
       startDate: electionInfo.startDate,
       endDate: electionInfo.endDate,
-      census: electionInfo.fromArchive
-        ? new ArchivedCensus(electionInfo.census.censusRoot, electionInfo.census.censusURL)
-        : await this.buildPublishedCensus(electionInfo),
+      census: await this.buildCensus(electionInfo),
       maxCensusSize: electionInfo.census.maxCensusSize,
       manuallyEnded: electionInfo.manuallyEnded,
       fromArchive: electionInfo.fromArchive,

test/integration/csp.test.ts

@@ -77,6 +77,7 @@ describe('CSP tests', () => {
         expect(election.title).toEqual(election.title);
         expect(election.voteCount).toEqual(numVotes);
         expect(election.results[0][0]).toEqual(election.results[0][1]);
+        expect(election.census).toBeInstanceOf(CspCensus);
         expect(election.census.size).toBeUndefined();
         expect(election.census.weight).toBeUndefined();
       });