1 files changed, 67 insertions, 0 deletions
diff --git a/spec/frontend/artifacts/graphql/cache_update_spec.js b/spec/frontend/artifacts/graphql/cache_update_spec.js
new file mode 100644
index 00000000000..4d610328298
--- /dev/null
+++ b/spec/frontend/artifacts/graphql/cache_update_spec.js
@@ -0,0 +1,67 @@
+import getJobArtifactsQuery from '~/artifacts/graphql/queries/get_job_artifacts.query.graphql';
+import { removeArtifactFromStore } from '~/artifacts/graphql/cache_update';
+
+describe('Artifact table cache updates', () => {
+  let store;
+
+  const cacheMock = {
+    project: {
+      jobs: {
+        nodes: [
+          { artifacts: { nodes: [{ id: 'foo' }] } },
+          { artifacts: { nodes: [{ id: 'bar' }] } },
+        ],
+      },
+    },
+  };
+
+  const query = getJobArtifactsQuery;
+  const variables = { fullPath: 'path/to/project' };
+
+  beforeEach(() => {
+    store = {
+      readQuery: jest.fn().mockReturnValue(cacheMock),
+      writeQuery: jest.fn(),
+    };
+  });
+
+  describe('removeArtifactFromStore', () => {
+    it('calls readQuery', () => {
+      removeArtifactFromStore(store, 'foo', query, variables);
+      expect(store.readQuery).toHaveBeenCalledWith({ query, variables });
+    });
+
+    it('writes the correct result in the cache', () => {
+      removeArtifactFromStore(store, 'foo', query, variables);
+      expect(store.writeQuery).toHaveBeenCalledWith({
+        query,
+        variables,
+        data: {
+          project: {
+            jobs: {
+              nodes: [{ artifacts: { nodes: [] } }, { artifacts: { nodes: [{ id: 'bar' }] } }],
+            },
+          },
+        },
+      });
+    });
+
+    it('does not remove an unknown artifact', () => {
+      removeArtifactFromStore(store, 'baz', query, variables);
+      expect(store.writeQuery).toHaveBeenCalledWith({
+        query,
+        variables,
+        data: {
+          project: {
+            jobs: {
+              nodes: [
+                { artifacts: { nodes: [{ id: 'foo' }] } },
+                { artifacts: { nodes: [{ id: 'bar' }] } },
+              ],
+            },
+          },
+        },
+      });
+    });
+  });
+});