def setUp(self):
self.user = User.create(username='******')
self.followers = ['user1', 'user2', 'user3']
self.bot = MockUpBot()
for username in self.followers:
Follower.create(user=self.user, username=username)
def set_users_followed(self, users):
"""
:type users: list(namedtuple)
"""
count = 0
for user in users:
unfollow_date = future_rand_date(date=user.follow_date)
Follower.create(user=self.user, username=user.username, unfollow_date=unfollow_date)
count += 1
return count
def test_get_users_to_unfollow(self):
c = Controller(username='******')
yesterday = datetime.datetime.now() - datetime.timedelta(days=1)
Follower.create(user=c.user,
username='******',
unfollow_date=yesterday)
Follower.create(user=c.user,
username='******',
following=False,
unfollow_date=yesterday)
users = c.get_users_to_unfollow()
self.assertEqual(users.count(), 1)
def set_users_unfollowed(self, users):
"""
:type users: list(namedtuple)
"""
usernames = [user.username for user in users]
query = (Follower.update(following=False)
.where(Follower.username in usernames,
Follower.following == True, # noqa
Follower.user == self.user.id))
modified_rows = query.execute()
return modified_rows
def setUp(self):
self.user = User.create(username='******')
self.followers = []
for username in ['user1', 'user2', 'user3']:
f = Follower.create(user=self.user, username=username)
self.followers.append(f)