integration_tests.rs 40 KB
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
// Copyright 2019-2021 Parity Technologies (UK) Ltd.
// This file is part of Polkadot.

// Polkadot is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.

// Polkadot is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.

// You should have received a copy of the GNU General Public License
// along with Polkadot.  If not, see <http://www.gnu.org/licenses/>.

//! Mocking utilities for testing with real pallets.

Shawn Tabrizi's avatar
Shawn Tabrizi committed
19
20
21
22
23
use crate::{
	auctions, crowdloan, paras_registrar,
	slot_range::SlotRange,
	slots,
	traits::{AuctionStatus, Auctioneer, Registrar as RegistrarT},
24
25
};
use frame_support::{
Shawn Tabrizi's avatar
Shawn Tabrizi committed
26
27
28
	assert_noop, assert_ok, parameter_types,
	traits::{Currency, GenesisBuild, KeyOwnerProofSystem, OnFinalize, OnInitialize},
	PalletId,
29
30
};
use frame_support_test::TestRandomness;
Shawn Tabrizi's avatar
Shawn Tabrizi committed
31
32
33
use frame_system::EnsureRoot;
use primitives::v1::{
	BlockNumber, HeadData, Header, Id as ParaId, ValidationCode, LOWEST_PUBLIC_ID,
34
};
Shawn Tabrizi's avatar
Shawn Tabrizi committed
35
36
37
38
39
40
use runtime_parachains::{configuration, paras, shared, Origin as ParaOrigin, ParaLifecycle};
use sp_core::{crypto::KeyTypeId, H256};
use sp_io::TestExternalities;
use sp_keystore::{testing::KeyStore, KeystoreExt};
use sp_runtime::traits::{BlakeTwo256, IdentityLookup, One};
use sp_std::sync::Arc;
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55

type UncheckedExtrinsic = frame_system::mocking::MockUncheckedExtrinsic<Test>;
type Block = frame_system::mocking::MockBlock<Test>;

type AccountId = u32;
type Balance = u32;
type Moment = u32;

frame_support::construct_runtime!(
	pub enum Test where
		Block = Block,
		NodeBlock = Block,
		UncheckedExtrinsic = UncheckedExtrinsic,
	{
		// System Stuff
56
57
58
		System: frame_system::{Pallet, Call, Config, Storage, Event<T>},
		Balances: pallet_balances::{Pallet, Call, Storage, Config<T>, Event<T>},
		Babe: pallet_babe::{Pallet, Call, Storage, Config, ValidateUnsigned},
59
60

		// Parachains Runtime
61
		Configuration: configuration::{Pallet, Call, Storage, Config<T>},
ferrell-code's avatar
ferrell-code committed
62
		Paras: paras::{Pallet, Origin, Call, Storage, Event, Config},
63
		ParasShared: shared::{Pallet, Call, Storage},
64
65

		// Para Onboarding Pallets
66
67
68
69
		Registrar: paras_registrar::{Pallet, Call, Storage, Event<T>},
		Auctions: auctions::{Pallet, Call, Storage, Event<T>},
		Crowdloan: crowdloan::{Pallet, Call, Storage, Event<T>},
		Slots: slots::{Pallet, Call, Storage, Event<T>},
70
71
72
	}
);

Shawn Tabrizi's avatar
Shawn Tabrizi committed
73
use crate::{auctions::Error as AuctionsError, crowdloan::Error as CrowdloanError};
74
75
76
77
78
79
80
81

parameter_types! {
	pub const BlockHashCount: u32 = 250;
	pub BlockWeights: frame_system::limits::BlockWeights =
		frame_system::limits::BlockWeights::simple_max(4 * 1024 * 1024);
}

impl frame_system::Config for Test {
82
	type BaseCallFilter = frame_support::traits::Everything;
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
	type BlockWeights = BlockWeights;
	type BlockLength = ();
	type DbWeight = ();
	type Origin = Origin;
	type Call = Call;
	type Index = u64;
	type BlockNumber = BlockNumber;
	type Hash = H256;
	type Hashing = BlakeTwo256;
	type AccountId = AccountId;
	type Lookup = IdentityLookup<AccountId>;
	type Header = Header;
	type Event = Event;
	type BlockHashCount = BlockHashCount;
	type Version = ();
	type PalletInfo = PalletInfo;
	type AccountData = pallet_balances::AccountData<Balance>;
	type OnNewAccount = ();
	type OnKilledAccount = ();
	type SystemWeightInfo = ();
	type SS58Prefix = ();
104
	type OnSetCode = ();
105
106
107
108
109
110
111
112
113
114
115
116
}

parameter_types! {
	pub const EpochDuration: u64 = 10;
	pub const ExpectedBlockTime: Moment = 6_000;
	pub const ReportLongevity: u64 = 10;
}

impl pallet_babe::Config for Test {
	type EpochDuration = EpochDuration;
	type ExpectedBlockTime = ExpectedBlockTime;
	type EpochChangeTrigger = pallet_babe::ExternalTrigger;
117
	type DisabledValidators = ();
118
	type KeyOwnerProofSystem = ();
Shawn Tabrizi's avatar
Shawn Tabrizi committed
119
120
121
122
	type KeyOwnerProof = <Self::KeyOwnerProofSystem as KeyOwnerProofSystem<(
		KeyTypeId,
		pallet_babe::AuthorityId,
	)>>::Proof;
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
	type KeyOwnerIdentification = <Self::KeyOwnerProofSystem as KeyOwnerProofSystem<(
		KeyTypeId,
		pallet_babe::AuthorityId,
	)>>::IdentificationTuple;
	type HandleEquivocation = ();
	type WeightInfo = ();
}

