m/metamask-extension
1
0
Fork 0
mirror of https://github.com/kremalicious/metamask-extension.git synced 2024-11-22 09:57:02 +01:00
Code

Sync NetworkController initializeProvider tests w/ core (#19297)

Browse Source 
This makes it easier to visually compare differences in the
NetworkController unit tests between core and this repo.
This commit is contained in:
Elliot Winkler 2023-06-02 20:41:57 -06:00 committed by GitHub
parent 63c402155d
commit a5c370cdfa
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 53 additions and 66 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

115
app/scripts/controllers/network/network-controller.test.ts
View File

@ -259,37 +259,34 @@ describe('NetworkController', () => {
});
describe('initializeProvider', () => {
it('throws if the provider configuration is invalid', async () => {
const invalidProviderConfig = {};
await withController(
/* @ts-expect-error We're intentionally passing bad input. */
{
state: {
providerConfig: invalidProviderConfig,
describe('when the type in the provider config is invalid', () => {
it('throws', async () => {
const invalidProviderConfig = {};
await withController(
/* @ts-expect-error We're intentionally passing bad input. */
{
state: {
providerConfig: invalidProviderConfig,
},
},
},
async ({ controller }) => {
await expect(async () => {
await controller.initializeProvider();
}).rejects.toThrow(
'NetworkController - #configureProvider - unknown type "undefined"',
);
},
);
async ({ controller }) => {
await expect(async () => {
await controller.initializeProvider();
}).rejects.toThrow("Unrecognized network type: 'undefined'");
},
);
});
});
for (const { networkType } of INFURA_NETWORKS) {
describe(`when the type in the provider configuration is "${networkType}"`, () => {
it(`initializes a provider pointed to the "${networkType}" Infura network`, async () => {
describe(`when the type in the provider config is "${networkType}"`, () => {
it(`creates a network client for the ${networkType} Infura network, capturing the resulting provider`, async () => {
await withController(
{
state: {
providerConfig: {
providerConfig: buildProviderConfig({
type: networkType,
// NOTE: This doesn't need to match the logical chain ID
// of the network selected, it just needs to exist
chainId: '0x9999999',
},
}),
},
infuraProjectId: 'some-infura-project-id',
},
@ -297,7 +294,8 @@ describe('NetworkController', () => {
const fakeProvider = buildFakeProvider([
{
request: {
method: 'test',
method: 'test_method',
params: [],
},
response: {
result: 'test response',
@ -305,27 +303,27 @@ describe('NetworkController', () => {
},
]);
const fakeNetworkClient = buildFakeClient(fakeProvider);
mockCreateNetworkClient()
.calledWith({
network: networkType,
infuraProjectId: 'some-infura-project-id',
type: NetworkClientType.Infura,
})
.mockReturnValue(fakeNetworkClient);
createNetworkClientMock.mockReturnValue(fakeNetworkClient);
await controller.initializeProvider();
expect(createNetworkClientMock).toHaveBeenCalledWith({
network: networkType,
infuraProjectId: 'some-infura-project-id',
type: NetworkClientType.Infura,
});
const { provider } = controller.getProviderAndBlockTracker();
assert(provider, 'Provider is somehow unset');
assert(provider, 'Provider is not set');
const promisifiedSendAsync = promisify(provider.sendAsync).bind(
provider,
);
const response = await promisifiedSendAsync({
id: '1',
const { result } = await promisifiedSendAsync({
id: 1,
jsonrpc: '2.0',
method: 'test',
method: 'test_method',
params: [],
});
expect(response.result).toBe('test response');
expect(result).toBe('test response');
},
);
});
@ -344,23 +342,14 @@ describe('NetworkController', () => {
describe(`when the type in the provider configuration is "rpc"`, () => {
describe('if chainId and rpcUrl are present in the provider config', () => {
it('initializes a provider pointed to the given RPC URL whose chain ID matches the configured chain ID', async () => {
it('creates a network client for a custom RPC endpoint using the provider config, capturing the resulting provider', async () => {
await withController(
{
state: {
providerConfig: {
type: 'rpc',
chainId: '0x1337',
rpcUrl: 'https://mock-rpc-url',
ticker: 'TEST',
},
networkConfigurations: {
testNetworkConfigurationId: {
rpcUrl: 'https://mock-rpc-url',
chainId: '0x1337',
ticker: 'TEST',
id: 'testNetworkConfigurationId',
},
type: NETWORK_TYPES.RPC,
chainId: toHex(1337),
rpcUrl: 'http://example.com',
},
},
},
@ -368,7 +357,8 @@ describe('NetworkController', () => {
const fakeProvider = buildFakeProvider([
{
request: {
method: 'test',
method: 'test_method',
params: [],
},
response: {
result: 'test response',
@ -376,28 +366,27 @@ describe('NetworkController', () => {
},
]);
const fakeNetworkClient = buildFakeClient(fakeProvider);
mockCreateNetworkClient()
.calledWith({
chainId: '0x1337',
rpcUrl: 'https://mock-rpc-url',
type: NetworkClientType.Custom,
})
.mockReturnValue(fakeNetworkClient);
mockCreateNetworkClient().mockReturnValue(fakeNetworkClient);
createNetworkClientMock.mockReturnValue(fakeNetworkClient);
await controller.initializeProvider();
expect(createNetworkClientMock).toHaveBeenCalledWith({
chainId: toHex(1337),
rpcUrl: 'http://example.com',
type: NetworkClientType.Custom,
});
const { provider } = controller.getProviderAndBlockTracker();
assert(provider, 'Provider is somehow unset');
assert(provider, 'Provider is not set');
const promisifiedSendAsync = promisify(provider.sendAsync).bind(
provider,
);
const response = await promisifiedSendAsync({
id: '1',
const { result } = await promisifiedSendAsync({
id: 1,
jsonrpc: '2.0',
method: 'test',
method: 'test_method',
params: [],
});
expect(response.result).toBe('test response');
expect(result).toBe('test response');
},
);
});

4
app/scripts/controllers/network/network-controller.ts
View File

@ -909,9 +909,7 @@ export class NetworkController extends EventEmitter {
}
this.#configureStandardProvider(rpcUrl, chainId);
} else {
throw new Error(
`NetworkController - #configureProvider - unknown type "${type}"`,
);
throw new Error(`Unrecognized network type: '${type}'`);
}
}