"""Test Unsubmit model.""" from product_review.api import db from product_review.models.unsubmit import Unsubmit def test_unsubmit_to_dict(): """Test Unsubmit.to_dict.""" test_object = Unsubmit(review_queue_id=1, user_id="foo", reason="bar") db.session.add(test_object) db.session.commit() found_model = Unsubmit.query.filter(Unsubmit.review_queue_id == 1).first() found_model_dict = found_model.to_dict() assert found_model_dict["review_queue_id"] == 1 assert found_model_dict["user_id"] == "foo" assert found_model_dict["reason"] == "bar" assert found_model_dict["unsubmit_id"] is not None assert found_model_dict["created_datetime"] is not None Unsubmit.query.delete()