parameter_types! {
	pub const MinimumPeriod: Moment = 6_000 / 2;
}

impl pallet_timestamp::Config for Test {
	type Moment = Moment;
	type OnTimestampSet = ();
	type MinimumPeriod = MinimumPeriod;
	type WeightInfo = ();
}

parameter_types! {
	pub static ExistentialDeposit: Balance = 1;
Gavin Wood's avatar
Gavin Wood committed
144
	pub const MaxReserves: u32 = 50;
145
146
147
148
149
150
151
152
153
154
}

impl pallet_balances::Config for Test {
	type MaxLocks = ();
	type Balance = Balance;
	type Event = Event;
	type DustRemoval = ();
	type ExistentialDeposit = ExistentialDeposit;
	type AccountStore = System;
	type WeightInfo = ();
Gavin Wood's avatar
Gavin Wood committed
155
156
	type MaxReserves = MaxReserves;
	type ReserveIdentifier = [u8; 8];
157
158
}

159
160
161
impl configuration::Config for Test {
	type WeightInfo = configuration::weights::WeightInfo<Test>;
}
162

Shawn Tabrizi's avatar
Shawn Tabrizi committed
163
impl shared::Config for Test {}
164
165
166

impl paras::Config for Test {
	type Origin = Origin;
167
	type Event = Event;
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
}

parameter_types! {
	pub const ParaDeposit: Balance = 500;
	pub const DataDepositPerByte: Balance = 1;
}

impl paras_registrar::Config for Test {
	type Event = Event;
	type OnSwap = (Crowdloan, Slots);
	type ParaDeposit = ParaDeposit;
	type DataDepositPerByte = DataDepositPerByte;
	type Currency = Balances;
	type Origin = Origin;
	type WeightInfo = crate::paras_registrar::TestWeightInfo;
}

parameter_types! {
	pub const EndingPeriod: BlockNumber = 10;
187
	pub const SampleLength: BlockNumber = 1;
188
189
190
191
192
}

impl auctions::Config for Test {
	type Event = Event;
	type Leaser = Slots;
193
	type Registrar = Registrar;
194
	type EndingPeriod = EndingPeriod;
195
	type SampleLength = SampleLength;
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
	type Randomness = TestRandomness<Self>;
	type InitiateOrigin = EnsureRoot<AccountId>;
	type WeightInfo = crate::auctions::TestWeightInfo;
}

parameter_types! {
	pub const LeasePeriod: BlockNumber = 100;
}

impl slots::Config for Test {
	type Event = Event;
	type Currency = Balances;
	type Registrar = Registrar;
	type LeasePeriod = LeasePeriod;
	type WeightInfo = crate::slots::TestWeightInfo;
}

parameter_types! {
Shawn Tabrizi's avatar
Shawn Tabrizi committed
214
	pub const CrowdloanId: PalletId = PalletId(*b"py/cfund");
215
216
217
	pub const SubmissionDeposit: Balance = 100;
	pub const MinContribution: Balance = 1;
	pub const RemoveKeysLimit: u32 = 100;
218
	pub const MaxMemoLength: u8 = 32;
219
220
221
222
}

impl crowdloan::Config for Test {
	type Event = Event;
Shawn Tabrizi's avatar
Shawn Tabrizi committed
223
	type PalletId = CrowdloanId;
224
225
226
227
228
	type SubmissionDeposit = SubmissionDeposit;
	type MinContribution = MinContribution;
	type RemoveKeysLimit = RemoveKeysLimit;
	type Registrar = Registrar;
	type Auctioneer = Auctions;
229
	type MaxMemoLength = MaxMemoLength;
230
231
232
233
234
	type WeightInfo = crate::crowdloan::TestWeightInfo;
}

/// Create a new set of test externalities.
pub fn new_test_ext() -> TestExternalities {
235
	let mut t = frame_system::GenesisConfig::default().build_storage::<Test>().unwrap();
236
237
	GenesisBuild::<Test>::assimilate_storage(
		&configuration::GenesisConfig {
Shawn Tabrizi's avatar
Shawn Tabrizi committed
238
239
240
241
			config: configuration::HostConfiguration {
				max_code_size: 2 * 1024 * 1024,      // 2 MB
				max_head_data_size: 1 * 1024 * 1024, // 1 MB
				..Default::default()
242
			},
Shawn Tabrizi's avatar
Shawn Tabrizi committed
243
244
245
246
		},
		&mut t,
	)
	.unwrap();
247
248
249
250
251
252
253
254
255
256
257
	let keystore = KeyStore::new();
	let mut ext: sp_io::TestExternalities = t.into();
	ext.register_extension(KeystoreExt(Arc::new(keystore)));
	ext.execute_with(|| System::set_block_number(1));
	ext
}

const BLOCKS_PER_SESSION: u32 = 10;

fn maybe_new_session(n: u32) {
	if n % BLOCKS_PER_SESSION == 0 {
Shawn Tabrizi's avatar
Shawn Tabrizi committed
258
		shared::Pallet::<Test>::set_session_index(shared::Pallet::<Test>::session_index() + 1);
259
260
261
262
263
264
265
266
267
		Paras::test_on_new_session();
	}
}

fn test_genesis_head(size: usize) -> HeadData {
	HeadData(vec![0u8; size])
}

fn test_validation_code(size: usize) -> ValidationCode {
asynchronous rob's avatar
asynchronous rob committed
268
	let validation_code = vec![0u8; size as usize];
269
270
271
272
273
274
275
276
277
278
279
	ValidationCode(validation_code)
}

fn para_origin(id: u32) -> ParaOrigin {
	ParaOrigin::Parachain(id.into())
}

