[Query Copilot V2] Unit tests for V2 Copilot (#1580)

* Add tests for V2 of copilot and fix query parameter feature flag

* Fix merge changes

src/Explorer/QueryCopilot/V2/Sidebar/QueryCopilotSidebar.test.tsx

@@ -1,10 +1,54 @@
+import { Stack } from "@fluentui/react";
+import { SampleBubble } from "Explorer/QueryCopilot/V2/Bubbles/Sample/SampleBubble";
 import { shallow } from "enzyme";
+import { useQueryCopilot } from "hooks/useQueryCopilot";
+import { withHooks } from "jest-react-hooks-shallow";
 import React from "react";
 import { QueryCopilotSidebar } from "./QueryCopilotSidebar";
 
-describe("QueryCopilotSidebar snapshot test", () => {
-  it("should render ", () => {
+describe("Query Copilot Sidebar snapshot test", () => {
+  const initialState = useQueryCopilot.getState();
+
+  beforeEach(() => {
+    useQueryCopilot.setState(initialState, true);
+  });
+
+  it("should render and set copilot used flag ", () => {
+    withHooks(() => {
+      useQueryCopilot.getState().setShowCopilotSidebar(true);
+      const wrapper = shallow(<QueryCopilotSidebar />);
+
+      expect(useQueryCopilot.getState().wasCopilotUsed).toBeTruthy();
+      expect(wrapper).toMatchSnapshot();
+    });
+  });
+
+  it("should render and not set copilot used flag ", () => {
+    withHooks(() => {
+      const wrapper = shallow(<QueryCopilotSidebar />);
+
+      expect(useQueryCopilot.getState().wasCopilotUsed).toBeFalsy();
+      expect(wrapper).toMatchSnapshot();
+    });
+  });
+
+  it("should render with chat messages", () => {
+    const message = "some test message";
+    useQueryCopilot.getState().setChatMessages([message]);
     const wrapper = shallow(<QueryCopilotSidebar />);
+
+    const messageContainer = wrapper.find(Stack).findWhere((x) => x.text() === message);
+
+    expect(messageContainer).toBeDefined();
+    expect(wrapper).toMatchSnapshot();
+  });
+
+  it("should render samples without messages", () => {
+    const wrapper = shallow(<QueryCopilotSidebar />);
+
+    const sampleBubble = wrapper.find(SampleBubble);
+
+    expect(sampleBubble).toBeDefined();
     expect(wrapper).toMatchSnapshot();
   });
 });