Add init faucet script

tests/integration-tests/src/flows/faucet/initFaucet.ts

@@ -0,0 +1,45 @@
+import { FlowProps } from '../../Flow'
+import { extendDebug } from '../../Debugger'
+import { FixtureRunner } from '../../Fixture'
+import { BuyMembershipHappyCaseFixture, TransferInvitesHappyCaseFixture } from '../../fixtures/membership'
+import { SetBudgetFixture } from '../../fixtures/workingGroups/SetBudgetFixture'
+import _ from 'lodash'
+import BN from 'bn.js'
+
+export default async function initFaucet({ api, env, query }: FlowProps): Promise<void> {
+  const debug = extendDebug('flow:initFaucet')
+  debug('Started')
+
+  // Get membership working group leader
+  const membershipWorkingGroup = 'membershipWorkingGroup'
+  const [, membershipLeader] = await api.getLeader(membershipWorkingGroup)
+
+  // Grant lead invitation quota
+  const proposalDetails = api.createType('ProposalDetails', { SetMembershipLeadInvitationQuota: 50 })
+
+  // The membership working group should have a budget allocated
+  const budgets: BN[] = [new BN(1000000)]
+  const setGroupBudgetFixture = new SetBudgetFixture(api, query, membershipWorkingGroup, budgets)
+  await new FixtureRunner(setGroupBudgetFixture).runWithQueryNodeChecks()
+
+  // Create a membership account for faucet
+  const faucetAccounts = (await api.createKeyPairs(1)).map(({ key }) => key.address)
+  const happyCaseFixture = new BuyMembershipHappyCaseFixture(api, query, faucetAccounts)
+  await new FixtureRunner(happyCaseFixture).runWithQueryNodeChecks()
+  const [faucetMemberId] = happyCaseFixture.getCreatedMembers()
+
+  // Give the faucet member accounts some funds (they need some funds for extrinsics)
+  await api.treasuryTransferBalanceToAccounts(faucetAccounts, new BN(200))
+
+  // Use the above membershipLeader to give faucet account permission to invite other members
+  // const transferInvitesHappyCaseFixture = new TransferInvitesHappyCaseFixture(
+  //   api,
+  //   query,
+  //   { memberId: membershipLeader.member_id, account: membershipLeader.role_account_id.toString() },
+  //   { memberId: faucetMemberId, account: faucetAccounts[0] },
+  //   10
+  // )
+  // await new FixtureRunner(transferInvitesHappyCaseFixture).runWithQueryNodeChecks()
+
+  debug('Done')
+}

tests/integration-tests/src/scenarios/setupNewChain.ts

@@ -1,6 +1,7 @@
 import electCouncil from '../flows/council/elect'
 import leaderSetup from '../flows/working-groups/leadOpening'
 import updateAccountsFlow from '../misc/updateAllWorkerRoleAccountsFlow'
+import initFaucet from '../flows/faucet/initFaucet'
 import initStorage, { singleBucketConfig as defaultStorageConfig } from '../flows/storage/initStorage'
 import initDistribution, { singleBucketConfig as defaultDistributionConfig } from '../flows/storage/initDistribution'
 import { scenario } from '../Scenario'
@@ -15,6 +16,10 @@ scenario(async ({ job }) => {
     job('initialize distribution system', initDistribution(defaultDistributionConfig)).requires(updateWorkerAccounts)
   }
 
+  if (!process.env.SKIP_FAUCET_INIT) {
+    job('Initialize Faucet', initFaucet).requires(leads)
+  }
+
   // TODO: Mock content
   // assign members known accounts?
   // assign council known accounts?