fn run_to_block(n: u32) {
	assert!(System::block_number() < n);
	while System::block_number() < n {
		let block_number = System::block_number();
280
		AllPallets::on_finalize(block_number);
281
282
283
284
		System::on_finalize(block_number);
		System::set_block_number(block_number + 1);
		System::on_initialize(block_number + 1);
		maybe_new_session(block_number + 1);
285
		AllPallets::on_initialize(block_number + 1);
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
	}
}

fn run_to_session(n: u32) {
	let block_number = BLOCKS_PER_SESSION * n;
	run_to_block(block_number);
}

fn last_event() -> Event {
	System::events().pop().expect("Event expected").event
}

#[test]
fn basic_end_to_end_works() {
	new_test_ext().execute_with(|| {
301
302
		let para_1 = LOWEST_PUBLIC_ID;
		let para_2 = LOWEST_PUBLIC_ID + 1;
303
304
		assert!(System::block_number().is_one());
		// User 1 and 2 will own parachains
305
306
		Balances::make_free_balance_be(&1, 1_000_000_000);
		Balances::make_free_balance_be(&2, 1_000_000_000);
307
308
309
		// First register 2 parathreads
		let genesis_head = Registrar::worst_head_data();
		let validation_code = Registrar::worst_validation_code();
310
		assert_ok!(Registrar::reserve(Origin::signed(1)));
311
312
		assert_ok!(Registrar::register(
			Origin::signed(1),
313
			ParaId::from(para_1),
314
315
316
			genesis_head.clone(),
			validation_code.clone(),
		));
317
		assert_ok!(Registrar::reserve(Origin::signed(2)));
318
319
		assert_ok!(Registrar::register(
			Origin::signed(2),
320
			ParaId::from(2001),
321
322
323
324
325
			genesis_head,
			validation_code,
		));

		// Paras should be onboarding
326
327
		assert_eq!(Paras::lifecycle(ParaId::from(para_1)), Some(ParaLifecycle::Onboarding));
		assert_eq!(Paras::lifecycle(ParaId::from(para_2)), Some(ParaLifecycle::Onboarding));
328
329
330
331
332
333
334
335

		// Start a new auction in the future
		let duration = 99u32;
		let lease_period_index_start = 4u32;
		assert_ok!(Auctions::new_auction(Origin::root(), duration, lease_period_index_start));

		// 2 sessions later they are parathreads
		run_to_session(2);
336
337
		assert_eq!(Paras::lifecycle(ParaId::from(para_1)), Some(ParaLifecycle::Parathread));
		assert_eq!(Paras::lifecycle(ParaId::from(para_2)), Some(ParaLifecycle::Parathread));
338
339
340
341
342

		// Para 1 will bid directly for slot 1, 2
		// Open a crowdloan for Para 2 for slot 3, 4
		assert_ok!(Crowdloan::create(
			Origin::signed(2),
343
			ParaId::from(para_2),
Shawn Tabrizi's avatar
Shawn Tabrizi committed
344
			1_000,                        // Cap
345
346
			lease_period_index_start + 2, // First Slot
			lease_period_index_start + 3, // Last Slot
Shawn Tabrizi's avatar
Shawn Tabrizi committed
347
			200,                          // Block End
348
349
			None,
		));
350
		let crowdloan_account = Crowdloan::fund_account_id(ParaId::from(para_2));
351
352
353
354

		// Auction ending begins on block 100, so we make a bid before then.
		run_to_block(90);

355
356
		Balances::make_free_balance_be(&10, 1_000_000_000);
		Balances::make_free_balance_be(&20, 1_000_000_000);
357
358
359
360

		// User 10 will bid directly for parachain 1
		assert_ok!(Auctions::bid(
			Origin::signed(10),
361
			ParaId::from(para_1),
Shawn Tabrizi's avatar
Shawn Tabrizi committed
362
			1,                            // Auction Index
363
364
			lease_period_index_start + 0, // First Slot
			lease_period_index_start + 1, // Last slot
Shawn Tabrizi's avatar
Shawn Tabrizi committed
365
			910,                          // Amount
366
367
		));

368
		// User 2 will be a contribute to crowdloan for parachain 2
369
		Balances::make_free_balance_be(&2, 1_000_000_000);
370
		assert_ok!(Crowdloan::contribute(Origin::signed(2), ParaId::from(para_2), 920, None));
371
372
373
374
375

		// Auction ends at block 110
		run_to_block(109);
		assert_eq!(
			last_event(),
376
			crowdloan::Event::<Test>::HandleBidResult(ParaId::from(para_2), Ok(())).into(),
377
378
		);
		run_to_block(110);
379
		assert_eq!(last_event(), auctions::Event::<Test>::AuctionClosed(1).into());
380
381
382

		// Paras should have won slots
		assert_eq!(
383
			slots::Leases::<Test>::get(ParaId::from(para_1)),
384
385
386
387
			// -- 1 --- 2 --- 3 --------- 4 ------------ 5 --------
			vec![None, None, None, Some((10, 910)), Some((10, 910))],
		);
		assert_eq!(
388
			slots::Leases::<Test>::get(ParaId::from(para_2)),
389
			// -- 1 --- 2 --- 3 --- 4 --- 5 ---------------- 6 --------------------------- 7 ----------------
Shawn Tabrizi's avatar
Shawn Tabrizi committed
390
391
392
393
394
395
396
397
398
			vec![
				None,
				None,
				None,
				None,
				None,
				Some((crowdloan_account, 920)),
				Some((crowdloan_account, 920))
			],
399
400
		);

401
		// Should not be able to contribute to a winning crowdloan
402
		Balances::make_free_balance_be(&3, 1_000_000_000);
Shawn Tabrizi's avatar
Shawn Tabrizi committed
403
404
405
406
		assert_noop!(
			Crowdloan::contribute(Origin::signed(3), ParaId::from(2001), 10, None),
			CrowdloanError::<Test>::BidOrLeaseActive
		);
407

408
409
410
411
412
		// New leases will start on block 400
		let lease_start_block = 400;
		run_to_block(lease_start_block);

		// First slot, Para 1 should be transitioning to Parachain
Shawn Tabrizi's avatar
Shawn Tabrizi committed
413
414
415
416
		assert_eq!(
			Paras::lifecycle(ParaId::from(para_1)),
			Some(ParaLifecycle::UpgradingParathread)
		);
417
		assert_eq!(Paras::lifecycle(ParaId::from(para_2)), Some(ParaLifecycle::Parathread));
418
419
420

		// Two sessions later, it has upgraded
		run_to_block(lease_start_block + 20);
421
422
		assert_eq!(Paras::lifecycle(ParaId::from(para_1)), Some(ParaLifecycle::Parachain));
		assert_eq!(Paras::lifecycle(ParaId::from(para_2)), Some(ParaLifecycle::Parathread));
423
424
425

		// Second slot nothing happens :)
		run_to_block(lease_start_block + 100);
426
427
		assert_eq!(Paras::lifecycle(ParaId::from(para_1)), Some(ParaLifecycle::Parachain));
		assert_eq!(Paras::lifecycle(ParaId::from(para_2)), Some(ParaLifecycle::Parathread));
428
429
430

		// Third slot, Para 2 should be upgrading, and Para 1 is downgrading
		run_to_block(lease_start_block + 200);
Shawn Tabrizi's avatar
Shawn Tabrizi committed
431
432
433
434
435
436
437
438
		assert_eq!(
			Paras::lifecycle(ParaId::from(para_1)),
			Some(ParaLifecycle::DowngradingParachain)
		);
		assert_eq!(
			Paras::lifecycle(ParaId::from(para_2)),
			Some(ParaLifecycle::UpgradingParathread)
		);
439
440
441

		// Two sessions later, they have transitioned
		run_to_block(lease_start_block + 220);
442
443
		assert_eq!(Paras::lifecycle(ParaId::from(para_1)), Some(ParaLifecycle::Parathread));
		assert_eq!(Paras::lifecycle(ParaId::from(para_2)), Some(ParaLifecycle::Parachain));
444
445
446

		// Fourth slot nothing happens :)
		run_to_block(lease_start_block + 300);
447
448
		assert_eq!(Paras::lifecycle(ParaId::from(para_1)), Some(ParaLifecycle::Parathread));
		assert_eq!(Paras::lifecycle(ParaId::from(para_2)), Some(ParaLifecycle::Parachain));
449
450
451

		// Fifth slot, Para 2 is downgrading
		run_to_block(lease_start_block + 400);
452
		assert_eq!(Paras::lifecycle(ParaId::from(para_1)), Some(ParaLifecycle::Parathread));
Shawn Tabrizi's avatar
Shawn Tabrizi committed
453
454
455
456
		assert_eq!(
			Paras::lifecycle(ParaId::from(para_2)),
			Some(ParaLifecycle::DowngradingParachain)
		);
457
458
459

		// Two sessions later, Para 2 is downgraded
		run_to_block(lease_start_block + 420);
460
461
		assert_eq!(Paras::lifecycle(ParaId::from(para_1)), Some(ParaLifecycle::Parathread));
		assert_eq!(Paras::lifecycle(ParaId::from(para_2)), Some(ParaLifecycle::Parathread));
462
463
464
465
466
467
468
	});
}

