systemprompt-mcp-notion
Version:
A specialized Model Context Protocol (MCP) server that integrates Notion into your AI workflows. This server enables seamless access to Notion through MCP, allowing AI agents to interact with pages, databases, and comments.
256 lines (255 loc) • 9.62 kB
JavaScript
import { SystemPromptService } from "../systemprompt-service.js";
// Mock fetch
const mockFetch = jest.fn();
global.fetch = mockFetch;
describe("SystemPromptService", () => {
let service;
const mockApiKey = "test-api-key";
beforeEach(() => {
jest.clearAllMocks();
process.env.SYSTEMPROMPT_API_KEY = mockApiKey;
service = new SystemPromptService();
});
afterEach(() => {
delete process.env.SYSTEMPROMPT_API_KEY;
});
describe("createPrompt", () => {
const validPromptData = {
instruction: {
static: "Test instruction",
state: "Test state",
dynamic: "Test dynamic",
},
input: {
name: "test_input",
description: "Test input description",
type: ["message"],
},
output: {
name: "test_output",
description: "Test output description",
type: ["message"],
},
metadata: {
title: "Test prompt",
description: "Test description",
tag: ["test"],
},
};
it("should create a prompt successfully", async () => {
const mockResponse = {
id: "test-uuid",
...validPromptData,
_link: "test-link",
metadata: {
...validPromptData.metadata,
created: "2024-01-01",
updated: "2024-01-01",
version: 1,
status: "draft",
author: "test",
log_message: "Created",
},
};
mockFetch.mockResolvedValueOnce({
ok: true,
json: () => Promise.resolve(mockResponse),
});
const result = await service.createPrompt(validPromptData);
expect(mockFetch).toHaveBeenCalledWith("https://api.systemprompt.io/v1/prompt", {
method: "POST",
headers: {
"Content-Type": "application/json",
"api-key": mockApiKey,
},
body: JSON.stringify(validPromptData),
});
expect(result).toEqual(mockResponse);
});
it("should handle API errors", async () => {
const errorMessage = "Invalid input";
mockFetch.mockResolvedValueOnce({
ok: false,
json: () => Promise.resolve({ message: errorMessage }),
});
await expect(service.createPrompt(validPromptData)).rejects.toThrow(errorMessage);
});
});
describe("editPrompt", () => {
const uuid = "test-uuid";
const updateData = {
instruction: {
static: "Updated instruction",
},
metadata: {
title: "Updated title",
description: "Test description",
},
};
it("should edit a prompt successfully", async () => {
const mockResponse = {
id: uuid,
instruction: { static: "Updated instruction" },
input: {
name: "test_input",
description: "Test input description",
type: ["message"],
},
output: {
name: "test_output",
description: "Test output description",
type: ["message"],
},
metadata: {
title: "Updated title",
description: "Test description",
created: "2024-01-01",
updated: "2024-01-01",
version: 1,
status: "draft",
author: "test",
log_message: "Updated",
},
_link: "test-link",
};
mockFetch.mockResolvedValueOnce({
ok: true,
json: () => Promise.resolve(mockResponse),
});
const result = await service.editPrompt(uuid, updateData);
expect(mockFetch).toHaveBeenCalledWith(`https://api.systemprompt.io/v1/prompt/${uuid}`, {
method: "PUT",
headers: {
"Content-Type": "application/json",
"api-key": mockApiKey,
},
body: JSON.stringify(updateData),
});
expect(result).toEqual(mockResponse);
});
it("should handle API errors", async () => {
const errorMessage = "Prompt not found";
mockFetch.mockResolvedValueOnce({
ok: false,
json: () => Promise.resolve({ message: errorMessage }),
});
await expect(service.editPrompt(uuid, updateData)).rejects.toThrow(errorMessage);
});
});
describe("createBlock", () => {
const validBlockData = {
content: "Test block content",
prefix: "test_block",
metadata: {
title: "Test block",
description: "Test description",
tag: ["test"],
},
};
it("should create a block successfully", async () => {
const mockResponse = {
id: "test-uuid",
...validBlockData,
_link: "test-link",
metadata: {
...validBlockData.metadata,
created: "2024-01-01",
updated: "2024-01-01",
version: 1,
status: "draft",
author: "test",
log_message: "Created",
},
};
mockFetch.mockResolvedValueOnce({
ok: true,
json: () => Promise.resolve(mockResponse),
});
const result = await service.createBlock(validBlockData);
expect(mockFetch).toHaveBeenCalledWith("https://api.systemprompt.io/v1/block", {
method: "POST",
headers: {
"Content-Type": "application/json",
"api-key": mockApiKey,
},
body: JSON.stringify(validBlockData),
});
expect(result).toEqual(mockResponse);
});
it("should handle API errors", async () => {
const errorMessage = "Invalid input";
mockFetch.mockResolvedValueOnce({
ok: false,
json: () => Promise.resolve({ message: errorMessage }),
});
await expect(service.createBlock(validBlockData)).rejects.toThrow(errorMessage);
});
});
describe("editBlock", () => {
const uuid = "test-uuid";
const updateData = {
content: "Updated content",
metadata: {
title: "Updated title",
description: "Test description",
},
};
it("should edit a block successfully", async () => {
const mockResponse = {
id: uuid,
content: "Updated content",
metadata: {
title: "Updated title",
description: "Test description",
created: "2024-01-01",
updated: "2024-01-01",
version: 1,
status: "draft",
author: "test",
log_message: "Updated",
},
_link: "test-link",
};
mockFetch.mockResolvedValueOnce({
ok: true,
json: () => Promise.resolve(mockResponse),
});
const result = await service.editBlock(uuid, updateData);
expect(mockFetch).toHaveBeenCalledWith(`https://api.systemprompt.io/v1/block/${uuid}`, {
method: "PUT",
headers: {
"Content-Type": "application/json",
"api-key": mockApiKey,
},
body: JSON.stringify(updateData),
});
expect(result).toEqual(mockResponse);
});
it("should handle API errors", async () => {
const errorMessage = "Block not found";
mockFetch.mockResolvedValueOnce({
ok: false,
json: () => Promise.resolve({ message: errorMessage }),
});
await expect(service.editBlock(uuid, updateData)).rejects.toThrow(errorMessage);
});
});
describe("error handling", () => {
it("should handle network errors", async () => {
mockFetch.mockRejectedValueOnce(new Error("Network error"));
await expect(service.createPrompt({})).rejects.toThrow("Network error");
});
it("should handle invalid JSON responses", async () => {
mockFetch.mockResolvedValueOnce({
ok: false,
json: () => Promise.reject(new Error("Invalid JSON")),
});
await expect(service.createPrompt({})).rejects.toThrow("API request failed");
});
it("should handle missing API key", () => {
delete process.env.SYSTEMPROMPT_API_KEY;
const service = new SystemPromptService();
expect(service["apiKey"]).toBe("");
});
});
});