def test_get_shortlink__hierarchical_link_does_not_exist__subpath_exists( self): self.assertEqual((None, None), helpers.get_shortlink('1.com', 'go', 'trotto')) self.assertEqual((None, None), helpers.get_shortlink('1.com', 'go', 'trotto/fr'))
def test_get_shortlink__same_shortlink_different_companies__alternate_namespace( self): self.assertEqual((ShortLink.get_by_id(117), 'https://1.drive.com/eng'), helpers.get_shortlink('1.com', 'eng', 'drive')) self.assertEqual((ShortLink.get_by_id(118), 'https://2.drive.com/eng'), helpers.get_shortlink('2.com', 'eng', 'drive'))
def get_go_link(path): requested_at = time.time() provided_shortpath = parse.unquote(path.strip('/')) shortpath_parts = provided_shortpath.split('/', 1) shortpath = '/'.join([shortpath_parts[0].lower()] + shortpath_parts[1:]) if not getattr(current_user, 'email', None): if request.args.get('s') == 'crx' and request.args.get('sc'): # see: go/484356182846856 return force_to_original_url() return redirect('/_/auth/login?%s' % parse.urlencode({'redirect_to': request.full_path})) matching_shortlink, destination = get_shortlink(current_user.organization, shortpath) if matching_shortlink: queue_event(requested_at, matching_shortlink.get_id(), destination, request.args.get('s') or 'other') return redirect(str(destination)) elif request.args.get('s') == 'crx' and request.args.get('sc'): return force_to_original_url() else: return redirect( '%s/?%s' % ('http://localhost:5007' if request.host.startswith('localhost') else '', parse.urlencode({'sp': shortpath})))
def test_get_shortlink__go_link_with_placeholder(self): self.assertEqual((models.ShortLink.get_by_id(116), 'http://looker.com/7/search/users'), helpers.get_shortlink('1.com', 'looker/7/users'))
def test_get_shortlink__simple_go_link(self): self.assertEqual((models.ShortLink.get_by_id(14), 'http://drive4.com'), helpers.get_shortlink('1.com', 'drive'))
def test_get_shortlink__link_does_not_exist(self): self.assertEqual((None, None), helpers.get_shortlink('1.com', 'slack'))
def test_get_shortlink__hierarchical_link_does_not_exist(self): self.assertEqual((None, None), helpers.get_shortlink('1.com', 'go', 'trotto/jots'))
def test_get_shortlink__hierarchical_link_exists_in_other_org(self): self.assertEqual((None, None), helpers.get_shortlink('2.com', 'go', 'trotto/docs'))
def test_get_shortlink__hierarchical_link_exists(self): self.assertEqual((ShortLink.get_by_id(16), 'http://www.trot.to/docs'), helpers.get_shortlink('1.com', 'go', 'trotto/docs')) self.assertEqual((ShortLink.get_by_id(17), 'http://fr.trot.to'), helpers.get_shortlink('1.com', 'go', 'trotto/fr/t'))