def test_update_note(self):
with self.assertRaises(NoteNotFound):
note = NotesService.get_note(VALID_ID)
NotesService.upsert_note(VALID_ID, "Hello, world!", utils.now())
NotesService.upsert_note(VALID_ID, "goodbye", utils.now())
note = NotesService.get_note(VALID_ID)
self.assertEqual(note.contents, "goodbye")
def test_upate_read_note(self):
note = NotesService.upsert_note(VALID_ID, "Test", utils.now())
self.assertFalse(note.read)
NotesService.mark_as_read(note)
self.assertTrue(note.read)
note = NotesService.upsert_note(VALID_ID, "foobar", utils.now(1000))
self.assertFalse(note.read)
def test_view_notes_query(self):
NotesService.upsert_note(NotesService.generate_note_id(), "aaaaaa",
utils.now())
NotesService.upsert_note(NotesService.generate_note_id(), "bbbbbb",
utils.now())
with self.client.get("/notes?q=aaa") as rv:
self.assertEqual(rv.status_code, 200)
data = rv.data.decode("utf-8")
self.assertTrue("aaaaaa" in data)
self.assertTrue("bbbbbb" not in data)
def test_hide_note(self):
params_list = [
{
"page": 1,
"pageSize": 10
},
{
"page": 1
},
{
"page": "foobar"
},
{
"page": "foobar",
"pageSize": "foo"
},
]
NotesService.upsert_note(VALID_ID, "hello", utils.now())
for params in params_list:
with self.subTest(**params):
rv = self.client.post("/notes/" + VALID_ID + "/hide",
data=params)
self.assertEqual(rv.status_code, 302)
self.assertTrue(NotesService.get_note(VALID_ID).hidden)
def test_note_mark_as_read(self):
note = NotesService.upsert_note(VALID_ID, "hello", utils.now())
self.assertFalse(note.read)
with self.client.get("/notes/" + VALID_ID) as rv:
self.assertEqual(rv.status_code, 200)
self.assertTrue(note.read)
def test_pagination_hidden(self):
NotesService.upsert_note(VALID_ID, "Test", utils.now())
pagination = NotesService.paginate_notes(page=1)
self.assertEqual(len(pagination.items), 1)
NotesService.hide_note(VALID_ID)
pagination = NotesService.paginate_notes(page=1)
self.assertEqual(len(pagination.items), 0)
def test_hide_note(self):
NotesService.upsert_note(VALID_ID, "Hello, world!", utils.now())
note = NotesService.get_note(VALID_ID)
self.assertFalse(note.hidden)
NotesService.hide_note(VALID_ID)
note = NotesService.get_note(VALID_ID)
self.assertTrue(note.hidden)
def test_invalid_token(self):
expired_token = AuthService.new_token(
expiration=utils.now() - timedelta(days=10)
)
tokens = ["", "hello", EXAMPLE_TOKEN, expired_token.id]
for token in tokens:
with self.subTest(token=token):
self.assertFalse(AuthService.is_valid_token(token))
def test_view_notes(self):
for _ in range(50):
NotesService.upsert_note(NotesService.generate_note_id(), "hello",
utils.now())
urls = ["/notes", "/notes?page=1", "/notes?page=foobar"]
for url in urls:
with self.subTest(url=url):
with self.client.get(url) as rv:
self.assertEqual(rv.status_code, 200)
def test_notes_search(self):
note_1_id = NotesService.generate_note_id()
note_2_id = NotesService.generate_note_id()
note_3_id = NotesService.generate_note_id()
NotesService.upsert_note(note_1_id, "hello world", utils.now())
NotesService.upsert_note(note_2_id, "hello there", utils.now(1))
NotesService.upsert_note(note_3_id, "aaaaaaaaaa", utils.now(2))
notes = NotesService.paginate_notes(1, query="hello world")
self.assertEqual(notes.total, 1)
self.assertEqual(notes.items[0].contents, "hello world")
notes = NotesService.paginate_notes(1, query="hello")
self.assertEqual(notes.total, 2)
self.assertEqual(notes.items[1].contents, "hello world")
self.assertEqual(notes.items[0].contents, "hello there")
notes = NotesService.paginate_notes(1, query="aa")
self.assertEqual(notes.total, 1)
self.assertEqual(notes.items[0].contents, "aaaaaaaaaa")
notes = NotesService.paginate_notes(1, query="foobarfoobar")
self.assertEqual(notes.total, 0)
def test_note_is_link(self):
notes = [
("http://example.com", True),
("https://place.com", True),
("https://place.com/foobar", True),
("https://place.com/foobar#foo", True),
("https://place.com/foobar?q=hello", True),
("hello there", False),
("something\nsomething", False),
("http://hello.com\nmore stuff", False),
("", False),
]
for note, is_link in notes:
with self.subTest(note=note):
note = NotesService.upsert_note(VALID_ID, note, utils.now())
self.assertEqual(is_note_link(note), is_link)
def test_pretty_preview(self):
notes_previews = [
("hello", "hello"),
("hello ", "hello"),
("hello \nfoobar", "hello"),
(" hello", "hello"),
("helloooooooooooooooo", "hellooooooooooo"),
("\nhello", "hello"),
("\n\n\nhello", "hello"),
("hello\nfoobar", "hello"),
("hello\nfooooooooooooooooo", "hello"),
("fooooooooooooooooo\nhello", "foooooooooooooo"),
("", ""),
]
for note, preview in notes_previews:
with self.subTest(note=note, preview=preview):
note = NotesService.upsert_note(
NotesService.generate_note_id(), note, utils.now()
)
self.assertEqual(preview, note.preview(15))
def test_previews_pagination(self):
total_notes = random.randint(20, 50)
page_size = 15
for _ in range(total_notes):
NotesService.upsert_note(
NotesService.generate_note_id(), "Test", utils.now()
)
ids = set()
for i in range(1, 1 + total_notes // page_size):
pagination = NotesService.paginate_notes(page=i, page_size=page_size)
self.assertLessEqual(len(pagination.items), page_size)
self.assertGreater(len(pagination.items), 0)
page_ids_set = {note.id for note in pagination.items}
self.assertEqual(len(page_ids_set), len(pagination.items))
previous_len = len(ids)
ids |= page_ids_set
self.assertEqual(len(ids), previous_len + len(page_ids_set))
def test_view_note(self):
NotesService.upsert_note(VALID_ID, "hello", utils.now())
with self.client.get("/notes/" + VALID_ID) as rv:
self.assertEqual(rv.status_code, 200)
def test_create_note(self):
NotesService.upsert_note(VALID_ID, "Hello, world!", utils.now())
note = NotesService.get_note(VALID_ID)
self.assertEqual(note.id, VALID_ID)
def test_delete_expired_tokens(self):
AuthService.new_token(expiration=utils.now() - timedelta(days=10))
self.assertEqual(Token.query.count(), 1)
AuthService.delete_expired_tokens()
self.assertEqual(Token.query.count(), 0)
