Beispiel #1
0
    def testClearUserData(self):
        user = users.User("*****@*****.**")
        first_name = "Master"
        second_name = "Roshi"
        ghost_name = "Babadook"

        ghost = Ghost(
            gmail=user,
            first_name=first_name,
            second_name=second_name,
            ghost_name=ghost_name)
        ghost.put()

        stored_ghost = Ghost.query().get()

        assert stored_ghost.gmail == user
        assert stored_ghost.first_name == first_name
        assert stored_ghost.second_name == second_name
        assert stored_ghost.ghost_name == ghost_name

        GhostrEngine.clear_user_data(stored_ghost)

        updated_stored_ghost = Ghost.query().get()

        assert updated_stored_ghost.gmail is None
        assert updated_stored_ghost.first_name is None
        assert updated_stored_ghost.second_name is None
Beispiel #2
0
    def post(self):
        user = users.get_current_user()
        previous_ghost = GhostrEngine.get_from_user(user)

        if previous_ghost:
            GhostrEngine.clear_user_data(previous_ghost)
        else:
            urlfetch.fetch(
                url='https://us-central1-ghostr.cloudfunctions.net/test-sms ',
                method=urlfetch.POST)

        ghost = GhostrEngine.get_from_ghostname(self.request.get('ghost_name'))
        ghost.first_name = self.request.get('first_name')
        ghost.second_name = self.request.get('second_name')
        ghost.gmail = user
        ghost.put()

        self.redirect('/?')