#[test]
fn basic_errors_fail() {
	new_test_ext().execute_with(|| {
		assert!(System::block_number().is_one());
469
		let para_id = LOWEST_PUBLIC_ID;
470
		// Can't double register
471
472
		Balances::make_free_balance_be(&1, 1_000_000_000);
		Balances::make_free_balance_be(&2, 1_000_000_000);
473
474
475

		let genesis_head = Registrar::worst_head_data();
		let validation_code = Registrar::worst_validation_code();
476
		assert_ok!(Registrar::reserve(Origin::signed(1)));
477
478
		assert_ok!(Registrar::register(
			Origin::signed(1),
479
			para_id,
480
481
482
			genesis_head.clone(),
			validation_code.clone(),
		));
483
		assert_ok!(Registrar::reserve(Origin::signed(2)));
Shawn Tabrizi's avatar
Shawn Tabrizi committed
484
485
486
487
		assert_noop!(
			Registrar::register(Origin::signed(2), para_id, genesis_head, validation_code,),
			paras_registrar::Error::<Test>::NotOwner
		);
488
489
490
491
492
493
494

		// Start an auction
		let duration = 99u32;
		let lease_period_index_start = 4u32;
		assert_ok!(Auctions::new_auction(Origin::root(), duration, lease_period_index_start));

		// Cannot create a crowdloan if you do not own the para
Shawn Tabrizi's avatar
Shawn Tabrizi committed
495
496
497
498
499
500
501
502
503
504
505
506
		assert_noop!(
			Crowdloan::create(
				Origin::signed(2),
				para_id,
				1_000,                        // Cap
				lease_period_index_start + 2, // First Slot
				lease_period_index_start + 3, // Last Slot
				200,                          // Block End
				None,
			),
			crowdloan::Error::<Test>::InvalidOrigin
		);
507
508
509
510
511
512
513
514
515
	});
}

