Mock client onConnect and onDisconnect

agama-project · May 28, 2024 · 5db3e70 · 5db3e70
1 parent e9de732
commit 5db3e70
Show file tree

Hide file tree

Showing 6 changed files with 12 additions and 1 deletion.
diff --git a/web/src/components/overview/L10nSection.test.jsx b/web/src/components/overview/L10nSection.test.jsx
@@ -50,6 +50,8 @@ beforeEach(() => {
   // if defined outside, the mock is cleared automatically
   createClient.mockImplementation(() => {
     return {
+      onConnect: jest.fn(),
+      onDisconnect: jest.fn(),
       l10n: l10nClientMock,
     };
   });

diff --git a/web/src/components/overview/NetworkSection.test.jsx b/web/src/components/overview/NetworkSection.test.jsx
@@ -54,6 +54,8 @@ beforeEach(() => {
   // if defined outside, the mock is cleared automatically
   createClient.mockImplementation(() => {
     return {
+      onConnect: jest.fn(),
+      onDisconnect: jest.fn(),
       network: {
         devices: devicesFn,
         onNetworkChange: onNetworkChangeEventFn,

diff --git a/web/src/components/overview/ProductSection.test.jsx b/web/src/components/overview/ProductSection.test.jsx
@@ -49,6 +49,8 @@ beforeEach(() => {
 
   createClient.mockImplementation(() => {
     return {
+      onConnect: jest.fn(),
+      onDisconnect: jest.fn(),
       product: {
         getIssues: jest.fn().mockResolvedValue(issues),
         onIssuesChange: jest.fn()

diff --git a/web/src/components/overview/SoftwareSection.test.jsx b/web/src/components/overview/SoftwareSection.test.jsx
@@ -45,6 +45,8 @@ let getIssuesFn = jest.fn().mockResolvedValue([]);
 beforeEach(() => {
   createClient.mockImplementation(() => {
     return {
+      onConnect: jest.fn(),
+      onDisconnect: jest.fn(),
       software: {
         getStatus: getStatusFn,
         getProgress: getProgressFn,

diff --git a/web/src/components/overview/StorageSection.test.jsx b/web/src/components/overview/StorageSection.test.jsx
@@ -47,6 +47,7 @@ const proposalResult = {
 };
 
 const storageMock = {
+
   probe: jest.fn().mockResolvedValue(0),
   proposal: {
     getAvailableDevices: jest.fn().mockResolvedValue(availableDevices),
@@ -70,7 +71,7 @@ let storage;
 beforeEach(() => {
   storage = { ...storageMock, proposal: { ...storageMock.proposal } };
 
-  createClient.mockImplementation(() => ({ storage }));
+  createClient.mockImplementation(() => ({ storage, onConnect: jest.fn(), onDisconnect: jest.fn() }));
 });
 
 it("probes storage if the storage devices are deprecated", async () => {

diff --git a/web/src/components/overview/UsersSection.test.jsx b/web/src/components/overview/UsersSection.test.jsx
@@ -54,6 +54,8 @@ beforeEach(() => {
   // if defined outside, the mock is cleared automatically
   createClient.mockImplementation(() => {
     return {
+      onConnect: jest.fn(),
+      onDisconnect: jest.fn(),
       users: {
         ...userClientMock,
       }