tasks_test.py

# Copyright 2015, Google, Inc.
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#    http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

import os

from gcp_devrel.testing import eventually_consistent
from gcp_devrel.testing.flaky import flaky
from google.cloud import datastore
import pytest

import tasks

PROJECT = os.environ['GCLOUD_PROJECT']


@pytest.yield_fixture
def client():
    client = datastore.Client(PROJECT)

    yield client

    # Delete anything created during the test.
    with client.batch():
        client.delete_multi(
            [x.key for x in client.query(kind='Task').fetch()])


@flaky
def test_create_client():
    tasks.create_client(PROJECT)


@flaky
def test_add_task(client):
    task_key = tasks.add_task(client, 'Test task')
    task = client.get(task_key)
    assert task
    assert task['description'] == 'Test task'


@flaky
def test_mark_done(client):
    task_key = tasks.add_task(client, 'Test task')
    tasks.mark_done(client, task_key.id)
    task = client.get(task_key)
    assert task
    assert task['done']


@flaky
def test_list_tasks(client):
    task1_key = tasks.add_task(client, 'Test task 1')
    task2_key = tasks.add_task(client, 'Test task 2')

    @eventually_consistent.call
    def _():
        task_list = tasks.list_tasks(client)
        assert [x.key for x in task_list] == [task1_key, task2_key]


@flaky
def test_delete_task(client):
    task_key = tasks.add_task(client, 'Test task 1')
    tasks.delete_task(client, task_key.id)
    assert client.get(task_key) is None


@flaky
def test_format_tasks(client):
    task1_key = tasks.add_task(client, 'Test task 1')
    tasks.add_task(client, 'Test task 2')
    tasks.mark_done(client, task1_key.id)

    @eventually_consistent.call
    def _():
        output = tasks.format_tasks(tasks.list_tasks(client))

        assert 'Test task 1' in output
        assert 'Test task 2' in output
        assert 'done' in output
        assert 'created' in output