#[test]
fn competing_slots() {
	// This test will verify that competing slots, from different sources will resolve appropriately.
	new_test_ext().execute_with(|| {
		assert!(System::block_number().is_one());
		let max_bids = 10u32;
516
		let para_id = LOWEST_PUBLIC_ID;
517
518

		// Create n paras and owners
Shawn Tabrizi's avatar
Shawn Tabrizi committed
519
		for n in 1..=max_bids {
520
			Balances::make_free_balance_be(&n, 1_000_000_000);
521
522
			let genesis_head = Registrar::worst_head_data();
			let validation_code = Registrar::worst_validation_code();
523
			assert_ok!(Registrar::reserve(Origin::signed(n)));
524
525
			assert_ok!(Registrar::register(
				Origin::signed(n),
526
				para_id + n - 1,
527
528
529
530
531
532
				genesis_head,
				validation_code,
			));
		}

		// Start a new auction in the future
533
		let duration = 149u32;
534
535
536
		let lease_period_index_start = 4u32;
		assert_ok!(Auctions::new_auction(Origin::root(), duration, lease_period_index_start));

537
538
539
		// Paras should be onboarded
		run_to_block(20); // session 2

Shawn Tabrizi's avatar
Shawn Tabrizi committed
540
		for n in 1..=max_bids {
541
			// Increment block number
542
			run_to_block(System::block_number() + 10);
543
544
545
546

			Balances::make_free_balance_be(&(n * 10), n * 1_000);

			let (start, end) = match n {
Shawn Tabrizi's avatar
Shawn Tabrizi committed
547
548
549
550
551
552
553
554
555
				1 => (0, 0),
				2 => (0, 1),
				3 => (0, 2),
				4 => (0, 3),
				5 => (1, 1),
				6 => (1, 2),
				7 => (1, 3),
				8 => (2, 2),
				9 => (2, 3),
556
557
558
559
				10 => (3, 3),
				_ => panic!("test not meant for this"),
			};

560
			// Users will bid directly for parachain
561
562
			assert_ok!(Auctions::bid(
				Origin::signed(n * 10),
563
				para_id + n - 1,
Shawn Tabrizi's avatar
Shawn Tabrizi committed
564
				1,                                // Auction Index
565
				lease_period_index_start + start, // First Slot
Shawn Tabrizi's avatar
Shawn Tabrizi committed
566
567
				lease_period_index_start + end,   // Last slot
				n * 900,                          // Amount
568
569
570
			));
		}

571
572
		// Auction should be done after ending period
		run_to_block(160);
573
574
575
576
577

		// Appropriate Paras should have won slots
		// 900 + 4500 + 2x 8100 = 21,600
		// 900 + 4500 + 7200 + 9000 = 21,600
		assert_eq!(
578
			slots::Leases::<Test>::get(para_id),
579
580
581
582
			// -- 1 --- 2 --- 3 ---------- 4 ------
			vec![None, None, None, Some((10, 900))],
		);
		assert_eq!(
583
			slots::Leases::<Test>::get(para_id + 4),
584
585
586
587
588
			// -- 1 --- 2 --- 3 --- 4 ---------- 5 -------
			vec![None, None, None, None, Some((50, 4500))],
		);
		// TODO: Is this right?
		assert_eq!(
589
			slots::Leases::<Test>::get(para_id + 8),
590
591
592
593
594
595
596
597
598
599
600
601
			// -- 1 --- 2 --- 3 --- 4 --- 5 ---------- 6 --------------- 7 -------
			vec![None, None, None, None, None, Some((90, 8100)), Some((90, 8100))],
		);
	});
}

#[test]
fn competing_bids() {
	// This test will verify that competing bids, from different sources will resolve appropriately.
	new_test_ext().execute_with(|| {
		assert!(System::block_number().is_one());

602
		let start_para = LOWEST_PUBLIC_ID - 1;
603
		// Create 3 paras and owners
Shawn Tabrizi's avatar
Shawn Tabrizi committed
604
		for n in 1..=3 {
605
			Balances::make_free_balance_be(&n, 1_000_000_000);
606
607
			let genesis_head = Registrar::worst_head_data();
			let validation_code = Registrar::worst_validation_code();
608
			assert_ok!(Registrar::reserve(Origin::signed(n)));
609
610
			assert_ok!(Registrar::register(
				Origin::signed(n),
611
				ParaId::from(start_para + n),
612
613
614
				genesis_head,
				validation_code,
			));
615
616
617
618
		}

		// Finish registration of paras.
		run_to_session(2);
619

620
621
622
623
624
625
		// Start a new auction in the future
		let starting_block = System::block_number();
		let duration = 99u32;
		let lease_period_index_start = 4u32;
		assert_ok!(Auctions::new_auction(Origin::root(), duration, lease_period_index_start));

Shawn Tabrizi's avatar
Shawn Tabrizi committed
626
		for n in 1..=3 {
627
628
629
			// Create a crowdloan for each para
			assert_ok!(Crowdloan::create(
				Origin::signed(n),
630
				ParaId::from(start_para + n),
Shawn Tabrizi's avatar
Shawn Tabrizi committed
631
				100_000,                      // Cap
632
633
				lease_period_index_start + 2, // First Slot
				lease_period_index_start + 3, // Last Slot
Shawn Tabrizi's avatar
Shawn Tabrizi committed
634
				200,                          // Block End,
635
636
637
638
				None,
			));
		}

Shawn Tabrizi's avatar
Shawn Tabrizi committed
639
		for n in 1..=9 {
640
			// Increment block number
641
			run_to_block(starting_block + n * 10);
642
643
644

			Balances::make_free_balance_be(&(n * 10), n * 1_000);

645
			let para = start_para + n % 3 + 1;
646
647
648
649
650
651

			if n % 2 == 0 {
				// User 10 will bid directly for parachain 1
				assert_ok!(Auctions::bid(
					Origin::signed(n * 10),
					ParaId::from(para),
Shawn Tabrizi's avatar
Shawn Tabrizi committed
652
					1,                            // Auction Index
653
654
					lease_period_index_start + 0, // First Slot
					lease_period_index_start + 1, // Last slot
Shawn Tabrizi's avatar
Shawn Tabrizi committed
655
					n * 900,                      // Amount
656
657
				));
			} else {
658
				// User 20 will be a contribute to crowdloan for parachain 2
659
660
661
662
663
664
665
666
667
668
				assert_ok!(Crowdloan::contribute(
					Origin::signed(n * 10),
					ParaId::from(para),
					n + 900,
					None,
				));
			}
		}

		// Auction should be done
669
		run_to_block(starting_block + 110);
670
671

		// Appropriate Paras should have won slots
672
		let crowdloan_2 = Crowdloan::fund_account_id(ParaId::from(2001));
673
		assert_eq!(
674
			slots::Leases::<Test>::get(ParaId::from(2000)),
675
			// -- 1 --- 2 --- 3 --- 4 --- 5 ------------- 6 ------------------------ 7 -------------
Shawn Tabrizi's avatar
Shawn Tabrizi committed
676
677
678
679
680
681
682
683
684
			vec![
				None,
				None,
				None,
				None,
				None,
				Some((crowdloan_2, 1812)),
				Some((crowdloan_2, 1812))
			],
685
686
		);
		assert_eq!(
687
			slots::Leases::<Test>::get(ParaId::from(2002)),
688
689
690
691
692
693
694
695
696
697
698
			// -- 1 --- 2 --- 3 ---------- 4 --------------- 5 -------
			vec![None, None, None, Some((80, 7200)), Some((80, 7200))],
		);
	});
}

