def setUp(self):
"""
Configurations to be made available before each
individual test case inheriting from this class.
"""
self.client = Client()
self.user = UserFactory()
self.access_token = generate_token(self.user)
def test_user_can_delete_an_article(self):
from conduit.apps.core.generate_token import generate_token
user = UserFactory(username="******")
token = generate_token(user)
ArticleFactory(slug="test-article-123", author=user.profile)
response = self.query_with_token(token, delete_article_mutation.format(
slug="test-article-123",
))
response_data = response.get("data")
success_message = response_data["deleteArticle"]["message"]
self.assertEqual(success_message, "Article deleted successfully")
def mutate(self, info, **kwargs):
email = Validation.validate_email(email=kwargs.get('email'))
password = Validation.validate_password(
password=kwargs.get('password'))
user = authenticate(username=email, password=password)
error_message = 'Invalid login credentials'
success_message = 'Login successful'
if user:
token = generate_token(user)
return LoginUser(user=user, token=token, success=success_message)
return LoginUser(errors=error_message)
def mutate(self, info, **kwargs):
username = kwargs.get('username')
email = kwargs.get('email')
password = kwargs.get('password')
validate_data = Validation().validate_data_fields(
username, email, password)
user = User(**validate_data)
message = "User created {} successfully.".format(email)
with SaveContextManager(model_instance=user) as user:
user.set_password(kwargs.get('password'))
user.save()
return RegisterUser(user=user,
success_message=message,
token=generate_token(user))
def setUp(self):
super(CommentTestCase, self).setUp()
self.test_article = ArticleFactory(slug="comment-edit-test-article")
self.user = UserFactory()
CommentFactory(article=self.test_article, author=self.user.profile)
self.token = generate_token(self.user)