forked from OpenZeppelin/ethernaut
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Vault.test.js
71 lines (60 loc) · 1.74 KB
/
Vault.test.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
const VaultFactory = artifacts.require('./levels/VaultFactory.sol');
const Vault = artifacts.require('./attacks/Vault.sol');
const VaultAttack = artifacts.require('./attacks/VaultAttack.sol');
const Ethernaut = artifacts.require('./Ethernaut.sol');
const {
BN,
constants,
expectEvent,
expectRevert,
} = require('openzeppelin-test-helpers');
const utils = require('../utils/TestUtils');
const { ethers, upgrades } = require('hardhat');
contract('Vault', function (accounts) {
let ethernaut;
let level;
let owner = accounts[1];
let player = accounts[0];
let statproxy;
before(async function () {
ethernaut = await utils.getEthernautWithStatsProxy();
level = await VaultFactory.new();
await ethernaut.registerLevel(level.address);
});
it('should fail if the player did not solve the level', async function () {
const instance = await utils.createLevelInstance(
ethernaut,
level.address,
player,
Vault
);
const completed = await utils.submitLevelInstance(
ethernaut,
level.address,
instance.address,
player
);
assert.isFalse(completed);
});
it('should allow the player to solve the level', async function () {
const instance = await utils.createLevelInstance(
ethernaut,
level.address,
player,
Vault
);
const attacker = await VaultAttack.new();
var hexStr = web3.utils.utf8ToHex(
await web3.eth.getStorageAt(instance.address, 1)
);
var password = web3.utils.hexToAscii(hexStr);
await attacker.attack(instance.address, password);
const completed = await utils.submitLevelInstance(
ethernaut,
level.address,
instance.address,
player
);
assert.isTrue(completed);
});
});