#[test]
fn basic_swap_works() {
	// This test will test a swap between a parachain and parathread works successfully.
	new_test_ext().execute_with(|| {
		assert!(System::block_number().is_one()); // So events are emitted
Shawn Tabrizi's avatar
Shawn Tabrizi committed
699
										  // User 1 and 2 will own paras
700
701
		Balances::make_free_balance_be(&1, 1_000_000_000);
		Balances::make_free_balance_be(&2, 1_000_000_000);
702
		// First register 2 parathreads with different data
703
		assert_ok!(Registrar::reserve(Origin::signed(1)));
704
705
		assert_ok!(Registrar::register(
			Origin::signed(1),
706
			ParaId::from(2000),
707
708
709
			test_genesis_head(10),
			test_validation_code(10),
		));
710
		assert_ok!(Registrar::reserve(Origin::signed(2)));
711
712
		assert_ok!(Registrar::register(
			Origin::signed(2),
713
			ParaId::from(2001),
714
715
716
717
718
			test_genesis_head(20),
			test_validation_code(20),
		));

		// Paras should be onboarding
719
720
		assert_eq!(Paras::lifecycle(ParaId::from(2000)), Some(ParaLifecycle::Onboarding));
		assert_eq!(Paras::lifecycle(ParaId::from(2001)), Some(ParaLifecycle::Onboarding));
721
722
723
724
725
726
727
728

		// Start a new auction in the future
		let duration = 99u32;
		let lease_period_index_start = 4u32;
		assert_ok!(Auctions::new_auction(Origin::root(), duration, lease_period_index_start));

		// 2 sessions later they are parathreads
		run_to_session(2);
729
730
		assert_eq!(Paras::lifecycle(ParaId::from(2000)), Some(ParaLifecycle::Parathread));
		assert_eq!(Paras::lifecycle(ParaId::from(2001)), Some(ParaLifecycle::Parathread));
731
732
733
734

		// Open a crowdloan for Para 1 for slots 0-3
		assert_ok!(Crowdloan::create(
			Origin::signed(1),
735
			ParaId::from(2000),
Shawn Tabrizi's avatar
Shawn Tabrizi committed
736
			1_000_000,                    // Cap
737
738
			lease_period_index_start + 0, // First Slot
			lease_period_index_start + 3, // Last Slot
Shawn Tabrizi's avatar
Shawn Tabrizi committed
739
			200,                          // Block End
740
741
			None,
		));
742
		let crowdloan_account = Crowdloan::fund_account_id(ParaId::from(2000));
743
744
745

		// Bunch of contributions
		let mut total = 0;
Shawn Tabrizi's avatar
Shawn Tabrizi committed
746
		for i in 10..20 {
747
			Balances::make_free_balance_be(&i, 1_000_000_000);
748
			assert_ok!(Crowdloan::contribute(Origin::signed(i), ParaId::from(2000), 900 - i, None));
749
750
751
752
753
754
755
756
757
758
759
760
761
762
			total += 900 - i;
		}
		assert!(total > 0);
		assert_eq!(Balances::free_balance(&crowdloan_account), total);

		// Go to end of auction where everyone won their slots
		run_to_block(200);

		// Deposit is appropriately taken
		// ----------------------------------------- para deposit --- crowdloan
		assert_eq!(Balances::reserved_balance(&1), (500 + 10 * 2 * 1) + 100);
		assert_eq!(Balances::reserved_balance(&2), 500 + 20 * 2 * 1);
		assert_eq!(Balances::reserved_balance(&crowdloan_account), total);
		// Crowdloan is appropriately set
763
764
		assert!(Crowdloan::funds(ParaId::from(2000)).is_some());
		assert!(Crowdloan::funds(ParaId::from(2001)).is_none());
765
766
767
768
769
770

		// New leases will start on block 400
		let lease_start_block = 400;
		run_to_block(lease_start_block);

		// Slots are won by Para 1
771
772
		assert!(!Slots::lease(ParaId::from(2000)).is_empty());
		assert!(Slots::lease(ParaId::from(2001)).is_empty());
773
774
775

		// 2 sessions later it is a parachain
		run_to_block(lease_start_block + 20);
776
777
		assert_eq!(Paras::lifecycle(ParaId::from(2000)), Some(ParaLifecycle::Parachain));
		assert_eq!(Paras::lifecycle(ParaId::from(2001)), Some(ParaLifecycle::Parathread));
778
779

		// Initiate a swap
Shawn Tabrizi's avatar
Shawn Tabrizi committed
780
781
782
783
784
785
786
787
788
789
		assert_ok!(Registrar::swap(
			para_origin(2000).into(),
			ParaId::from(2000),
			ParaId::from(2001)
		));
		assert_ok!(Registrar::swap(
			para_origin(2001).into(),
			ParaId::from(2001),
			ParaId::from(2000)
		));
790

791
792
		assert_eq!(Paras::lifecycle(ParaId::from(2000)), Some(ParaLifecycle::DowngradingParachain));
		assert_eq!(Paras::lifecycle(ParaId::from(2001)), Some(ParaLifecycle::UpgradingParathread));
793
794
795

		// 2 session later they have swapped
		run_to_block(lease_start_block + 40);
796
797
		assert_eq!(Paras::lifecycle(ParaId::from(2000)), Some(ParaLifecycle::Parathread));
		assert_eq!(Paras::lifecycle(ParaId::from(2001)), Some(ParaLifecycle::Parachain));
798
799

		// Deregister parathread
800
		assert_ok!(Registrar::deregister(para_origin(2000).into(), ParaId::from(2000)));
801
802
803
804
		// Correct deposit is unreserved
		assert_eq!(Balances::reserved_balance(&1), 100); // crowdloan deposit left over
		assert_eq!(Balances::reserved_balance(&2), 500 + 20 * 2 * 1);
		// Crowdloan ownership is swapped
805
806
		assert!(Crowdloan::funds(ParaId::from(2000)).is_none());
		assert!(Crowdloan::funds(ParaId::from(2001)).is_some());
807
		// Slot is swapped
808
809
		assert!(Slots::lease(ParaId::from(2000)).is_empty());
		assert!(!Slots::lease(ParaId::from(2001)).is_empty());
810
811

		// Cant dissolve
Shawn Tabrizi's avatar
Shawn Tabrizi committed
812
813
814
815
816
817
818
819
		assert_noop!(
			Crowdloan::dissolve(Origin::signed(1), ParaId::from(2000)),
			CrowdloanError::<Test>::InvalidParaId
		);
		assert_noop!(
			Crowdloan::dissolve(Origin::signed(2), ParaId::from(2001)),
			CrowdloanError::<Test>::NotReadyToDissolve
		);
820
821
822
823
824
825

		// Go way in the future when the para is offboarded
		run_to_block(lease_start_block + 1000);

		// Withdraw of contributions works
		assert_eq!(Balances::free_balance(&crowdloan_account), total);
Shawn Tabrizi's avatar
Shawn Tabrizi committed
826
		for i in 10..20 {
827
			assert_ok!(Crowdloan::withdraw(Origin::signed(i), i, ParaId::from(2001)));
828
829
830
831
		}
		assert_eq!(Balances::free_balance(&crowdloan_account), 0);

		// Dissolve returns the balance of the person who put a deposit for crowdloan
832
		assert_ok!(Crowdloan::dissolve(Origin::signed(1), ParaId::from(2001)));
833
834
835
836
		assert_eq!(Balances::reserved_balance(&1), 0);
		assert_eq!(Balances::reserved_balance(&2), 500 + 20 * 2 * 1);

		// Final deregister sets everything back to the start
837
		assert_ok!(Registrar::deregister(para_origin(2001).into(), ParaId::from(2001)));
838
839
840
841
842
843
844
845
		assert_eq!(Balances::reserved_balance(&2), 0);
	})
}

