generated from nhcarrigan/template
fix: tests
This commit is contained in:
@@ -9,6 +9,8 @@ describe("stateMapper", () => {
|
|||||||
type: "system",
|
type: "system",
|
||||||
subtype: "init",
|
subtype: "init",
|
||||||
session_id: "test-session",
|
session_id: "test-session",
|
||||||
|
cwd: "/home/test",
|
||||||
|
tools: ["Read", "Write", "Edit"],
|
||||||
};
|
};
|
||||||
expect(mapMessageToState(message)).toBe("idle");
|
expect(mapMessageToState(message)).toBe("idle");
|
||||||
});
|
});
|
||||||
@@ -16,8 +18,7 @@ describe("stateMapper", () => {
|
|||||||
it("returns null for non-init system messages", () => {
|
it("returns null for non-init system messages", () => {
|
||||||
const message: ClaudeStreamMessage = {
|
const message: ClaudeStreamMessage = {
|
||||||
type: "system",
|
type: "system",
|
||||||
subtype: "other",
|
subtype: "compact_boundary",
|
||||||
session_id: "test-session",
|
|
||||||
};
|
};
|
||||||
expect(mapMessageToState(message)).toBeNull();
|
expect(mapMessageToState(message)).toBeNull();
|
||||||
});
|
});
|
||||||
@@ -26,9 +27,6 @@ describe("stateMapper", () => {
|
|||||||
const message: ClaudeStreamMessage = {
|
const message: ClaudeStreamMessage = {
|
||||||
type: "assistant",
|
type: "assistant",
|
||||||
message: {
|
message: {
|
||||||
id: "test",
|
|
||||||
type: "message",
|
|
||||||
role: "assistant",
|
|
||||||
content: [
|
content: [
|
||||||
{
|
{
|
||||||
type: "tool_use",
|
type: "tool_use",
|
||||||
@@ -39,7 +37,6 @@ describe("stateMapper", () => {
|
|||||||
],
|
],
|
||||||
model: "claude-3",
|
model: "claude-3",
|
||||||
stop_reason: "tool_use",
|
stop_reason: "tool_use",
|
||||||
usage: { input_tokens: 100, output_tokens: 50 },
|
|
||||||
},
|
},
|
||||||
};
|
};
|
||||||
expect(mapMessageToState(message)).toBe("searching");
|
expect(mapMessageToState(message)).toBe("searching");
|
||||||
@@ -49,9 +46,6 @@ describe("stateMapper", () => {
|
|||||||
const message: ClaudeStreamMessage = {
|
const message: ClaudeStreamMessage = {
|
||||||
type: "assistant",
|
type: "assistant",
|
||||||
message: {
|
message: {
|
||||||
id: "test",
|
|
||||||
type: "message",
|
|
||||||
role: "assistant",
|
|
||||||
content: [
|
content: [
|
||||||
{
|
{
|
||||||
type: "tool_use",
|
type: "tool_use",
|
||||||
@@ -62,7 +56,6 @@ describe("stateMapper", () => {
|
|||||||
],
|
],
|
||||||
model: "claude-3",
|
model: "claude-3",
|
||||||
stop_reason: "tool_use",
|
stop_reason: "tool_use",
|
||||||
usage: { input_tokens: 100, output_tokens: 50 },
|
|
||||||
},
|
},
|
||||||
};
|
};
|
||||||
expect(mapMessageToState(message)).toBe("coding");
|
expect(mapMessageToState(message)).toBe("coding");
|
||||||
@@ -72,9 +65,6 @@ describe("stateMapper", () => {
|
|||||||
const message: ClaudeStreamMessage = {
|
const message: ClaudeStreamMessage = {
|
||||||
type: "assistant",
|
type: "assistant",
|
||||||
message: {
|
message: {
|
||||||
id: "test",
|
|
||||||
type: "message",
|
|
||||||
role: "assistant",
|
|
||||||
content: [
|
content: [
|
||||||
{
|
{
|
||||||
type: "tool_use",
|
type: "tool_use",
|
||||||
@@ -85,7 +75,6 @@ describe("stateMapper", () => {
|
|||||||
],
|
],
|
||||||
model: "claude-3",
|
model: "claude-3",
|
||||||
stop_reason: "tool_use",
|
stop_reason: "tool_use",
|
||||||
usage: { input_tokens: 100, output_tokens: 50 },
|
|
||||||
},
|
},
|
||||||
};
|
};
|
||||||
expect(mapMessageToState(message)).toBe("mcp");
|
expect(mapMessageToState(message)).toBe("mcp");
|
||||||
@@ -95,9 +84,6 @@ describe("stateMapper", () => {
|
|||||||
const message: ClaudeStreamMessage = {
|
const message: ClaudeStreamMessage = {
|
||||||
type: "assistant",
|
type: "assistant",
|
||||||
message: {
|
message: {
|
||||||
id: "test",
|
|
||||||
type: "message",
|
|
||||||
role: "assistant",
|
|
||||||
content: [
|
content: [
|
||||||
{
|
{
|
||||||
type: "tool_use",
|
type: "tool_use",
|
||||||
@@ -108,7 +94,6 @@ describe("stateMapper", () => {
|
|||||||
],
|
],
|
||||||
model: "claude-3",
|
model: "claude-3",
|
||||||
stop_reason: "tool_use",
|
stop_reason: "tool_use",
|
||||||
usage: { input_tokens: 100, output_tokens: 50 },
|
|
||||||
},
|
},
|
||||||
};
|
};
|
||||||
expect(mapMessageToState(message)).toBe("thinking");
|
expect(mapMessageToState(message)).toBe("thinking");
|
||||||
@@ -118,13 +103,9 @@ describe("stateMapper", () => {
|
|||||||
const message: ClaudeStreamMessage = {
|
const message: ClaudeStreamMessage = {
|
||||||
type: "assistant",
|
type: "assistant",
|
||||||
message: {
|
message: {
|
||||||
id: "test",
|
|
||||||
type: "message",
|
|
||||||
role: "assistant",
|
|
||||||
content: [{ type: "text", text: "Hello, Naomi!" }],
|
content: [{ type: "text", text: "Hello, Naomi!" }],
|
||||||
model: "claude-3",
|
model: "claude-3",
|
||||||
stop_reason: "end_turn",
|
stop_reason: "end_turn",
|
||||||
usage: { input_tokens: 100, output_tokens: 50 },
|
|
||||||
},
|
},
|
||||||
};
|
};
|
||||||
expect(mapMessageToState(message)).toBe("typing");
|
expect(mapMessageToState(message)).toBe("typing");
|
||||||
@@ -135,7 +116,6 @@ describe("stateMapper", () => {
|
|||||||
type: "result",
|
type: "result",
|
||||||
subtype: "success",
|
subtype: "success",
|
||||||
result: "Task completed",
|
result: "Task completed",
|
||||||
session_id: "test-session",
|
|
||||||
};
|
};
|
||||||
expect(mapMessageToState(message)).toBe("success");
|
expect(mapMessageToState(message)).toBe("success");
|
||||||
});
|
});
|
||||||
@@ -144,8 +124,6 @@ describe("stateMapper", () => {
|
|||||||
const message: ClaudeStreamMessage = {
|
const message: ClaudeStreamMessage = {
|
||||||
type: "result",
|
type: "result",
|
||||||
subtype: "error_max_turns",
|
subtype: "error_max_turns",
|
||||||
error: "Max turns exceeded",
|
|
||||||
session_id: "test-session",
|
|
||||||
};
|
};
|
||||||
expect(mapMessageToState(message)).toBe("error");
|
expect(mapMessageToState(message)).toBe("error");
|
||||||
});
|
});
|
||||||
@@ -153,7 +131,7 @@ describe("stateMapper", () => {
|
|||||||
it("returns null for user messages", () => {
|
it("returns null for user messages", () => {
|
||||||
const message: ClaudeStreamMessage = {
|
const message: ClaudeStreamMessage = {
|
||||||
type: "user",
|
type: "user",
|
||||||
message: { role: "user", content: "Hello" },
|
message: { content: [{ type: "text", text: "Hello" }] },
|
||||||
};
|
};
|
||||||
expect(mapMessageToState(message)).toBeNull();
|
expect(mapMessageToState(message)).toBeNull();
|
||||||
});
|
});
|
||||||
@@ -164,16 +142,12 @@ describe("stateMapper", () => {
|
|||||||
const message: ClaudeStreamMessage = {
|
const message: ClaudeStreamMessage = {
|
||||||
type: "assistant",
|
type: "assistant",
|
||||||
message: {
|
message: {
|
||||||
id: "test",
|
|
||||||
type: "message",
|
|
||||||
role: "assistant",
|
|
||||||
content: [
|
content: [
|
||||||
{ type: "text", text: "Hello!" },
|
{ type: "text", text: "Hello!" },
|
||||||
{ type: "text", text: "How are you?" },
|
{ type: "text", text: "How are you?" },
|
||||||
],
|
],
|
||||||
model: "claude-3",
|
model: "claude-3",
|
||||||
stop_reason: "end_turn",
|
stop_reason: "end_turn",
|
||||||
usage: { input_tokens: 100, output_tokens: 50 },
|
|
||||||
},
|
},
|
||||||
};
|
};
|
||||||
expect(extractTextFromMessage(message)).toBe("Hello!\nHow are you?");
|
expect(extractTextFromMessage(message)).toBe("Hello!\nHow are you?");
|
||||||
@@ -183,9 +157,6 @@ describe("stateMapper", () => {
|
|||||||
const message: ClaudeStreamMessage = {
|
const message: ClaudeStreamMessage = {
|
||||||
type: "assistant",
|
type: "assistant",
|
||||||
message: {
|
message: {
|
||||||
id: "test",
|
|
||||||
type: "message",
|
|
||||||
role: "assistant",
|
|
||||||
content: [
|
content: [
|
||||||
{
|
{
|
||||||
type: "tool_use",
|
type: "tool_use",
|
||||||
@@ -196,7 +167,6 @@ describe("stateMapper", () => {
|
|||||||
],
|
],
|
||||||
model: "claude-3",
|
model: "claude-3",
|
||||||
stop_reason: "tool_use",
|
stop_reason: "tool_use",
|
||||||
usage: { input_tokens: 100, output_tokens: 50 },
|
|
||||||
},
|
},
|
||||||
};
|
};
|
||||||
expect(extractTextFromMessage(message)).toBeNull();
|
expect(extractTextFromMessage(message)).toBeNull();
|
||||||
@@ -219,7 +189,6 @@ describe("stateMapper", () => {
|
|||||||
type: "result",
|
type: "result",
|
||||||
subtype: "success",
|
subtype: "success",
|
||||||
result: "Completed successfully",
|
result: "Completed successfully",
|
||||||
session_id: "test-session",
|
|
||||||
};
|
};
|
||||||
expect(extractTextFromMessage(message)).toBe("Completed successfully");
|
expect(extractTextFromMessage(message)).toBe("Completed successfully");
|
||||||
});
|
});
|
||||||
@@ -230,9 +199,6 @@ describe("stateMapper", () => {
|
|||||||
const message: ClaudeStreamMessage = {
|
const message: ClaudeStreamMessage = {
|
||||||
type: "assistant",
|
type: "assistant",
|
||||||
message: {
|
message: {
|
||||||
id: "test",
|
|
||||||
type: "message",
|
|
||||||
role: "assistant",
|
|
||||||
content: [
|
content: [
|
||||||
{
|
{
|
||||||
type: "tool_use",
|
type: "tool_use",
|
||||||
@@ -249,7 +215,6 @@ describe("stateMapper", () => {
|
|||||||
],
|
],
|
||||||
model: "claude-3",
|
model: "claude-3",
|
||||||
stop_reason: "tool_use",
|
stop_reason: "tool_use",
|
||||||
usage: { input_tokens: 100, output_tokens: 50 },
|
|
||||||
},
|
},
|
||||||
};
|
};
|
||||||
const tools = extractToolInfo(message);
|
const tools = extractToolInfo(message);
|
||||||
@@ -267,7 +232,7 @@ describe("stateMapper", () => {
|
|||||||
it("returns empty array for non-assistant messages", () => {
|
it("returns empty array for non-assistant messages", () => {
|
||||||
const message: ClaudeStreamMessage = {
|
const message: ClaudeStreamMessage = {
|
||||||
type: "user",
|
type: "user",
|
||||||
message: { role: "user", content: "Hello" },
|
message: { content: [{ type: "text", text: "Hello" }] },
|
||||||
};
|
};
|
||||||
expect(extractToolInfo(message)).toEqual([]);
|
expect(extractToolInfo(message)).toEqual([]);
|
||||||
});
|
});
|
||||||
|
|||||||
Reference in New Issue
Block a user