initial commit

tests/test_child_api.py

@@ -0,0 +1,179 @@
+import pytest
+import os
+from flask import Flask
+from api.child_api import child_api
+from db.db import child_db, reward_db, task_db
+from tinydb import Query
+from models.child import Child
+
+@pytest.fixture
+def client():
+    app = Flask(__name__)
+    app.register_blueprint(child_api)
+    app.config['TESTING'] = True
+    with app.test_client() as client:
+        yield client
+
+@pytest.fixture(scope="session", autouse=True)
+def cleanup_db():
+    yield
+    child_db.close()
+    reward_db.close()
+    task_db.close()
+    if os.path.exists('children.json'):
+        os.remove('children.json')
+    if os.path.exists('rewards.json'):
+        os.remove('rewards.json')
+    if os.path.exists('tasks.json'):
+        os.remove('tasks.json')
+
+def test_add_child(client):
+    response = client.put('/child/add', json={'name': 'Alice', 'age': 8})
+    assert response.status_code == 201
+    assert b'Child Alice added.' in response.data
+    children = child_db.all()
+    assert any(c.get('name') == 'Alice' and c.get('age') == 8 and c.get('image_id') == 'boy01' for c in children)
+
+    response = client.put('/child/add', json={'name': 'Mike', 'age': 4, 'image_id': 'girl02'})
+    assert response.status_code == 201
+    assert b'Child Mike added.' in response.data
+    children = child_db.all()
+    assert any(c.get('name') == 'Mike' and c.get('age') == 4 and c.get('image_id') == 'girl02' for c in children)
+
+def test_list_children(client):
+    child_db.truncate()
+    child_db.insert(Child(name='Alice', age=8, image_id="boy01").to_dict())
+    child_db.insert(Child(name='Mike', age=4, image_id="boy01").to_dict())
+    response = client.get('/child/list')
+    assert response.status_code == 200
+    data = response.json
+    assert len(data['children']) == 2
+
+def test_assign_and_remove_task(client):
+    client.put('/child/add', json={'name': 'Bob', 'age': 10})
+    children = client.get('/child/list').get_json()['children']
+    child_id = children[0]['id']
+    response = client.post(f'/child/{child_id}/assign-task', json={'task_id': 'task123'})
+    assert response.status_code == 200
+    ChildQuery = Query()
+    child = child_db.search(ChildQuery.id == child_id)[0]
+    assert 'task123' in child.get('tasks', [])
+    response = client.post(f'/child/{child_id}/remove-task', json={'task_id': 'task123'})
+    assert response.status_code == 200
+    child = child_db.search(ChildQuery.id == child_id)[0]
+    assert 'task123' not in child.get('tasks', [])
+
+def test_get_child_not_found(client):
+    response = client.get('/child/nonexistent-id')
+    assert response.status_code == 404
+    assert b'Child not found' in response.data
+
+def test_remove_task_not_found(client):
+    response = client.post('/child/nonexistent-id/remove-task', json={'task_id': 'task123'})
+    assert response.status_code == 404
+    assert b'Child not found' in response.data
+
+def test_remove_reward_not_found(client):
+    response = client.post('/child/nonexistent-id/remove-reward', json={'reward_id': 'reward123'})
+    assert response.status_code == 404
+    assert b'Child not found' in response.data
+
+def test_affordable_rewards(client):
+    reward_db.insert({'id': 'r_cheep', 'name': 'Sticker', 'cost': 5})
+    reward_db.insert({'id': 'r_exp', 'name': 'Bike', 'cost': 20})
+    client.put('/child/add', json={'name': 'Charlie', 'age': 9})
+    children = client.get('/child/list').get_json()['children']
+    child_id = next((c['id'] for c in children if c.get('name') == 'Charlie'), children[0]['id'])
+    ChildQuery = Query()
+    child_db.update({'points': 10}, ChildQuery.id == child_id)
+    client.post(f'/child/{child_id}/assign-reward', json={'reward_id': 'r_cheep'})
+    client.post(f'/child/{child_id}/assign-reward', json={'reward_id': 'r_exp'})
+    resp = client.get(f'/child/{child_id}/affordable-rewards')
+    assert resp.status_code == 200
+    data = resp.get_json()
+    affordable_ids = [r['id'] for r in data['affordable_rewards']]
+    assert 'r_cheep' in affordable_ids and 'r_exp' not in affordable_ids
+
+def test_reward_status(client):
+    reward_db.insert({'id': 'r1', 'name': 'Candy', 'cost': 3, 'image': 'candy.png'})
+    reward_db.insert({'id': 'r2', 'name': 'Game', 'cost': 8, 'image': 'game.png'})
+    reward_db.insert({'id': 'r3', 'name': 'Trip', 'cost': 15, 'image': 'trip.png'})
+    client.put('/child/add', json={'name': 'Dana', 'age': 11})
+    children = client.get('/child/list').get_json()['children']
+    child_id = next((c['id'] for c in children if c.get('name') == 'Dana'), children[0]['id'])
+    ChildQuery = Query()
+    child_db.update({'points': 7}, ChildQuery.id == child_id)
+    for rid in ['r1', 'r2', 'r3']:
+        client.post(f'/child/{child_id}/assign-reward', json={'reward_id': rid})
+    resp = client.get(f'/child/{child_id}/reward-status')
+    assert resp.status_code == 200
+    data = resp.get_json()
+    mapping = {s['id']: s['points_needed'] for s in data['reward_status']}
+    assert mapping['r1'] == 0 and mapping['r2'] == 1 and mapping['r3'] == 8
+
+def test_list_child_tasks_returns_tasks(client):
+    task_db.insert({'id': 't_list_1', 'name': 'Task One', 'points': 2, 'is_good': True})
+    task_db.insert({'id': 't_list_2', 'name': 'Task Two', 'points': 3, 'is_good': False})
+    child_db.insert({
+        'id': 'child_list_1',
+        'name': 'Eve',
+        'age': 8,
+        'points': 0,
+        'tasks': ['t_list_1', 't_list_2', 't_missing'],
+        'rewards': []
+    })
+    resp = client.get('/child/child_list_1/list-tasks')
+    assert resp.status_code == 200
+    data = resp.get_json()
+    returned_ids = {t['id'] for t in data['child_tasks']}
+    assert returned_ids == {'t_list_1', 't_list_2'}
+    for t in data['child_tasks']:
+        assert 'name' in t and 'points' in t and 'is_good' in t
+
+def test_list_assignable_tasks_returns_expected_ids(client):
+    child_db.truncate()
+    task_db.truncate()
+    task_db.insert({'id': 'tA', 'name': 'Task A', 'points': 1, 'is_good': True})
+    task_db.insert({'id': 'tB', 'name': 'Task B', 'points': 2, 'is_good': True})
+    task_db.insert({'id': 'tC', 'name': 'Task C', 'points': 3, 'is_good': False})
+    client.put('/child/add', json={'name': 'Zoe', 'age': 7})
+    child_id = client.get('/child/list').get_json()['children'][0]['id']
+    client.post(f'/child/{child_id}/assign-task', json={'task_id': 'tA'})
+    client.post(f'/child/{child_id}/assign-task', json={'task_id': 'tC'})
+    resp = client.get(f'/child/{child_id}/list-assignable-tasks')
+    assert resp.status_code == 200
+    data = resp.get_json()
+    assert len(data['assignable_tasks']) == 1
+    assert data['assignable_tasks'][0]['id'] == 'tB'
+    assert data['count'] == 1
+
+def test_list_assignable_tasks_when_none_assigned(client):
+    child_db.truncate()
+    task_db.truncate()
+    ids = ['t1', 't2', 't3']
+    for i, tid in enumerate(ids, 1):
+        task_db.insert({'id': tid, 'name': f'Task {i}', 'points': i, 'is_good': True})
+    client.put('/child/add', json={'name': 'Liam', 'age': 6})
+    child_id = client.get('/child/list').get_json()['children'][0]['id']
+    resp = client.get(f'/child/{child_id}/list-assignable-tasks')
+    assert resp.status_code == 200
+    data = resp.get_json()
+    returned_ids = {t['id'] for t in data['assignable_tasks']}
+    assert returned_ids == set(ids)
+    assert data['count'] == len(ids)
+
+def test_list_assignable_tasks_empty_task_db(client):
+    child_db.truncate()
+    task_db.truncate()
+    client.put('/child/add', json={'name': 'Mia', 'age': 5})
+    child_id = client.get('/child/list').get_json()['children'][0]['id']
+    resp = client.get(f'/child/{child_id}/list-assignable-tasks')
+    assert resp.status_code == 200
+    data = resp.get_json()
+    assert data['assignable_tasks'] == []
+    assert data['count'] == 0
+
+def test_list_assignable_tasks_child_not_found(client):
+    resp = client.get('/child/does-not-exist/list-assignable-tasks')
+    assert resp.status_code == 404
+    assert b'Child not found' in resp.data