def createUser(self, twitter_user, account_status, messages_per_day): user = User() user.twitter = twitter_user user.username = twitter_user user.account_status = account_status user.messages_per_day = messages_per_day user.total_points = 0 user.put() return user
def setUp(self): self.testbed = testbed.Testbed() self.testbed.activate() self.testbed.init_datastore_v3_stub() # Preparing datastore by prepopulating some data user = User() user.username = "******" user.twitter = "da_zbur" user.total_points = 0 user.put()
def testAddNewLearnListItem(self): # Preparing datastore by prepopulating some data user = User() user.username = "******" user.twitter = "ny_blin" user.total_points = 0 user.put() json_file = open("files/direct_message2.json") message_json = simplejson.load(json_file) twitter_dm = DirectMessage.NewFromJsonDict(message_json) processMessage(twitter_dm) query = LearnList.all().filter('twitter_user ='******'ny_blin') results = query.fetch(2) self.assertEqual(1, len(results)) self.assertEqual('ny_blin', results[0].twitter_user) self.assertEqual(2.5, results[0].interval_days) self.assertEqual(1.5, results[0].efactor) self.assertEqual(1, results[0].total_served) now_plus_two = datetime.date.today() +\ datetime.timedelta(days=2) self.assertEqual(now_plus_two, results[0].next_serve_date)
def post(self): twitter = Twitter.getInstance() model = model_from_request(self.request, RegisterModel) # validate data; on error, redisplay form with error messages if not model.validate(): self.view(model) return # save new user user = User() user.username = model.twitter_name user.twitter = model.twitter_name user.email = "" pwd_hash = hashlib.md5() pwd_hash.update(model.password) user.password = pwd_hash.hexdigest() user.timezone = "" user.next_run_time = 0 user.message_type = "reply" user.account_status = "enabled" user.repeat_times = 6 user.followed_by = "" user.i_follow = "" user.messages_per_day = 10 user.default_source_lang = "en" user.follow_lang_list = "en" user.total_points = 0 user.put() try: twitter.api.CreateFriendship(model.twitter_name) except: pass # put him into session set_current_user(user) # redirect to the home page self.redirect("/profile")