Using jest mock for socket.io events in react native - react-native

Question

I am trying to write tests for my react native app but am running into issues because we use sockets for connection and data transmission. The very basic initial test code looks like this:
import React from "react";
import renderer from "react-test-renderer";
import { SessionTab } from "./index";
beforeEach(() => {
jest.spyOn(console, "warn").mockImplementation(() => {});
});
beforeAll(() => {
ReactDOM.createPortal = jest.fn((element, node) => {
return element;
});
});
test("renders correctly", () => {
const tree = renderer.create(<SessionTab />).toJSON();
expect(tree).toMatchSnapshot();
});
The error I'm getting when I run my test looks like this:
6 | export default store = createStore(rootReducer, applyMiddleware(thunk));
7 |
> 8 | SocketDriver.OnBroadcast('admin.getModel', async () => {
| ^
9 | let messageQueue = await SocketDriver.messageQueue.getQueue();
10 |
11 | SocketDriver.emitToServer('system.reportModel', { data: {
The function being referenced there looks like this:
static OnBroadcast(event, handler) {
SocketDriver.listen(event);
event += '.broadcast';
SocketDriver.events[event].addListener(event, handler);
}
I think this is where using jest mocks could help, but even though I've read some documentation, I'm unclear how to implement. How would I set up a jest mock for the OnBroadcast event above?