def test_save_comment_sets_id():
from xss_demo.models import (
DB,
Comment,
)
comment = Comment('Just some text', 'admin', 0)
assert comment.id is None
DB.save(comment)
assert comment.id is not None
def test_save_comment_sets_id():
from xss_demo.models import (
DB,
Comment,
)
comment = Comment('Just some text', 'admin', 0)
assert comment.id is None
DB.save(comment)
assert comment.id is not None
def test_save_comment_invalid_id():
from xss_demo.models import (
DB,
Comment,
)
comment = Comment('Just some text', 'admin', 0)
comment.id = 99
with pytest.raises(ValueError):
DB.save(comment)
def test_save_post_sets_id():
from xss_demo.models import (
DB,
Post,
)
post = Post('Post 1', 'Just some text', 'admin')
assert post.id is None
DB.save(post)
assert post.id is not None
def test_save_post_sets_id():
from xss_demo.models import (
DB,
Post,
)
post = Post('Post 1', 'Just some text', 'admin')
assert post.id is None
DB.save(post)
assert post.id is not None
def test_save_post_invalid_id():
from xss_demo.models import (
DB,
Post,
)
post = Post('Post 1', 'Just some text', 'admin')
post.id = 99
with pytest.raises(ValueError):
DB.save(post)
def test_save_comment_invalid_id():
from xss_demo.models import (
DB,
Comment,
)
comment = Comment('Just some text', 'admin', 0)
comment.id = 99
with pytest.raises(ValueError):
DB.save(comment)
def test_save_post_invalid_id():
from xss_demo.models import (
DB,
Post,
)
post = Post('Post 1', 'Just some text', 'admin')
post.id = 99
with pytest.raises(ValueError):
DB.save(post)
def test_get_comment_returns_same_data():
from xss_demo.models import (
DB,
Comment,
)
comment = Comment('Just some text', 'admin', 0)
DB.save(comment)
comment2 = DB.get(Comment, comment.id)
assert comment is not comment2 # different objects
assert comment.__dict__ == comment2.__dict__
def test_get_post_returns_same_data():
from xss_demo.models import (
DB,
Post,
)
post = Post('Post 1', 'Just some text', 'admin')
DB.save(post)
post2 = DB.get(Post, post.id)
assert post is not post2 # different objects
assert post.__dict__ == post2.__dict__
def test_get_comment_returns_same_data():
from xss_demo.models import (
DB,
Comment,
)
comment = Comment('Just some text', 'admin', 0)
DB.save(comment)
comment2 = DB.get(Comment, comment.id)
assert comment is not comment2 # different objects
assert comment.__dict__ == comment2.__dict__
def test_get_post_returns_same_data():
from xss_demo.models import (
DB,
Post,
)
post = Post('Post 1', 'Just some text', 'admin')
DB.save(post)
post2 = DB.get(Post, post.id)
assert post is not post2 # different objects
assert post.__dict__ == post2.__dict__
def test_delete_post():
from xss_demo.models import (
DB,
Post,
)
post = Post('Post 1', 'Just some text', 'admin')
DB.save(post)
original_id = post.id
DB.delete(post)
assert DB._db['posts'][original_id] is None
assert post.id is None
def test_delete_comment():
from xss_demo.models import (
DB,
Comment,
)
comment = Comment('Just some text', 'admin', 0)
DB.save(comment)
original_id = comment.id
DB.delete(comment)
assert DB._db['comments'][original_id] is None
assert comment.id is None
def test_delete_post():
from xss_demo.models import (
DB,
Post,
)
post = Post('Post 1', 'Just some text', 'admin')
DB.save(post)
original_id = post.id
DB.delete(post)
assert DB._db['posts'][original_id] is None
assert post.id is None
def test_delete_comment():
from xss_demo.models import (
DB,
Comment,
)
comment = Comment('Just some text', 'admin', 0)
DB.save(comment)
original_id = comment.id
DB.delete(comment)
assert DB._db['comments'][original_id] is None
assert comment.id is None
def test_save_existing_post_writes_data():
from xss_demo.models import (
DB,
Post,
)
post = Post('Post 1', 'Just some text', 'admin')
DB.save(post)
original_id = post.id
post.content = 'Modified text'
DB.save(post)
assert post.id == original_id
assert DB.get(Post, original_id).content == 'Modified text'
def test_cant_get_deleted_comment():
from xss_demo.models import (
DB,
Comment,
)
comment = Comment('Just some text', 'admin', 0)
DB.save(comment)
original_id = comment.id
DB.delete(comment)
with pytest.raises(ValueError):
DB.get(Comment, original_id)
def test_get_all_comments():
from xss_demo.models import (
DB,
Comment,
)
comment = Comment('Just some text', 'admin', 0)
DB.save(comment)
saved_id = comment.id
all_comments = DB.get_all(Comment)
assert len(all_comments) >= 1
assert any(comment.id == saved_id for comment in all_comments)
def test_cant_get_deleted_comment():
from xss_demo.models import (
DB,
Comment,
)
comment = Comment('Just some text', 'admin', 0)
DB.save(comment)
original_id = comment.id
DB.delete(comment)
with pytest.raises(ValueError):
DB.get(Comment, original_id)
def test_get_all_comments():
from xss_demo.models import (
DB,
Comment,
)
comment = Comment('Just some text', 'admin', 0)
DB.save(comment)
saved_id = comment.id
all_comments = DB.get_all(Comment)
assert len(all_comments) >= 1
assert any(comment.id == saved_id for comment in all_comments)
def test_save_existing_comment_writes_data():
from xss_demo.models import (
DB,
Comment,
)
comment = Comment('Just some text', 'admin', 0)
DB.save(comment)
original_id = comment.id
comment.message = 'Modified text'
DB.save(comment)
assert comment.id == original_id
assert DB.get(Comment, original_id).message == 'Modified text'
def test_save_existing_comment_writes_data():
from xss_demo.models import (
DB,
Comment,
)
comment = Comment('Just some text', 'admin', 0)
DB.save(comment)
original_id = comment.id
comment.message = 'Modified text'
DB.save(comment)
assert comment.id == original_id
assert DB.get(Comment, original_id).message == 'Modified text'
def test_get_all_posts():
from xss_demo.models import (
DB,
Post,
)
post = Post('Post 1', 'Just some text', 'admin')
DB.save(post)
saved_id = post.id
all_posts = DB.get_all(Post)
assert len(all_posts) >= 1
assert any(post.id == saved_id for post in all_posts)
def test_cant_get_deleted_post():
from xss_demo.models import (
DB,
Post,
)
post = Post('Post 1', 'Just some text', 'admin')
DB.save(post)
original_id = post.id
DB.delete(post)
with pytest.raises(ValueError):
DB.get(Post, original_id)
def test_cant_get_deleted_post():
from xss_demo.models import (
DB,
Post,
)
post = Post('Post 1', 'Just some text', 'admin')
DB.save(post)
original_id = post.id
DB.delete(post)
with pytest.raises(ValueError):
DB.get(Post, original_id)
def test_get_all_posts():
from xss_demo.models import (
DB,
Post,
)
post = Post('Post 1', 'Just some text', 'admin')
DB.save(post)
saved_id = post.id
all_posts = DB.get_all(Post)
assert len(all_posts) >= 1
assert any(post.id == saved_id for post in all_posts)
def test_save_existing_post_writes_data():
from xss_demo.models import (
DB,
Post,
)
post = Post('Post 1', 'Just some text', 'admin')
DB.save(post)
original_id = post.id
post.content = 'Modified text'
DB.save(post)
assert post.id == original_id
assert DB.get(Post, original_id).content == 'Modified text'