#[test]
fn crowdloan_ending_period_bid() {
	new_test_ext().execute_with(|| {
		assert!(System::block_number().is_one()); // So events are emitted
Shawn Tabrizi's avatar
Shawn Tabrizi committed
846
										  // User 1 and 2 will own paras
847
848
		Balances::make_free_balance_be(&1, 1_000_000_000);
		Balances::make_free_balance_be(&2, 1_000_000_000);
849
		// First register 2 parathreads
850
		assert_ok!(Registrar::reserve(Origin::signed(1)));
851
852
		assert_ok!(Registrar::register(
			Origin::signed(1),
853
			ParaId::from(2000),
854
855
856
			test_genesis_head(10),
			test_validation_code(10),
		));
857
		assert_ok!(Registrar::reserve(Origin::signed(2)));
858
859
		assert_ok!(Registrar::register(
			Origin::signed(2),
860
			ParaId::from(2001),
861
862
863
864
865
			test_genesis_head(20),
			test_validation_code(20),
		));

		// Paras should be onboarding
866
867
		assert_eq!(Paras::lifecycle(ParaId::from(2000)), Some(ParaLifecycle::Onboarding));
		assert_eq!(Paras::lifecycle(ParaId::from(2001)), Some(ParaLifecycle::Onboarding));
868
869
870
871
872
873
874
875

		// Start a new auction in the future
		let duration = 99u32;
		let lease_period_index_start = 4u32;
		assert_ok!(Auctions::new_auction(Origin::root(), duration, lease_period_index_start));

		// 2 sessions later they are parathreads
		run_to_session(2);
876
877
		assert_eq!(Paras::lifecycle(ParaId::from(2000)), Some(ParaLifecycle::Parathread));
		assert_eq!(Paras::lifecycle(ParaId::from(2001)), Some(ParaLifecycle::Parathread));
878
879
880
881

		// Open a crowdloan for Para 1 for slots 0-3
		assert_ok!(Crowdloan::create(
			Origin::signed(1),
882
			ParaId::from(2000),
Shawn Tabrizi's avatar
Shawn Tabrizi committed
883
			1_000_000,                    // Cap
884
885
			lease_period_index_start + 0, // First Slot
			lease_period_index_start + 3, // Last Slot
Shawn Tabrizi's avatar
Shawn Tabrizi committed
886
			200,                          // Block End
887
888
			None,
		));
889
		let crowdloan_account = Crowdloan::fund_account_id(ParaId::from(2000));
890
891
892

		// Bunch of contributions
		let mut total = 0;
Shawn Tabrizi's avatar
Shawn Tabrizi committed
893
		for i in 10..20 {
894
			Balances::make_free_balance_be(&i, 1_000_000_000);
895
			assert_ok!(Crowdloan::contribute(Origin::signed(i), ParaId::from(2000), 900 - i, None));
896
897
898
899
900
901
			total += 900 - i;
		}
		assert!(total > 0);
		assert_eq!(Balances::free_balance(&crowdloan_account), total);

		// Bid for para 2 directly
902
		Balances::make_free_balance_be(&2, 1_000_000_000);
903
904
		assert_ok!(Auctions::bid(
			Origin::signed(2),
905
			ParaId::from(2001),
Shawn Tabrizi's avatar
Shawn Tabrizi committed
906
			1,                            // Auction Index
907
908
			lease_period_index_start + 0, // First Slot
			lease_period_index_start + 1, // Last slot
Shawn Tabrizi's avatar
Shawn Tabrizi committed
909
			900,                          // Amount
910
911
912
913
914
		));

		// Go to beginning of ending period
		run_to_block(100);

915
		assert_eq!(Auctions::auction_status(100), AuctionStatus::<u32>::EndingPeriod(0, 0));
916
		let mut winning = [None; SlotRange::SLOT_RANGE_COUNT];
917
		winning[SlotRange::ZeroOne as u8 as usize] = Some((2, ParaId::from(2001), 900));
Shawn Tabrizi's avatar
Shawn Tabrizi committed
918
919
		winning[SlotRange::ZeroThree as u8 as usize] =
			Some((crowdloan_account, ParaId::from(2000), total));
920

921
		assert_eq!(Auctions::winning(0), Some(winning));
922
923
924

		run_to_block(101);

925
		Balances::make_free_balance_be(&1234, 1_000_000_000);
926
		assert_ok!(Crowdloan::contribute(Origin::signed(1234), ParaId::from(2000), 900, None));
927
928
929

		// Data propagates correctly
		run_to_block(102);
930
		let mut winning = [None; SlotRange::SLOT_RANGE_COUNT];
931
		winning[SlotRange::ZeroOne as u8 as usize] = Some((2, ParaId::from(2001), 900));
Shawn Tabrizi's avatar
Shawn Tabrizi committed
932
933
		winning[SlotRange::ZeroThree as u8 as usize] =
			Some((crowdloan_account, ParaId::from(2000), total + 900));
934
		assert_eq!(Auctions::winning(2), Some(winning));
935
936
	})
}
937
938
939
940
941
942
943
944
945
946
947
948

