Пример #1
0
func ExampleSetupOAuth() {
	SetupOAuth("userkey", new(urlp), new(st), new(userp))

	s := new(Stack)
	session := Sessions("123123123213", "mangoauth", &CookieOptions{Path: "/", MaxAge: 3600 * 24 * 7})
	s.Middleware(session)

	m := http.NewServeMux()
	m.HandleFunc("/login", s.HandlerFunc(OAuthLogin))
	m.HandleFunc("/logout", s.HandlerFunc(Logout))
	m.HandleFunc("/oauthready", s.HandlerFunc(OAuthReady))

	restrictedStack := new(Stack)
	restrictedStack.Middleware(session, MustLoggedIn)

	profile := func(env Env) (status Status, headers Headers, body Body) {
		return 200, Headers{}, Body("Hi, this is my profile")
	}

	m.HandleFunc("/profile", restrictedStack.HandlerFunc(profile))

	fakelogin := func(env Env) (status Status, headers Headers, body Body) {
		s := env.Session()
		s["userkey"] = "user1"
		return 200, Headers{}, Body("")
	}

	m.HandleFunc("/fakelogin", s.HandlerFunc(fakelogin))

	ts := httptest.NewServer(m)
	defer ts.Close()

	is := integrationtest.NewSession()

	is.Get(ts.URL + "/login")
	for k, _ := range storageImpl {
		fmt.Printf("have TemporaryCredential stored: %+v\n", strings.Contains(k, "TemporaryCredential"))
	}

	// logged in can access profile
	is.Get(ts.URL + "/fakelogin")
	storageImpl["User:user1"] = &user{"user1"}
	r, _ := is.Get(ts.URL + "/profile")
	b, _ := ioutil.ReadAll(r.Body)
	if strings.Contains(string(b), "Hi, this is my profile") {
		fmt.Println("can access profile when logged in")
	}

	// didn't log in can not access profile
	delete(storageImpl, "User:user1")
	r, _ = is.Get(ts.URL + "/profile")
	b, _ = ioutil.ReadAll(r.Body)
	if !strings.Contains(string(b), "Hi, this is my profile") {
		fmt.Println("yes, can not access profile if not logged in")
	}
	//Output:
	//have TemporaryCredential stored: true
	//can access profile when logged in
	//yes, can not access profile if not logged in
}
Пример #2
0
func fetch(num_iid string, dimension bool) (imgs []*Image, err error) {
	s := integrationtest.NewSession()
	// r, err := http.Get(fmt.Sprintf("http://item.taobao.com/item.htm?id=%s", num_iid))
	r1, err := s.Get(fmt.Sprintf("http://item.taobao.com/item.htm?id=%s", num_iid))
	defer close(r1)

	if err != nil {
		if Verbose {
			log.Printf("taobaoimg: get taobao item page error: %s\n", err)
		}
		return
	}

	buf := bytes.NewBuffer([]byte{})
	matchIndex := descUrlRegexp.FindReaderIndex(bufio.NewReader(io.TeeReader(r1.Body, buf)))
	if len(matchIndex) == 0 {
		return
	}
	descURL := buf.String()[matchIndex[0]:matchIndex[1]]
	r2, err := http.Get(descURL)
	if err != nil {
		if Verbose {
			log.Printf("taobaoimg: get taobao desc url %s error: %s\n", descURL, err)
		}
		return
	}
	b, _ := ioutil.ReadAll(r2.Body)
	defer close(r2)
	matches := imgTagRegexp.FindAllStringSubmatch(string(b), -1)
	for _, match := range matches {
		var img *Image
		if dimension {
			img, err = DecodeImage(match[1])
		} else {
			img = &Image{
				URL: match[1],
			}
		}
		if img != nil {
			imgs = append(imgs, img)
		}
	}

	return
}
Пример #3
0
func TestThumbnailLoader(t *testing.T) {

	mgodb.Setup("localhost", "tenpu_test")
	mgodb.DropCollections(collectionName, thumbnailsCollectionName)

	m := &maker{}
	_, meta, _, _ := m.MakeForUpload(nil)

	http.HandleFunc("/thumbpostupload", tenpu.MakeUploader(m))
	http.HandleFunc("/thumbload", thumbnails.MakeLoader(&thumbnails.Configuration{
		Maker: m,
		ThumbnailStorageMaker: &ThumbnailStorageMaker{},
		ThumbnailSpecs: []*thumbnails.ThumbnailSpec{
			{Name: "icon", Width: 100},
		},
	}))

	ts := httptest.NewServer(http.DefaultServeMux)
	defer ts.Close()

	var err error

	s := integrationtest.NewSession()
	res := integrationtest.Must(s.PostMultipart(ts.URL+"/thumbpostupload", func(w *multipart.Writer) {
		w.WriteField("OwnerId", "my12345")
		p1, err := w.CreateFormFile("t", "t.jpg")
		if err != nil {
			panic(err)
		}
		tf, err1 := os.Open("t.jpg")
		if err1 != nil {
			panic(err1)
		}
		defer tf.Close()

		io.Copy(p1, tf)
	}))

	b, _ := ioutil.ReadAll(res.Body)
	strb := string(b)
	if !strings.Contains(strb, "my12345") {
		t.Errorf("%+v", strb)
	}

	atts := meta.Attachments("my12345")
	if len(atts) != 1 {
		t.Errorf("%+v", atts)
	}

	res, err = http.Get(ts.URL + fmt.Sprintf("/thumbload?id=%s&thumb=icon", atts[0].Id))
	if err != nil {
		panic(err)
	}

	f, err2 := os.OpenFile("thumbGenerated.jpg", os.O_CREATE|os.O_RDWR, 0666)
	if err2 != nil {
		panic(err2)
	}

	defer f.Close()

	io.Copy(f, res.Body)

	http.Get(ts.URL + fmt.Sprintf("/thumbload?id=%s&thumb=icon", atts[0].Id))

	var thumbs []thumbnails.Thumbnail
	mgodb.FindAll(thumbnailsCollectionName, bson.M{}, &thumbs)
	if len(thumbs) != 1 {
		t.Errorf("%+v", thumbs)
	}
}