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)