def create_test_note1_and_note2_by_user1(self):
user_data = {
"username": '******',
'password': '******'
}
user = UserModel(**user_data)
user.save()
res = self.client.get("/users")
data = json.loads(res.data)
self.assertEqual(res.status_code, 200)
self.assertEqual(data[0]["username"], user_data["username"])
notes_data = [
{
"note": 'test note 1'
},
{
"note": 'test note 2'
},
]
ids = []
for note_data in notes_data:
note = NoteModel(author_id=user.id, **note_data)
note.save()
ids.append(note.id)
return user, user_data
def test_get_note_by_id(self):
user_data = {"username": '******', "password": '******'}
user = UserModel(**user_data)
user.save()
notes_data = [
{
"text": 'Text note 1',
},
{
"text": 'Text note 2',
},
]
ids = []
for note_data in notes_data:
note = NoteModel(author_id=user.id, **note_data)
note.save()
ids.append(note.id)
headers = {
'Authorization':
'Basic ' +
b64encode(f"{user_data['username']}:{user_data['password']}".
encode('ascii')).decode('utf-8')
}
res = self.client.get('/notes', headers=headers)
self.assertEqual(res.status_code, 200)
data = json.loads(res.data)
self.assertEqual(len(data), 2)
def create_test_user1_note1_tag1_tag1(self):
user_data = {
"username": '******',
'password': '******'
}
user = UserModel(**user_data)
user.save()
res = self.client.get("/users")
data = json.loads(res.data)
self.assertEqual(res.status_code, 200)
self.assertEqual(data[0]["username"], user_data["username"])
note_data1 = {
"note": 'test note 1'
}
note = NoteModel(author_id=user.id, **note_data1)
note.save()
tags_data = [
{
"name": 'test1 tag'
},
{
"name": 'test2 tag'
}
]
for tag_data in tags_data:
tag = TagModel(author_id=user.id, **tag_data)
tag.save()
return user, note, user_data, tags_data
def post(self, **kwargs):
user = UserModel(**kwargs)
try:
user.save()
return user, 201
except:
abort(404, error=f"An error occurred while adding new user" \
"or a user with such name is already exist. " \
"You can only add a unique name")
def test_user_get_by_id(self):
user_data = {"username": '******', 'password': '******'}
user = UserModel(**user_data)
user.save()
user_id = user.id
response = self.client.get(f'/users/{user_id}')
data = json.loads(response.data)
self.assertEqual(response.status_code, 200)
self.assertEqual(data["username"], user_data["username"])
def create_test_user1(self):
user_data = {"username": '******', 'password': '******'}
user = UserModel(**user_data)
user.save()
res = self.client.get("/users")
data = json.loads(res.data)
self.assertEqual(res.status_code, 200)
self.assertEqual(data[0]["username"], user_data["username"])
return user_data
def post(cls):
data = cls.parser.parse_args()
#Check if the username already exists
#if True, return an error message
#else, create a new user object, save it and return a success message
if UserModel.get_by_username(data['username']):
return {"message": "The username already exists"}, 409
if UserModel.get_by_email(data['email']):
return {"message": "The email already exists"}, 409
user = UserModel(data['username'], data['email'], data['password'])
user.save()
return {"message": "User created successfully"}, 201
def create_test_tag1_by_user1(self):
user_data1 = {"username": '******', 'password': '******'}
user1 = UserModel(**user_data1)
user1.save()
res = self.client.get("/users")
data = json.loads(res.data)
self.assertEqual(res.status_code, 200)
self.assertEqual(data[0]["username"], user_data1["username"])
tag_data1 = {"name": 'test tag'}
tag1 = TagModel(author_id=user1.id, **tag_data1)
tag1.save()
return tag1, user1, user_data1
def create_test_user1_and_user2(db_create):
users_data = [
{
"username": '******',
'password': '******'
},
{
"username": '******',
'password': '******'
},
]
users_lst = []
for user_data in users_data:
user = UserModel(**user_data)
user.save()
users_lst.append(user)
return users_lst
def create_test_tag1_and_tag2_by_user1(self):
user_data1 = {"username": '******', 'password': '******'}
user1 = UserModel(**user_data1)
user1.save()
res = self.client.get("/users")
data = json.loads(res.data)
self.assertEqual(res.status_code, 200)
self.assertEqual(data[0]["username"], user_data1["username"])
tags_data = [{"name": 'test1 tag'}, {"name": 'test2 tag'}]
ids = []
for tag_data in tags_data:
tag = TagModel(author_id=user1.id, **tag_data)
tag.save()
ids.append(tag.id)
return user1, user_data1, tags_data
def create_test_user1_note1_by_user1(self):
user_data1 = {
"username": '******',
'password': '******'
}
user1 = UserModel(**user_data1)
user1.save()
res = self.client.get("/users")
data = json.loads(res.data)
self.assertEqual(res.status_code, 200)
self.assertEqual(data[0]["username"], user_data1["username"])
note_data1 = {
"note": 'test note 1'
}
note1 = NoteModel(author_id=user1.id, **note_data1)
note1.save()
return note1, user1, user_data1
def test_users_get(self):
users_data = [
{
"username": '******',
'password': '******'
},
{
"username": '******',
'password': '******'
},
]
for user_data in users_data:
user = UserModel(**user_data)
user.save()
res = self.client.get('/users')
data = json.loads(res.data)
self.assertEqual(res.status_code, 200)
print(data)
self.assertEqual(data[0]["username"], users_data[0]["username"])
self.assertEqual(data[1]["username"], users_data[1]["username"])
def create_test_user2_note2_by_user2(self):
user_data2 = {
"username": '******',
'password': '******'
}
user2 = UserModel(**user_data2)
user2.save()
res = self.client.get("/users")
data = json.loads(res.data)
self.assertEqual(res.status_code, 200)
self.assertEqual(data[1]["username"], user_data2["username"])
note_data = {
"note": 'test note 2'
}
note = NoteModel(author_id=user2.id, **note_data)
note.save()
return note, user2, user_data2
def create_test_user1_and_user2(self):
users_data = [
{
"username": '******',
'password': '******'
},
{
"username": '******',
'password': '******'
},
]
ids_lst = []
for user_data in users_data:
user = UserModel(**user_data)
user.save()
user_id = user.id
ids_lst.append(user_id)
res = self.client.get('/users')
data = json.loads(res.data)
self.assertEqual(res.status_code, 200)
self.assertEqual(data[0]["username"], users_data[0]["username"])
self.assertEqual(data[1]["username"], users_data[1]["username"])
return ids_lst
def create_test_user1(db_create):
user1_data = {"username": '******', 'password': '******'}
user1 = UserModel(**user1_data)
user1.save()
return user1, user1_data
def create_test_user2(db_create):
user2_data = {"username": '******', 'password': '******'}
user2 = UserModel(**user2_data)
user2.save()
return user2, user2_data
def create_user():
new_user = UserModel('lazarus', '*****@*****.**', 'xbt3ybot9')
new_user.save()
