File: gaetest_common.py

package info (click to toggle)
wtforms 2.0.1-1
  • links: PTS, VCS
  • area: main
  • in suites: jessie, jessie-kfreebsd
  • size: 964 kB
  • ctags: 1,360
  • sloc: python: 5,163; makefile: 73
file content (40 lines) | stat: -rw-r--r-- 951 bytes parent folder | download | duplicates (4)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
"""
This contains common tools for gae tests, and also sets up the environment.

It should be the first import in the unit tests.
"""
# -- First setup paths
import sys
import os
my_dir = os.path.dirname(os.path.abspath(__file__))
WTFORMS_DIR = os.path.abspath(os.path.join(my_dir, '..', '..'))
sys.path.insert(0, WTFORMS_DIR)

SAMPLE_AUTHORS = (
    ('Bob', 'Boston'),
    ('Harry', 'Houston'),
    ('Linda', 'London'),
)


class DummyPostData(dict):
    def getlist(self, key):
        v = self[key]
        if not isinstance(v, (list, tuple)):
            v = [v]
        return v


def fill_authors(Author):
    """
    Fill authors from SAMPLE_AUTHORS.
    Model is passed so it can be either an NDB or DB model.
    """
    AGE_BASE = 30
    authors = []
    for name, city in SAMPLE_AUTHORS:
        author = Author(name=name, city=city, age=AGE_BASE)
        author.put()
        authors.append(author)
        AGE_BASE += 1
    return authors