110 lines
2.9 KiB
Python
110 lines
2.9 KiB
Python
import os.path
|
|
import tempfile
|
|
from dataclasses import dataclass
|
|
from unittest import mock
|
|
|
|
import dataset
|
|
import pytest
|
|
|
|
from butterrobot import db
|
|
|
|
|
|
@dataclass
|
|
class DummyItem:
|
|
id: int
|
|
foo: str
|
|
|
|
|
|
class DummyQuery(db.Query):
|
|
tablename = "dummy"
|
|
obj = DummyItem
|
|
|
|
|
|
class MockDatabase:
|
|
def __init__(self):
|
|
self.temp_dir = tempfile.TemporaryDirectory()
|
|
|
|
def __enter__(self):
|
|
db_path = os.path.join(self.temp_dir.name, "db.sqlite")
|
|
db.db = dataset.connect(f"sqlite:///{db_path}")
|
|
|
|
def __exit__(self, exc_type, exc_val, exc_tb):
|
|
self.temp_dir.cleanup()
|
|
|
|
|
|
def test_query_create_ok():
|
|
with MockDatabase():
|
|
assert DummyQuery.create(foo="bar")
|
|
|
|
|
|
def test_query_delete_ok():
|
|
with MockDatabase():
|
|
item_id = DummyQuery.create(foo="bar")
|
|
assert DummyQuery.delete(item_id)
|
|
|
|
|
|
def test_query_exists_by_id_ok():
|
|
with MockDatabase():
|
|
assert not DummyQuery.exists(id=1)
|
|
item_id = DummyQuery.create(foo="bar")
|
|
assert DummyQuery.exists(id=item_id)
|
|
|
|
|
|
def test_query_exists_by_attribute_ok():
|
|
with MockDatabase():
|
|
assert not DummyQuery.exists(id=1)
|
|
item_id = DummyQuery.create(foo="bar")
|
|
assert DummyQuery.exists(foo="bar")
|
|
|
|
|
|
def test_query_get_ok():
|
|
with MockDatabase():
|
|
item_id = DummyQuery.create(foo="bar")
|
|
item = DummyQuery.get(id=item_id)
|
|
assert item.id
|
|
|
|
|
|
def test_query_all_ok():
|
|
with MockDatabase():
|
|
assert len(list(DummyQuery.all())) == 0
|
|
[DummyQuery.create(foo="bar") for i in range(0, 3)]
|
|
assert len(list(DummyQuery.all())) == 3
|
|
|
|
|
|
def test_update_ok():
|
|
with MockDatabase():
|
|
expected = "bar2"
|
|
item_id = DummyQuery.create(foo="bar")
|
|
assert DummyQuery.update(item_id, foo=expected)
|
|
item = DummyQuery.get(id=item_id)
|
|
assert item.foo == expected
|
|
|
|
|
|
def test_create_user_sets_password_ok():
|
|
password = "password"
|
|
with MockDatabase():
|
|
user_id = db.UserQuery.create(username="foo", password=password)
|
|
user = db.UserQuery.get(id=user_id)
|
|
assert user.password == db.UserQuery._hash_password(password)
|
|
|
|
|
|
def test_user_check_credentials_ok():
|
|
with MockDatabase():
|
|
username = "foo"
|
|
password = "bar"
|
|
user_id = db.UserQuery.create(username=username, password=password)
|
|
user = db.UserQuery.get(id=user_id)
|
|
user = db.UserQuery.check_credentials(username, password)
|
|
assert isinstance(user, db.UserQuery.obj)
|
|
|
|
|
|
def test_user_check_credentials_ko():
|
|
with MockDatabase():
|
|
username = "foo"
|
|
password = "bar"
|
|
user_id = db.UserQuery.create(username=username, password=password)
|
|
user = db.UserQuery.get(id=user_id)
|
|
assert not db.UserQuery.check_credentials(username, "error")
|
|
assert not db.UserQuery.check_credentials("error", password)
|
|
assert not db.UserQuery.check_credentials("error", "error")
|