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) { 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) 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 { if err != nil {
panic("Unknown error: " + err.Error()) 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)) 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 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() user := create_dummy_user()
assert.False(user.IsFollowed) assert.False(user.IsFollowed)
profile.SaveUser(user) err := profile.SaveUser(user)
assert.NoError(err)
profile.SetUserFollowed(&user, true) profile.SetUserFollowed(&user, true)
assert.True(user.IsFollowed) 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.Equal(user.ID, user_reloaded.ID) // Verify it's the same user
assert.True(user_reloaded.IsFollowed) 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) user_reloaded, err = profile.GetUserByHandle(user.Handle)
require.NoError(t, err) require.NoError(t, err)
assert.Equal(user.ID, user_reloaded.ID) // Verify it's the same user assert.Equal(user.ID, user_reloaded.ID) // Verify it's the same user