@@ -3,6 +3,7 @@ import JWTDecode
3
3
import TorusUtils
4
4
import UIKit
5
5
import XCTest
6
+ import curveSecp256k1
6
7
7
8
final class MockSDKTest : XCTestCase {
8
9
func test_jwtDecodeTest( ) {
@@ -11,11 +12,11 @@ final class MockSDKTest: XCTestCase {
11
12
print ( decodedData)
12
13
}
13
14
14
- func testGetTorusKey( ) async {
15
+ func testGetTorusKey( ) async throws {
15
16
let expectation = XCTestExpectation ( description: " getTorusKey should correctly proxy input and output to/from TorusUtils " )
16
17
17
- let expectedPrivateKey = fakeData. generatePrivateKey ( )
18
- let expectedPublicAddress = fakeData. generatePublicKey ( )
18
+ let expectedPrivateKey = try fakeData. generatePrivateKey ( )
19
+ let expectedPublicAddress = try fakeData. generatePublicKey ( )
19
20
let expectedVerifier = fakeData. generateVerifier ( )
20
21
let expectedVerfierId = fakeData. generateRandomEmail ( of: 6 )
21
22
@@ -48,11 +49,11 @@ final class MockSDKTest: XCTestCase {
48
49
}
49
50
}
50
51
51
- func testGetAggregateTorusKey( ) async {
52
+ func testGetAggregateTorusKey( ) async throws {
52
53
let expectation = XCTestExpectation ( description: " getAggregateTorusKey should correctly proxy input and output to/from TorusUtils " )
53
54
54
- let expectedPrivateKey = fakeData. generatePrivateKey ( )
55
- let expectedPublicAddress = fakeData. generatePublicKey ( )
55
+ let expectedPrivateKey = try fakeData. generatePrivateKey ( )
56
+ let expectedPublicAddress = try fakeData. generatePublicKey ( )
56
57
let expectedVerifier = fakeData. generateVerifier ( )
57
58
let expectedVerfierId = fakeData. generateRandomEmail ( of: 6 )
58
59
@@ -96,15 +97,14 @@ class fakeData {
96
97
return String . randomString ( length: 10 )
97
98
}
98
99
99
- static func generatePrivateKey( ) -> String {
100
- let privateKey = Data . randomOfLength ( 32 )
101
- return ( privateKey? . toHexString ( ) ) !
100
+ static func generatePrivateKey( ) throws -> String {
101
+ let privateKey = curveSecp256k1 . SecretKey ( )
102
+ return try privateKey. serialize ( )
102
103
}
103
104
104
- static func generatePublicKey( ) -> String {
105
- let privateKey = Data . randomOfLength ( 32 ) !
106
- let publicKey = SECP256K1 . privateToPublic ( privateKey: privateKey) ? . subdata ( in: 1 ..< 65 )
107
- return publicKey!. toHexString ( )
105
+ static func generatePublicKey( ) throws -> String {
106
+ // drop "04"
107
+ return try String ( curveSecp256k1. SecretKey ( ) . toPublic ( ) . serialize ( compressed: false ) . dropFirst ( 2 ) )
108
108
}
109
109
110
110
static func generateRandomEmail( of length: Int ) -> String {
0 commit comments