refine message_history test

Chainlit · May 16, 2023 · 8d453b4 · 8d453b4
1 parent 38f3214
commit 8d453b4
Show file tree

Hide file tree

Showing 4 changed files with 20 additions and 7 deletions.
diff --git a/cypress/e2e/message_history/spec.cy.ts b/cypress/e2e/message_history/spec.cy.ts
@@ -1,4 +1,8 @@
-import { openHistory, submitMessage } from "../../support/testUtils";
+import {
+  closeHistory,
+  openHistory,
+  submitMessage,
+} from "../../support/testUtils";
 
 describe("Message History", () => {
   before(() => {
@@ -9,15 +13,21 @@ describe("Message History", () => {
   });
 
   it("should be able to show the last message in the message history", () => {
+    openHistory();
+
+    cy.get(".history-item").should("have.length", 0);
+    cy.get("#history-empty").should("exist");
+
+    closeHistory();
+
     const timestamp = Date.now().toString();
 
     submitMessage(timestamp);
     cy.wait(["@message"]);
 
-    cy.get(".history-item").should("have.length", 0);
-
     openHistory();
 
+    cy.get("#history-empty").should("not.exist");
     cy.get(".history-item").should("have.length", 1);
     cy.get(".history-item").eq(0).should("contain", timestamp).click();
     cy.get(".history-item").should("have.length", 0);

diff --git a/cypress/support/testUtils.ts b/cypress/support/testUtils.ts
@@ -9,3 +9,7 @@ export function openHistory() {
   cy.get(`#chat-input`).should("not.be.disabled");
   cy.get(`#chat-input`).type(`{upArrow}`);
 }
+
+export function closeHistory() {
+  cy.get(`body`).click();
+}
diff --git a/src/chainlit/frontend/src/components/chat/history/button.tsx b/src/chainlit/frontend/src/components/chat/history/button.tsx
@@ -95,7 +95,7 @@ export default function HistoryButton({ onClick, onOpen, chats }: Props) {
   const empty =
     chats?.length === 0 ? (
       // @ts-ignore
-      <div key="empty" disabled>
+      <div key="empty" id="history-empty" disabled>
         <Typography
           color="text.secondary"
           sx={{
@@ -112,7 +112,7 @@ export default function HistoryButton({ onClick, onOpen, chats }: Props) {
 
   const loading = !chats ? (
     // @ts-ignore
-    <div key="loading" disabled>
+    <div key="loading" id="history-loading" disabled>
       <Typography
         color="text.secondary"
         sx={{
@@ -198,7 +198,6 @@ export default function HistoryButton({ onClick, onOpen, chats }: Props) {
     <Menu
       autoFocus
       anchorEl={anchorEl}
-      id="account-menu"
       open={open}
       onClose={() => setOpen(false)}
       PaperProps={{

diff --git a/src/pyproject.toml b/src/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "chainlit"
-version = "0.1.101"
+version = "0.1.102"
 keywords = ['LLM', 'Agents', 'gen ai', 'chat ui', 'chatbot ui', 'langchain']
 description = "A faster way to build chatbot UIs."
 authors = ["Chainlit"]