Michael Scoff
7 years ago
parent
ab3365505d
commit
38bc99e6da
@ -0,0 +1,125 @@ |
|||||||
|
// Copyright SIX DAY LLC. All rights reserved. |
||||||
|
|
||||||
|
import XCTest |
||||||
|
@testable import Trust |
||||||
|
import KeychainSwift |
||||||
|
import Geth |
||||||
|
|
||||||
|
class EtherKeystoreTests: XCTestCase { |
||||||
|
|
||||||
|
func testInitialization() { |
||||||
|
let keystore = FakeEtherKeystore() |
||||||
|
|
||||||
|
XCTAssertNotNil(keystore) |
||||||
|
XCTAssertEqual(false, keystore.hasAccounts) |
||||||
|
} |
||||||
|
|
||||||
|
func testCreateWallet() { |
||||||
|
let keystore = FakeEtherKeystore() |
||||||
|
let password = "test" |
||||||
|
|
||||||
|
let account = keystore.createAccout(password: password) |
||||||
|
let retrivedPassword = keystore.getPassword(for: account) |
||||||
|
|
||||||
|
XCTAssertEqual(password, retrivedPassword) |
||||||
|
XCTAssertEqual(1, keystore.accounts.count) |
||||||
|
} |
||||||
|
|
||||||
|
func testSetAndGetPasswordForAccount() { |
||||||
|
let keystore = FakeEtherKeystore() |
||||||
|
let password = "test" |
||||||
|
let account: Account = .make() |
||||||
|
|
||||||
|
let setPassword = keystore.setPassword(password, for: account) |
||||||
|
let retrivedPassword = keystore.getPassword(for: account) |
||||||
|
|
||||||
|
XCTAssertEqual(true, setPassword) |
||||||
|
XCTAssertEqual(retrivedPassword, password) |
||||||
|
} |
||||||
|
|
||||||
|
func testImport() { |
||||||
|
let keystore = FakeEtherKeystore() |
||||||
|
let password = "test" |
||||||
|
|
||||||
|
let result = keystore.importKeystore( |
||||||
|
value: "{\"address\":\"5e9c27156a612a2d516c74c7a80af107856f8539\",\"crypto\":{\"cipher\":\"aes-128-ctr\",\"ciphertext\":\"5eb0c790d1fb27824c78acac9233241b340c329b46aba08c6533b70ab67ea74f\",\"cipherparams\":{\"iv\":\"e5ab559977af075eda00a97c8f0ce506\"},\"kdf\":\"scrypt\",\"kdfparams\":{\"dklen\":32,\"n\":4096,\"p\":6,\"r\":8,\"salt\":\"b43142f34caf2b3b39c16f52344701f800711589f799cdae1827ac2f844f9602\"},\"mac\":\"c6ccaecca7896974dacac91a8116216ec287930bc74bfd7694a94f08bd992095\"},\"id\":\"e3554f73-4d0a-40a0-b721-fc801623d5ba\",\"version\":3}", |
||||||
|
password: password |
||||||
|
) |
||||||
|
|
||||||
|
guard case let .success(account) = result else { |
||||||
|
return XCTFail() |
||||||
|
} |
||||||
|
|
||||||
|
XCTAssertEqual("0x5E9c27156a612a2D516C74c7a80af107856F8539", account.address.address) |
||||||
|
XCTAssertEqual(1, keystore.accounts.count) |
||||||
|
} |
||||||
|
|
||||||
|
func testRecentlyUsedAccount() { |
||||||
|
let keystore = FakeEtherKeystore() |
||||||
|
let password = "test" |
||||||
|
|
||||||
|
XCTAssertNil(keystore.recentlyUsedAccount) |
||||||
|
|
||||||
|
let account = keystore.createAccout(password: password) |
||||||
|
|
||||||
|
keystore.recentlyUsedAccount = account |
||||||
|
|
||||||
|
XCTAssertEqual(account, keystore.recentlyUsedAccount) |
||||||
|
|
||||||
|
keystore.recentlyUsedAccount = nil |
||||||
|
|
||||||
|
XCTAssertNil(keystore.recentlyUsedAccount) |
||||||
|
} |
||||||
|
|
||||||
|
func testSignTransaction() { |
||||||
|
let keystore = FakeEtherKeystore() |
||||||
|
let account = keystore.createAccout(password: "test") |
||||||
|
|
||||||
|
let transaction = keystore.signTransaction( |
||||||
|
amount: GethNewBigInt(1), |
||||||
|
account: account, |
||||||
|
address: .make(), |
||||||
|
nonce: 0, |
||||||
|
speed: .regular |
||||||
|
) |
||||||
|
|
||||||
|
guard case .success = transaction else { |
||||||
|
return XCTAssertFalse(true) |
||||||
|
} |
||||||
|
|
||||||
|
XCTAssertTrue(true) |
||||||
|
} |
||||||
|
|
||||||
|
func testDeleteAccount() { |
||||||
|
let keystore = FakeEtherKeystore() |
||||||
|
let password = "test" |
||||||
|
let account = keystore.createAccout(password: password) |
||||||
|
|
||||||
|
XCTAssertEqual(1, keystore.accounts.count) |
||||||
|
|
||||||
|
let result = keystore.delete(account: account, password: password) |
||||||
|
|
||||||
|
guard case .success = result else { |
||||||
|
return XCTFail() |
||||||
|
} |
||||||
|
|
||||||
|
XCTAssertEqual(0, keystore.accounts.count) |
||||||
|
} |
||||||
|
|
||||||
|
func testDeleteAccountFail() { |
||||||
|
let keystore = FakeEtherKeystore() |
||||||
|
let password = "test" |
||||||
|
let account = keystore.createAccout(password: password) |
||||||
|
|
||||||
|
XCTAssertEqual(1, keystore.accounts.count) |
||||||
|
|
||||||
|
let result = keystore.delete(account: account, password: "invalidPassword") |
||||||
|
|
||||||
|
guard case .failure = result else { |
||||||
|
return XCTFail() |
||||||
|
} |
||||||
|
|
||||||
|
XCTAssertEqual(1, keystore.accounts.count) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
Loading…
Reference in new issue