#[test]
fn auction_bid_requires_registered_para() {
	new_test_ext().execute_with(|| {
		assert!(System::block_number().is_one()); // So events are emitted

		// Start a new auction in the future
		let duration = 99u32;
		let lease_period_index_start = 4u32;
		assert_ok!(Auctions::new_auction(Origin::root(), duration, lease_period_index_start));

		// Can't bid with non-registered paras
949
		Balances::make_free_balance_be(&1, 1_000_000_000);
Shawn Tabrizi's avatar
Shawn Tabrizi committed
950
951
952
953
954
955
956
957
958
959
960
		assert_noop!(
			Auctions::bid(
				Origin::signed(1),
				ParaId::from(2000),
				1,                            // Auction Index
				lease_period_index_start + 0, // First Slot
				lease_period_index_start + 1, // Last slot
				900,                          // Amount
			),
			AuctionsError::<Test>::ParaNotRegistered
		);
961
962

		// Now we register the para
963
		assert_ok!(Registrar::reserve(Origin::signed(1)));
964
965
		assert_ok!(Registrar::register(
			Origin::signed(1),
966
			ParaId::from(2000),
967
968
969
970
971
			test_genesis_head(10),
			test_validation_code(10),
		));

		// Still can't bid until it is fully onboarded
Shawn Tabrizi's avatar
Shawn Tabrizi committed
972
973
974
975
976
977
978
979
980
981
982
		assert_noop!(
			Auctions::bid(
				Origin::signed(1),
				ParaId::from(2000),
				1,                            // Auction Index
				lease_period_index_start + 0, // First Slot
				lease_period_index_start + 1, // Last slot
				900,                          // Amount
			),
			AuctionsError::<Test>::ParaNotRegistered
		);
983
984
985
986
987

		// Onboarded on Session 2
		run_to_session(2);

		// Success
988
		Balances::make_free_balance_be(&1, 1_000_000_000);
989
990
		assert_ok!(Auctions::bid(
			Origin::signed(1),
991
			ParaId::from(2000),
Shawn Tabrizi's avatar
Shawn Tabrizi committed
992
			1,                            // Auction Index
993
994
			lease_period_index_start + 0, // First Slot
			lease_period_index_start + 1, // Last slot
Shawn Tabrizi's avatar
Shawn Tabrizi committed
995
			900,                          // Amount
996
997
998
		));
	});
}
999
1000

#[test]
For faster browsing, not all history is shown. View entire blame