stacosys/tests/test_api.py

#!/usr/bin/env python
# -*- coding: UTF-8 -*-

import json
import logging

import pytest

from stacosys.db import dao, database
from stacosys.interface import api, app


def init_test_db():
    c1 = dao.create_comment("/site1", "Bob", "/bob.site", "", "comment 1")
    c2 = dao.create_comment("/site2", "Bill", "/bill.site", "", "comment 2")
    c3 = dao.create_comment("/site3", "Jack", "/jack.site", "", "comment 3")
    dao.publish_comment(c1)
    dao.publish_comment(c3)
    assert c2


@pytest.fixture
def client():
    logger = logging.getLogger(__name__)
    database.configure("sqlite:memory://db.sqlite")
    init_test_db()
    logger.info(f"start interface {api}")
    return app.test_client()


def test_api_ping(client):
    resp = client.get("/api/ping")
    assert resp.data == b"OK"


def test_api_count_global(client):
    resp = client.get("/api/comments/count")
    d = json.loads(resp.data)
    assert d and d["count"] == 2


def test_api_count_url(client):
    resp = client.get("/api/comments/count?url=/site1")
    d = json.loads(resp.data)
    assert d and d["count"] == 1
    resp = client.get("/api/comments/count?url=/site2")
    d = json.loads(resp.data)
    assert d and d["count"] == 0


def test_api_comment(client):
    resp = client.get("/api/comments?url=/site1")
    d = json.loads(resp.data)
    assert d and len(d["data"]) == 1
    comment = d["data"][0]
    assert comment["author"] == "Bob"
    assert comment["content"] == "comment 1"


def test_api_comment_not_found(client):
    resp = client.get("/api/comments?url=/site2")
    d = json.loads(resp.data)
    assert d and d["data"] == []