def test_noderequest_cookies(self):
"""
"""
f = open(os.path.join(self.tmp_template_dir, 'select_llama.sql'), 'w')
f.write("""
select :llama as llama;
""")
f.close()
with self.app.app_context():
with self.app.test_client() as c:
init_db()
page = insert_node(name='page', value=None)
insert_route(path='/page/', node_id=page)
llama = insert_node(name='llama', value=None)
insert_node_node(node_id=page, target_node_id=llama)
insert_query(name='select_llama.sql', node_id=llama)
c.set_cookie('localhost', 'llama', 'chuck')
rv = c.get('/page/', follow_redirects=True)
assert 200 == rv.status_code
self.app.logger.debug('test: %s', rv.data)
assert 'chuck' in rv.data
def test_noderequest_cookies(self):
"""
"""
f = open(os.path.join(self.tmp_template_dir, 'select_llama.sql'), 'w')
f.write("""
select :llama as llama;
""")
f.close()
with self.app.app_context():
with self.app.test_client() as c:
init_db()
page = insert_node(name='page', value=None)
insert_route(path='/page/', node_id=page)
llama = insert_node(name='llama', value=None)
insert_node_node(node_id=page, target_node_id=llama)
insert_query(name='select_llama.sql', node_id=llama)
c.set_cookie('localhost', 'llama', 'chuck')
rv = c.get('/page/', follow_redirects=True)
assert 200 == rv.status_code
self.app.logger.debug('test: %s', rv.data)
assert 'chuck' in rv.data
def test_some_unicode_as_value_in_template(self):
"""
"""
f = open(os.path.join(self.tmp_template_dir, 'template_unicode.html'), 'w')
f.write("""
<h1>template_unicode</h1>
{{ isit|safe }}
""")
f.close()
f = open(os.path.join(self.tmp_template_dir, 'isit.html'), 'w')
f.write("""
<div>template with a unicode {{ value }}</div>
""")
f.close()
with self.app.app_context():
with self.app.test_client() as c:
init_db()
test_id = insert_node(name='page', value=None)
insert_route(path='/test/1/', node_id=test_id)
add_template_for_node('template_unicode.html', test_id)
isit = insert_node(name='isit', value=u'Àрpĺè')
add_template_for_node('isit.html', isit)
insert_node_node(node_id=test_id, target_node_id=isit)
rv = c.get('/test/1/', follow_redirects=True)
assert u'Àрpĺè' in rv.data.decode('utf-8')
def test_noderequest(self):
"""
"""
f = open(os.path.join(self.tmp_template_dir, 'select_pagenames.sql'),
'w')
f.write("""
select 'yup' as test where :pagename in ('apple', 'pear', 'grapes');
""")
f.close()
with self.app.app_context():
with self.app.test_client() as c:
init_db()
page = insert_node(name='page', value=None)
insert_route(path='/page/<pagename>/', node_id=page)
pagenames = insert_node(name='pagenames', value=None)
insert_node_node(node_id=page, target_node_id=pagenames)
insert_query(name='select_pagenames.sql', node_id=pagenames)
rv = c.get('/page/cucumber/', follow_redirects=True)
assert 200 == rv.status_code
self.app.logger.debug('test: %s', rv.data)
assert 'yup' not in rv.data
rv = c.get('/page/pear/', follow_redirects=True)
assert 200 == rv.status_code
self.app.logger.debug('test: %s', rv.data)
assert 'yup' in rv.data
def test_some_unicode_as_value_in_template(self):
"""
"""
f = open(os.path.join(self.tmp_template_dir, 'template_unicode.html'),
'w')
f.write("""
<h1>template_unicode</h1>
{{ isit|safe }}
""")
f.close()
f = open(os.path.join(self.tmp_template_dir, 'isit.html'), 'w')
f.write("""
<div>template with a unicode {{ value }}</div>
""")
f.close()
with self.app.app_context():
with self.app.test_client() as c:
init_db()
test_id = insert_node(name='page', value=None)
insert_route(path='/test/1/', node_id=test_id)
add_template_for_node('template_unicode.html', test_id)
isit = insert_node(name='isit', value=u'Àрpĺè')
add_template_for_node('isit.html', isit)
insert_node_node(node_id=test_id, target_node_id=isit)
rv = c.get('/test/1/', follow_redirects=True)
assert u'Àрpĺè' in rv.data.decode('utf-8')
def test_reading_in_a_document(self):
"""
The custom 'readfile' jinja2 filter reads the file from the DOCUMENT_FOLDER.
"""
f = open(os.path.join(self.tmp_template_dir, 'imasimplefile.txt'), 'w')
f.write("""
Hello, this is just a file.
""")
f.close()
f = open(os.path.join(self.tmp_template_dir, 'template.html'), 'w')
f.write("""
<h1>template</h1>
{{ simplefilename }}
<br>
{{ simplefilename|readfile }}
""")
f.close()
with self.app.app_context():
with self.app.test_client() as c:
init_db()
a = insert_node(name='simplefilename',
value='imasimplefile.txt')
apage = insert_node(name='apage', value=None)
insert_node_node(node_id=apage, target_node_id=a)
insert_route(path='/a/', node_id=apage)
add_template_for_node('template.html', apage)
rv = c.get('/a/', follow_redirects=True)
assert 'Hello' in rv.data
def test_noderequest(self):
"""
"""
f = open(os.path.join(self.tmp_template_dir, 'select_pagenames.sql'), 'w')
f.write("""
select 'yup' as test where :pagename in ('apple', 'pear', 'grapes');
""")
f.close()
with self.app.app_context():
with self.app.test_client() as c:
init_db()
page = insert_node(name='page', value=None)
insert_route(path='/page/<pagename>/', node_id=page)
pagenames = insert_node(name='pagenames', value=None)
insert_node_node(node_id=page, target_node_id=pagenames)
insert_query(name='select_pagenames.sql', node_id=pagenames)
rv = c.get('/page/cucumber/', follow_redirects=True)
assert 200 == rv.status_code
self.app.logger.debug('test: %s', rv.data)
assert 'yup' not in rv.data
rv = c.get('/page/pear/', follow_redirects=True)
assert 200 == rv.status_code
self.app.logger.debug('test: %s', rv.data)
assert 'yup' in rv.data
def test_reading_in_a_document(self):
"""
The custom 'readfile' jinja2 filter reads the file from the DOCUMENT_FOLDER.
"""
f = open(os.path.join(self.tmp_template_dir, 'imasimplefile.txt'), 'w')
f.write("""
Hello, this is just a file.
""")
f.close()
f = open(os.path.join(self.tmp_template_dir, 'template.html'), 'w')
f.write("""
<h1>template</h1>
{{ simplefilename }}
<br>
{{ simplefilename|readfile }}
""")
f.close()
with self.app.app_context():
with self.app.test_client() as c:
init_db()
a = insert_node(name='simplefilename', value='imasimplefile.txt')
apage = insert_node(name='apage', value=None)
insert_node_node(node_id=apage, target_node_id=a)
insert_route(path='/a/', node_id=apage)
add_template_for_node('template.html', apage)
rv = c.get('/a/', follow_redirects=True)
assert 'Hello' in rv.data
def add_item_with_attributes_to_collection(collection_name, collection_node_id, item_attr_list):
item_node_id = insert_node(name='{0}_item'.format(collection_name), value=None)
insert_query(name='select_link_node_from_node.sql', node_id=item_node_id)
insert_node_node(node_id=collection_node_id, target_node_id=item_node_id)
for item_attr_name in item_attr_list:
value = raw_input("Enter item attribute value for '{0}': ".format(item_attr_name))
# set value to none if it's an empty string
value = value if len(value) else None
item_attr_node_id = insert_node(name=item_attr_name, value=value)
insert_node_node(node_id=item_node_id, target_node_id=item_attr_node_id)
def test_simple_use_case(self):
"Add a picture and apply a template."
f = open(os.path.join(self.tmp_template_dir, 'simple.html'), 'w')
f.write("""
<!doctype html>
<html><head><title>test</title></head>
<body>
<div>
{{ cat|safe }}
</div>
</body>
</html>
""")
f.close()
f = open(os.path.join(self.tmp_template_dir, 'img.html'), 'w')
f.write("""
<img src="{{ url_for('send_media_file', filename=path) }}"/>
""")
f.close()
with self.app.app_context():
with self.app.test_client() as c:
init_db()
init_picture_tables()
# Create a blank jpg
catjpg = open(os.path.join(self.tmp_template_dir, 'cat.jpg'), 'wb')
img = Image.new("RGB", (300,200))
img.save(fp=catjpg)
# The 'cat' node is what will link the cat.jpg to.
cat = insert_node(name="cat", value=None)
add_picture_for_node(node_id=cat, filepath='cat.jpg')
# Set a img template around the 'cat'
add_template_for_node('img.html', cat)
# Setup a page to show the cat.jpg with the img template
page = insert_node(name='page', value=None)
insert_route(path='/cat/', node_id=page)
add_template_for_node('simple.html', page)
# Link the page with cat node
insert_node_node(node_id=page, target_node_id=cat)
rv = c.get('/cat/', follow_redirects=True)
assert '<title>test</title>' in rv.data
assert '<img src="/media/cat.jpg"/>' in rv.data
def init():
"Initialize the current directory with base starting files and database."
if not os.path.exists('site.cfg'):
f = open('site.cfg', 'w')
f.write(SITECFG)
f.close()
try:
os.mkdir('queries')
except OSError:
pass
try:
os.mkdir('templates')
except OSError:
pass
htmlfile = os.path.join('templates', 'homepage.html')
if not os.path.exists(htmlfile):
f = open(htmlfile, 'w')
f.write("""
<!doctype html>
<html>
<head>
<title>Chill</title>
</head>
<body>
<p>{{ homepage_content }}</p>
</body>
</html>
""")
f.close()
app = make_app(config='site.cfg', DEBUG=True)
with app.app_context():
app.logger.info("initializing database")
init_db()
homepage = insert_node(name='homepage', value=None)
insert_route(path='/', node_id=homepage)
insert_query(name='select_link_node_from_node.sql', node_id=homepage)
add_template_for_node('homepage.html', homepage)
homepage_content = insert_node(name='homepage_content', value="Cascading, Highly Irrelevant, Lost Llamas")
insert_node_node(node_id=homepage, target_node_id=homepage_content)
def test_unicode_value(self):
"""
"""
with self.app.app_context():
with self.app.test_client() as c:
init_db()
a_id = insert_node(name='a', value=None)
insert_route(path='/', node_id=a_id)
content = insert_node(name='content', value=u'Àрpĺè')
insert_node_node(node_id=a_id, target_node_id=content)
rv = c.get('/', follow_redirects=True)
assert 200 == rv.status_code
assert '\u00c0\u0440p\u013a\u00e8' in rv.data.decode('utf-8')
def test_unicode_value(self):
"""
"""
with self.app.app_context():
with self.app.test_client() as c:
init_db()
a_id = insert_node(name='a', value=None)
insert_route(path='/', node_id=a_id)
content = insert_node(name='content', value=u'Àрpĺè')
insert_node_node(node_id=a_id, target_node_id=content)
rv = c.get('/', follow_redirects=True)
assert 200 == rv.status_code
assert '\u00c0\u0440p\u013a\u00e8' in rv.data.decode('utf-8')
def test_value(self):
"""
"""
with self.app.app_context():
with self.app.test_client() as c:
init_db()
a_id = insert_node(name='a', value=None)
insert_route(path='/', node_id=a_id)
content = insert_node(name='content', value='apple')
insert_node_node(node_id=a_id, target_node_id=content)
rv = c.get('/', follow_redirects=True)
assert 200 == rv.status_code
#self.app.logger.debug('test: %s', rv.data)
assert 'apple' in rv.data
def test_value(self):
"""
"""
with self.app.app_context():
with self.app.test_client() as c:
init_db()
a_id = insert_node(name='a', value=None)
insert_route(path='/', node_id=a_id)
content = insert_node(name='content', value='apple')
insert_node_node(node_id=a_id, target_node_id=content)
rv = c.get('/', follow_redirects=True)
assert 200 == rv.status_code
#self.app.logger.debug('test: %s', rv.data)
assert 'apple' in rv.data
def test_page_uri(self):
"Expand the page_uri shortcode"
f = open(os.path.join(self.tmp_template_dir, 'simple.html'), 'w')
f.write("""
<!doctype html>
<html><head><title>test</title></head>
<body>
<div>
{{ cat|shortcodes }}
</div>
</body>
</html>
""")
f.close()
with self.app.app_context():
with self.app.test_client() as c:
init_db()
page = insert_node(name='page', value=None)
insert_route(path='/', node_id=page)
add_template_for_node('simple.html', page)
catpage = insert_node(name='acat', value="a page for cat")
insert_route(path='/cat/', node_id=catpage)
text = "something link for cat page that does exist = '[chill page_uri cat ]' link for dog that does not exist = '[chill page_uri dog]'"
textnode = insert_node(name='cat', value=text)
insert_node_node(node_id=page, target_node_id=textnode)
rv = c.get('/', follow_redirects=True)
assert "something link for cat page that does exist = '/cat/' link for dog that does not exist = '/dog/'" in rv.data
assert "[chill page_uri cat ]" not in rv.data
rv = c.get('/cat/', follow_redirects=True)
assert "a page for cat" in rv.data
rv = c.get('/dog/', follow_redirects=True)
assert 404 == rv.status_code
def test_page_uri(self):
"Expand the page_uri shortcode"
f = open(os.path.join(self.tmp_template_dir, 'simple.html'), 'w')
f.write("""
<!doctype html>
<html><head><title>test</title></head>
<body>
<div>
{{ cat|shortcodes }}
</div>
</body>
</html>
""")
f.close()
with self.app.app_context():
with self.app.test_client() as c:
init_db()
page = insert_node(name='page', value=None)
insert_route(path='/', node_id=page)
add_template_for_node('simple.html', page)
catpage = insert_node(name='acat', value="a page for cat")
insert_route(path='/cat/', node_id=catpage)
text = "something link for cat page that does exist = '[chill page_uri cat ]' link for dog that does not exist = '[chill page_uri dog]'"
textnode = insert_node(name='cat', value=text)
insert_node_node(node_id=page, target_node_id=textnode)
rv = c.get('/', follow_redirects=True)
assert "something link for cat page that does exist = '/cat/' link for dog that does not exist = '/dog/'" in rv.data
assert "[chill page_uri cat ]" not in rv.data
rv = c.get('/cat/', follow_redirects=True)
assert "a page for cat" in rv.data
rv = c.get('/dog/', follow_redirects=True)
assert 404 == rv.status_code
def test_add_and_link(self):
"Add a picture to the database and link it to two nodes"
with self.app.app_context():
with self.app.test_client() as c:
init_db()
init_picture_tables()
apage = insert_node(name="apage", value=None)
insert_route(path='/', node_id=apage)
for f in ('apple 200 300', 'banana 500 100', 'carrot 100 30'):
(name, width, height) = f.split(' ')
width = int(width)
height = int(height)
node = insert_node(name=name, value=None)
# Create a.jpg in tmp dir media_folder
jpg = open(os.path.join(self.tmp_template_dir, '{0}.jpg'.format(name)), 'wb')
img = Image.new("RGB", (width,height))
img.save(fp=jpg)
add_picture_for_node(node_id=node, filepath='{0}.jpg'.format(name))
insert_node_node(node_id=apage, target_node_id=node)
# Make a 'pictures' node and add all the pictures to it.
pictures = insert_node(name='pictures', value=None)
for name in ('apple.jpg', 'banana.jpg', 'carrot.jpg'):
link_picturename_for_node(node_id=pictures, picturename=name)
insert_node_node(node_id=apage, target_node_id=pictures)
rv = c.get('/', follow_redirects=True)
#self.app.logger.debug('test: %s', rv.data)
rv_json = json.loads(rv.data)
assert 200 == rv_json['apple']['width']
assert 500 == rv_json['banana']['width']
assert 100 == rv_json['banana']['height']
assert 30 == rv_json['carrot']['height']
assert 'apple.jpg' in [x['path'] for x in rv_json['pictures']]
def test_link(self):
"""
Link to any node
"""
with self.app.app_context():
init_db()
a_id = insert_node(name='a', value=None)
b_id = insert_node(name='b', value=None)
c_id = insert_node(name='c', value="c")
d_id = insert_node(name='d', value="d")
# a -> c, b -> c
# a -> d
insert_node_node(node_id=a_id, target_node_id=c_id)
insert_node_node(node_id=a_id, target_node_id=d_id)
insert_node_node(node_id=b_id, target_node_id=c_id)
result = db.execute(text(fetch_query_string('select_link_node_from_node.sql')), node_id=a_id)
result = [x['node_id'] for x in result]
assert c_id in result
assert d_id in result
assert a_id not in result
result = db.execute(text(fetch_query_string('select_link_node_from_node.sql')), node_id=b_id)
result = [x['node_id'] for x in result]
assert c_id in result
assert d_id not in result
assert a_id not in result
def test_link(self):
"""
Link to any node
"""
with self.app.app_context():
init_db()
a_id = insert_node(name='a', value=None)
b_id = insert_node(name='b', value=None)
c_id = insert_node(name='c', value="c")
d_id = insert_node(name='d', value="d")
# a -> c, b -> c
# a -> d
insert_node_node(node_id=a_id, target_node_id=c_id)
insert_node_node(node_id=a_id, target_node_id=d_id)
insert_node_node(node_id=b_id, target_node_id=c_id)
result = db.query(
fetch_query_string('select_link_node_from_node.sql'),
fetchall=True,
**{'node_id': a_id})
result = [x.get('node_id', None) for x in result]
assert c_id in result
assert d_id in result
assert a_id not in result
result = db.query(
fetch_query_string('select_link_node_from_node.sql'),
fetchall=True,
**{'node_id': b_id})
result = [x.get('node_id', None) for x in result]
assert c_id in result
assert d_id not in result
assert a_id not in result
def test_link(self):
"""
Link to any node
"""
with self.app.app_context():
init_db()
a_id = insert_node(name='a', value=None)
b_id = insert_node(name='b', value=None)
c_id = insert_node(name='c', value="c")
d_id = insert_node(name='d', value="d")
# a -> c, b -> c
# a -> d
insert_node_node(node_id=a_id, target_node_id=c_id)
insert_node_node(node_id=a_id, target_node_id=d_id)
insert_node_node(node_id=b_id, target_node_id=c_id)
c = db.cursor()
result = c.execute(fetch_query_string('select_link_node_from_node.sql'), {'node_id': a_id}).fetchall()
(result, col_names) = rowify(result, c.description)
result = [x.get('node_id', None) for x in result]
assert c_id in result
assert d_id in result
assert a_id not in result
result = c.execute(fetch_query_string('select_link_node_from_node.sql'), {'node_id': b_id}).fetchall()
(result, col_names) = rowify(result, c.description)
result = [x.get('node_id', None) for x in result]
assert c_id in result
assert d_id not in result
assert a_id not in result
def test_route(self):
"Expand the route shortcode"
f = open(os.path.join(self.tmp_template_dir, 'simple.html'), 'w')
f.write("""
<!doctype html>
<html><head><title>test</title></head>
<body>
<div>
{{ cat|shortcodes }}
</div>
</body>
</html>
""")
f.close()
with self.app.app_context():
with self.app.test_client() as c:
init_db()
page = insert_node(name='page', value=None)
insert_route(path='/', node_id=page)
add_template_for_node('simple.html', page)
catimg = insert_node(name='acat',
value="<img alt='a picture of a cat'/>")
insert_route(path='/cat/picture/', node_id=catimg)
text = "something [chill route /cat/picture/ ] [blah blah[chill route /dog/pic] the end"
textnode = insert_node(name='cat', value=text)
insert_node_node(node_id=page, target_node_id=textnode)
rv = c.get('/', follow_redirects=True)
assert "something <img alt='a picture of a cat'/> [blah blah<!-- 404 '/dog/pic' --> the end" in rv.data
assert "[chill route /cat/picture/ ]" not in rv.data
rv = c.get('/cat/picture/', follow_redirects=True)
assert "<img alt='a picture of a cat'/>" in rv.data
def test_route_with_unicode(self):
"Expand the route shortcode with unicode contents"
f = open(os.path.join(self.tmp_template_dir, 'simple.html'), 'w')
f.write("""
<!doctype html>
<html><head><title>test</title></head>
<body>
<div>
{{ cat|shortcodes }}
</div>
</body>
</html>
""")
f.close()
with self.app.app_context():
with self.app.test_client() as c:
init_db()
page = insert_node(name='page', value=None)
insert_route(path='/', node_id=page)
add_template_for_node('simple.html', page)
catimg = insert_node(name='acat', value=u"Àрpĺè")
insert_route(path='/cat/picture/', node_id=catimg)
text = "something [chill route /cat/picture/ ] [blah blah[chill route /dog/pic] the end"
textnode = insert_node(name='cat', value=text)
insert_node_node(node_id=page, target_node_id=textnode)
rv = c.get('/', follow_redirects=True)
assert "something Àрpĺè [blah blah<!-- 404 '/dog/pic' --> the end" in rv.data
assert "[chill route /cat/picture/ ]" not in rv.data
rv = c.get('/cat/picture/', follow_redirects=True)
assert "Àрpĺè" in rv.data
def test_delete_node_with_link(self):
"""
Delete a node also will delete from link
"""
with self.app.app_context():
init_db()
a_id = insert_node(name='a', value=None)
b_id = insert_node(name='b', value=None)
c_id = insert_node(name='c', value="c")
d_id = insert_node(name='d', value="d")
# a -> c, b -> c
# a -> d
insert_node_node(node_id=a_id, target_node_id=c_id)
insert_node_node(node_id=a_id, target_node_id=d_id)
insert_node_node(node_id=b_id, target_node_id=c_id)
result = db.query(
fetch_query_string('select_link_node_from_node.sql'),
fetchall=True,
**{'node_id': a_id})
result = [x.get('node_id', None) for x in result]
assert c_id in result
assert d_id in result
assert a_id not in result
result = db.query(
fetch_query_string('select_link_node_from_node.sql'),
fetchall=True,
**{'node_id': b_id})
result = [x.get('node_id', None) for x in result]
assert c_id in result
assert d_id not in result
assert a_id not in result
# now delete (should use the 'on delete cascade' sql bit)
trans = db.transaction()
db.query(fetch_query_string('delete_node_for_id.sql'),
**{'node_id': a_id})
trans.commit()
result = db.query(fetch_query_string('select_node_from_id.sql'),
fetchall=True,
**{'node_id': a_id})
assert len(result) == 0
result = db.query(
fetch_query_string('select_link_node_from_node.sql'),
fetchall=True,
**{'node_id': a_id})
assert len(result) == 0
result = db.query(
fetch_query_string('select_node_node_from_node_id.sql'),
fetchall=True,
**{'node_id': a_id})
assert len(result) == 0
def test_add(self):
"Add a picture to the database"
with self.app.app_context():
with self.app.test_client() as c:
init_db()
init_picture_tables()
a = insert_node(name="apicture", value=None)
# Create a.jpg in tmp dir media_folder
ajpg = open(os.path.join(self.tmp_template_dir, 'a.jpg'), 'wb')
img = Image.new("RGB", (100,100))
img.save(fp=ajpg)
add_picture_for_node(node_id=a, filepath='a.jpg')
apage = insert_node(name="apage", value=None)
insert_node_node(node_id=apage, target_node_id=a)
insert_route(path='/', node_id=apage)
rv = c.get('/', follow_redirects=True)
rv_json = json.loads(rv.data)
assert 100 == rv_json['apicture']['width']
def test_delete_node_with_link(self):
"""
Delete a node also will delete from link
"""
with self.app.app_context():
init_db()
a_id = insert_node(name='a', value=None)
b_id = insert_node(name='b', value=None)
c_id = insert_node(name='c', value="c")
d_id = insert_node(name='d', value="d")
# a -> c, b -> c
# a -> d
insert_node_node(node_id=a_id, target_node_id=c_id)
insert_node_node(node_id=a_id, target_node_id=d_id)
insert_node_node(node_id=b_id, target_node_id=c_id)
result = db.execute(text(
fetch_query_string('select_link_node_from_node.sql')),
node_id=a_id)
result = [x['node_id'] for x in result]
assert c_id in result
assert d_id in result
assert a_id not in result
result = db.execute(text(
fetch_query_string('select_link_node_from_node.sql')),
node_id=b_id)
result = [x['node_id'] for x in result]
assert c_id in result
assert d_id not in result
assert a_id not in result
# now delete (should use the 'on delete cascade' sql bit)
db.execute(text(fetch_query_string('delete_node_for_id.sql')),
node_id=a_id)
result = db.execute(text(
fetch_query_string('select_node_from_id.sql')),
node_id=a_id).fetchall()
assert len(result) == 0
result = db.execute(text(
fetch_query_string('select_link_node_from_node.sql')),
node_id=a_id).fetchall()
assert len(result) == 0
result = db.execute(text(
fetch_query_string('select_node_node_from_node_id.sql')),
node_id=a_id).fetchall()
assert len(result) == 0
def test_delete_node_with_link(self):
"""
Delete a node also will delete from link
"""
with self.app.app_context():
init_db()
a_id = insert_node(name='a', value=None)
b_id = insert_node(name='b', value=None)
c_id = insert_node(name='c', value="c")
d_id = insert_node(name='d', value="d")
# a -> c, b -> c
# a -> d
insert_node_node(node_id=a_id, target_node_id=c_id)
insert_node_node(node_id=a_id, target_node_id=d_id)
insert_node_node(node_id=b_id, target_node_id=c_id)
c = db.cursor()
result = c.execute(fetch_query_string('select_link_node_from_node.sql'), {'node_id': a_id}).fetchall()
(result, col_names) = rowify(result, c.description)
result = [x.get('node_id', None) for x in result]
assert c_id in result
assert d_id in result
assert a_id not in result
result = c.execute(fetch_query_string('select_link_node_from_node.sql'), {'node_id': b_id}).fetchall()
(result, col_names) = rowify(result, c.description)
result = [x.get('node_id', None) for x in result]
assert c_id in result
assert d_id not in result
assert a_id not in result
# now delete (should use the 'on delete cascade' sql bit)
c = db.cursor()
c.execute(fetch_query_string('delete_node_for_id.sql'), {'node_id': a_id})
db.commit()
result = c.execute(fetch_query_string('select_node_from_id.sql'), {'node_id': a_id}).fetchall()
(result, col_names) = rowify(result, c.description)
assert len(result) == 0
c = db.cursor()
result = c.execute(fetch_query_string('select_link_node_from_node.sql'), {'node_id': a_id}).fetchall()
(result, col_names) = rowify(result, c.description)
assert len(result) == 0
result = c.execute(fetch_query_string('select_node_node_from_node_id.sql'), {'node_id': a_id}).fetchall()
(result, col_names) = rowify(result, c.description)
assert len(result) == 0
def test_delete_node_with_link(self):
"""
Delete a node also will delete from link
"""
with self.app.app_context():
init_db()
a_id = insert_node(name='a', value=None)
b_id = insert_node(name='b', value=None)
c_id = insert_node(name='c', value="c")
d_id = insert_node(name='d', value="d")
# a -> c, b -> c
# a -> d
insert_node_node(node_id=a_id, target_node_id=c_id)
insert_node_node(node_id=a_id, target_node_id=d_id)
insert_node_node(node_id=b_id, target_node_id=c_id)
result = db.execute(text(fetch_query_string('select_link_node_from_node.sql')), node_id=a_id)
result = [x['node_id'] for x in result]
assert c_id in result
assert d_id in result
assert a_id not in result
result = db.execute(text(fetch_query_string('select_link_node_from_node.sql')), node_id=b_id)
result = [x['node_id'] for x in result]
assert c_id in result
assert d_id not in result
assert a_id not in result
# now delete (should use the 'on delete cascade' sql bit)
db.execute(text(fetch_query_string('delete_node_for_id.sql')), node_id=a_id)
result = db.execute(text(fetch_query_string('select_node_from_id.sql')), node_id=a_id).fetchall()
assert len(result) == 0
result = db.execute(text(fetch_query_string('select_link_node_from_node.sql')), node_id=a_id).fetchall()
assert len(result) == 0
result = db.execute(text(fetch_query_string('select_node_node_from_node_id.sql')), node_id=a_id).fetchall()
assert len(result) == 0
def test_markdown_document(self):
"""
Use 'readfile' and 'markdown' filter together.
"""
md = """
Heading
=======
Sub-heading
-----------
### Another deeper heading
Paragraphs are separated
by a blank line.
Leave 2 spaces at the end of a line to do a
line break
Text attributes *italic*, **bold**,
onospace
A [link](http://example.com).
Shopping list:
* apples
* oranges
* pears
Numbered list:
1. apples
2. oranges
3. pears
The rain---not the reign---in
Spain.
"""
html = """<h1>Heading</h1>
<h2>Sub-heading</h2>
<h3>Another deeper heading</h3>
<p>Paragraphs are separated
by a blank line.</p>
<p>Leave 2 spaces at the end of a line to do a
line break</p>
<p>Text attributes <em>italic</em>, <strong>bold</strong>,
onospace
A <a href="http://example.com">link</a>.</p>
<p>Shopping list:</p>
<ul>
<li>apples</li>
<li>oranges</li>
<li>pears</li>
</ul>
<p>Numbered list:</p>
<ol>
<li>apples</li>
<li>oranges</li>
<li>pears</li>
</ol>
<p>The rain---not the reign---in
Spain.</p>"""
f = open(os.path.join(self.tmp_template_dir, 'imasimplefile.md'), 'w')
f.write(md)
f.close()
f = open(os.path.join(self.tmp_template_dir, 'template.html'), 'w')
f.write("""
{{ simplefilename|readfile|markdown }}
""")
f.close()
with self.app.app_context():
with self.app.test_client() as c:
init_db()
a = insert_node(name='simplefilename',
value='imasimplefile.md')
apage = insert_node(name='apage', value=None)
insert_node_node(node_id=apage, target_node_id=a)
insert_route(path='/a/', node_id=apage)
add_template_for_node('template.html', apage)
rv = c.get('/a/', follow_redirects=True)
assert html in rv.data
def mode_database_functions():
"Select a function to perform from chill.database"
print globals()['mode_database_functions'].__doc__
selection = True
database_functions = [
'init_db',
'insert_node',
'insert_node_node',
'delete_node',
'select_node',
'insert_route',
'insert_query',
'add_template_for_node',
'fetch_query_string',
]
while selection:
choices = database_functions + [
'help',
]
selection = select(choices)
if selection:
print globals().get(selection).__doc__
if selection == 'init_db':
confirm = raw_input("Initialize new database y/n? [n] ")
if confirm == 'y':
init_db()
elif selection == 'insert_node':
name = raw_input("Node name: ")
value = raw_input("Node value: ")
node = insert_node(name=name, value=value or None)
print "name: %s \nid: %s" % (name, node)
elif selection == 'insert_query':
sqlfile = choose_query_file()
if sqlfile:
node = existing_node_input()
if node >= 0:
insert_query(name=sqlfile, node_id=node)
print "adding %s to node id: %s" % (sqlfile, node)
elif selection == 'insert_node_node':
print "Add parent node id"
node = existing_node_input()
print "Add target node id"
target_node = existing_node_input()
if node >= 0 and target_node >= 0:
insert_node_node(node_id=node, target_node_id=target_node)
elif selection == 'delete_node':
node = existing_node_input()
if node >= 0:
delete_node(node_id=node)
elif selection == 'select_node':
node = existing_node_input()
if node >= 0:
result = select_node(node_id=node)
print safe_dump(dict(zip(result[0].keys(), result[0].values())), default_flow_style=False)
elif selection == 'insert_route':
path = raw_input('path: ')
weight = raw_input('weight: ') or None
method = raw_input('method: ') or 'GET'
node = existing_node_input()
if node >= 0:
insert_route(path=path, node_id=node, weight=weight, method=method)
elif selection == 'add_template_for_node':
folder = current_app.config.get('THEME_TEMPLATE_FOLDER')
choices = map(os.path.basename,
glob(os.path.join(folder, '*'))
)
choices.sort()
templatefile = select(choices)
if templatefile:
node = existing_node_input()
if node >= 0:
add_template_for_node(name=templatefile, node_id=node)
print "adding %s to node id: %s" % (templatefile, node)
elif selection == 'fetch_query_string':
sqlfile = choose_query_file()
if sqlfile:
sql = fetch_query_string(sqlfile)
print sql
elif selection == 'help':
print "------"
for f in database_functions:
print "\n** %s **" % f
print globals().get(f).__doc__
print "------"
else:
pass
def test_rules(self):
f = open(os.path.join(self.tmp_template_dir, 'insert_promoattr.sql'),
'w')
f.write("""
insert into PromoAttr (node_id, title, description) values (:node_id, :title, :description);
""")
f.close()
f = open(os.path.join(self.tmp_template_dir, 'select_promoattr.sql'),
'w')
f.write("""
select * from PromoAttr where node_id = :node_id;
""")
f.close()
f = open(os.path.join(self.tmp_template_dir, 'select_promos.sql'), 'w')
f.write("""
select id as node_id, * from Node where name = 'promo' order by id limit 2 offset 13;
""")
f.close()
f = open(os.path.join(self.tmp_template_dir, 'select_mainmenu.sql'),
'w')
f.write("""
select name as link from Node where name like 'page_' order by link;
""")
f.close()
f = open(os.path.join(self.tmp_template_dir, 'select_pageattr.sql'),
'w')
f.write("""
select 'example title' as title, 'a description of the page' as description;
""")
f.close()
expected = {
"mainmenu": [{
"link": "page1"
}, {
"link": "page2"
}, {
"link": "page3"
}],
"pageattr": {
"description": "a description of the page",
"title": "example title"
},
"promos": [{
"promo": {
"description": "aaaaaaaaaaaaa",
"node_id": 20,
"title": "promo 13"
}
}, {
"promo": {
"description": "aaaaaaaaaaaaaa",
"node_id": 21,
"title": "promo 14"
}
}]
}
with self.app.app_context():
with self.app.test_client() as c:
init_db()
trans = db.transaction()
db.query("""
create table PromoAttr (
node_id integer,
abc integer,
title varchar(255),
description text
);
""")
trans.commit()
page_id = insert_node(name='page1', value=None)
print page_id
insert_route(path='/page1/', node_id=page_id)
pageattr_id = insert_node(name='pageattr', value=None)
print pageattr_id
insert_node_node(node_id=page_id, target_node_id=pageattr_id)
insert_query(name='select_pageattr.sql', node_id=pageattr_id)
mainmenu_id = insert_node(name='mainmenu', value=None)
insert_node_node(node_id=page_id, target_node_id=mainmenu_id)
insert_query(name='select_mainmenu.sql', node_id=mainmenu_id)
# Add some other pages that will be shown in menu as just links
insert_node(name='page2', value=None)
insert_node(name='page3', value=None)
promos_id = insert_node(name='promos', value=None)
insert_node_node(node_id=page_id, target_node_id=promos_id)
insert_query(name='select_promos.sql', node_id=promos_id)
for a in range(0, 100):
a_id = insert_node(name='promo', value=None)
trans = db.transaction()
db.query(
fetch_query_string('insert_promoattr.sql'), **{
'node_id': a_id,
'title': 'promo %i' % a,
'description': 'a' * a
})
trans.commit()
# wire the promo to it's attr
insert_query(name='select_promoattr.sql', node_id=a_id)
rv = c.get('/page1', follow_redirects=True)
print rv
assert 200 == rv.status_code
rv_json = json.loads(rv.data)
assert set(expected.keys()) == set(rv_json.keys())
assert set(expected['pageattr'].keys()) == set(
rv_json['pageattr'].keys())
def test_rules(self):
f = open(os.path.join(self.tmp_template_dir, 'insert_promoattr.sql'), 'w')
f.write("""
insert into PromoAttr (node_id, title, description) values (:node_id, :title, :description);
""")
f.close()
f = open(os.path.join(self.tmp_template_dir, 'select_promoattr.sql'), 'w')
f.write("""
select * from PromoAttr where node_id = :node_id;
""")
f.close()
f = open(os.path.join(self.tmp_template_dir, 'select_promos.sql'), 'w')
f.write("""
select id as node_id, * from Node where name = 'promo' order by id limit 2 offset 13;
""")
f.close()
f = open(os.path.join(self.tmp_template_dir, 'select_mainmenu.sql'), 'w')
f.write("""
select name as link from Node where name like 'page_' order by link;
""")
f.close()
f = open(os.path.join(self.tmp_template_dir, 'select_pageattr.sql'), 'w')
f.write("""
select 'example title' as title, 'a description of the page' as description;
""")
f.close()
expected = {
"mainmenu": [
{ "link": "page1" },
{ "link": "page2" },
{ "link": "page3" }
],
"pageattr": {
"description": "a description of the page",
"title": "example title"
},
"promos": [
{ "promo": {
"description": "aaaaaaaaaaaaa",
"node_id": 20,
"title": "promo 13"
}
},
{ "promo": {
"description": "aaaaaaaaaaaaaa",
"node_id": 21,
"title": "promo 14"
}
}
]
}
with self.app.app_context():
with self.app.test_client() as c:
init_db()
db.execute(text("""
create table PromoAttr (
node_id integer,
abc integer,
title varchar(255),
description text
);
"""))
page_id = insert_node(name='page1', value=None)
insert_route(path='/page1/', node_id=page_id)
pageattr_id = insert_node(name='pageattr', value=None)
insert_node_node(node_id=page_id, target_node_id=pageattr_id)
insert_query(name='select_pageattr.sql', node_id=pageattr_id)
mainmenu_id = insert_node(name='mainmenu', value=None)
insert_node_node(node_id=page_id, target_node_id=mainmenu_id)
insert_query(name='select_mainmenu.sql', node_id=mainmenu_id)
# Add some other pages that will be shown in menu as just links
insert_node(name='page2', value=None)
insert_node(name='page3', value=None)
promos_id = insert_node(name='promos', value=None)
insert_node_node(node_id=page_id, target_node_id=promos_id)
insert_query(name='select_promos.sql', node_id=promos_id)
for a in range(0,100):
a_id = insert_node(name='promo', value=None)
db.execute(text(fetch_query_string('insert_promoattr.sql')), **{'node_id':a_id, 'title':'promo %i' % a, 'description': 'a'*a})
# wire the promo to it's attr
insert_query(name='select_promoattr.sql', node_id=a_id)
rv = c.get('/page1', follow_redirects=True)
assert 200 == rv.status_code
rv_json = json.loads(rv.data)
assert set(expected.keys()) == set(rv_json.keys())
assert set(expected['pageattr'].keys()) == set(rv_json['pageattr'].keys())
def mode_collection():
"""
Manage an existing collection node.
"""
print globals()['mode_collection'].__doc__
collection_node_id = existing_node_input()
value = render_value_for_node(collection_node_id)
if not value:
return None
print "Collection length: {0}".format(len(value))
print safe_dump(value, default_flow_style=False)
item_attr_list = []
if len(value):
for key in value.items()[0][1].keys():
m = re.match(r'(.*) \((\d+)\)', key)
item_attr_list.append(m.group(1))
selection = True
while selection:
selection = select([
'View collection',
'Add item',
'Add attribute',
'Remove item',
'Remove attribute',
'Purge collection'
])
if selection == 'View collection':
print safe_dump(value, default_flow_style=False)
elif selection == 'Purge collection':
confirm = raw_input("Delete all {0} items and their {1} attributes from the collection? y/n\n".format(len(value.keys()), len(item_attr_list)))
if confirm == 'y':
delete_node(node_id=collection_node_id)
purge_collection(value.keys())
elif selection == 'Remove item':
item_node_id = existing_node_input()
if item_node_id < 0:
return
value = render_value_for_node(item_node_id)
print safe_dump(value, default_flow_style=False)
confirm = raw_input("Delete this node and it's attributes? y/n\n").format(len(value.keys()), len(item_attr_list))
if confirm == 'y':
delete_node(node_id=item_node_id)
purge_collection(value.keys())
elif selection == 'Add item':
result = select_node(node_id=collection_node_id)
collection_name = result[0].get('name')
add_item_with_attributes_to_collection(
collection_name=collection_name,
collection_node_id=collection_node_id,
item_attr_list=item_attr_list)
elif selection == 'Remove attribute':
print "Select the attribute that will be removed:"
attribute_selection = select(item_attr_list)
if attribute_selection:
confirm = raw_input("Delete attribute '{0}' from all {1} items in the collection? y/n\n".format(attribute_selection, len(value.keys())))
if confirm == 'y':
for item_key, item in value.items():
for key in item.keys():
m = re.match(r'(.*) \((\d+)\)', key)
if m.group(1) == attribute_selection:
delete_node(node_id=m.group(2))
break
elif selection == 'Add attribute':
item_attr = raw_input("Add a collection item attribute name: ")
if item_attr:
item_index = 0
for item_key, item in value.items():
item_index += 1
m = re.match(r'(.*) \((\d+)\)', item_key)
item_value = render_value_for_node(m.group(2))
print "item {0} of {1} items".format(item_index, len(value))
print safe_dump(item_value, default_flow_style=False)
new_attr_value = raw_input("Enter item attribute value for '{0}': ".format(item_attr))
# set value to none if it's an empty string
new_attr_value = new_attr_value if len(new_attr_value) else None
item_attr_node_id = insert_node(name=item_attr, value=new_attr_value)
insert_node_node(node_id=m.group(2), target_node_id=item_attr_node_id)
# Update the value after each operation
value = render_value_for_node(collection_node_id)
def test_markdown_document(self):
"""
Use 'readfile' and 'markdown' filter together.
"""
md = """
Heading
=======
Sub-heading
-----------
### Another deeper heading
Paragraphs are separated
by a blank line.
Leave 2 spaces at the end of a line to do a
line break
Text attributes *italic*, **bold**,
onospace
A [link](http://example.com).
Shopping list:
* apples
* oranges
* pears
Numbered list:
1. apples
2. oranges
3. pears
The rain---not the reign---in
Spain.
"""
html = """<h1>Heading</h1>
<h2>Sub-heading</h2>
<h3>Another deeper heading</h3>
<p>Paragraphs are separated
by a blank line.</p>
<p>Leave 2 spaces at the end of a line to do a
line break</p>
<p>Text attributes <em>italic</em>, <strong>bold</strong>,
onospace
A <a href="http://example.com">link</a>.</p>
<p>Shopping list:</p>
<ul>
<li>apples</li>
<li>oranges</li>
<li>pears</li>
</ul>
<p>Numbered list:</p>
<ol>
<li>apples</li>
<li>oranges</li>
<li>pears</li>
</ol>
<p>The rain---not the reign---in
Spain.</p>"""
f = open(os.path.join(self.tmp_template_dir, 'imasimplefile.md'), 'w')
f.write(md)
f.close()
f = open(os.path.join(self.tmp_template_dir, 'template.html'), 'w')
f.write("""
{{ simplefilename|readfile|markdown }}
""")
f.close()
with self.app.app_context():
with self.app.test_client() as c:
init_db()
a = insert_node(name='simplefilename', value='imasimplefile.md')
apage = insert_node(name='apage', value=None)
insert_node_node(node_id=apage, target_node_id=a)
insert_route(path='/a/', node_id=apage)
add_template_for_node('template.html', apage)
rv = c.get('/a/', follow_redirects=True)
assert html in rv.data