me/offline-twitter
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix linting error

Browse Source
This commit is contained in:
Alessio 2022-02-26 16:06:51 -08:00
parent 373b9ce16e
commit 857b9c5255
2 changed files with 9 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
persistence/user_queries.go
View File

@ -196,14 +196,15 @@ func (p Profile) CheckUserContentDownloadNeeded(user scraper.User) bool {
*/
func (p Profile) SetUserFollowed(user *scraper.User, is_followed bool) {
result, err := p.DB.Exec("update users set is_followed = ? where id = ?", is_followed, user.ID)
if err != nil {
panic(fmt.Sprintf("Error inserting user with handle %q: %s", user.Handle, err.Error()))
}
count, err := result.RowsAffected()
if err != nil {
panic("Unknown error: " + err.Error())
}
if count, _ := result.RowsAffected(); count != 1 {
if count != 1 {
panic(fmt.Sprintf("User with handle %q not found", user.Handle))
}
if err != nil {
panic(fmt.Sprintf("Error inserting user with handle %q: %s", user.Handle, err.Error()))
}
user.IsFollowed = is_followed
}

6
persistence/user_queries_test.go
View File

@ -225,7 +225,8 @@ func TestFollowUnfollowUser(t *testing.T) {
user := create_dummy_user()
assert.False(user.IsFollowed)
profile.SaveUser(user)
err := profile.SaveUser(user)
assert.NoError(err)
profile.SetUserFollowed(&user, true)
assert.True(user.IsFollowed)
@ -236,7 +237,8 @@ func TestFollowUnfollowUser(t *testing.T) {
assert.Equal(user.ID, user_reloaded.ID) // Verify it's the same user
assert.True(user_reloaded.IsFollowed)
profile.SaveUser(user) // should NOT un-set is_followed
err = profile.SaveUser(user) // should NOT un-set is_followed
assert.NoError(err)
user_reloaded, err = profile.GetUserByHandle(user.Handle)
require.NoError(t, err)
assert.Equal(user.ID, user_reloaded.ID) // Verify it's the same user