Exemplo n.º 1
0
func TestDomainsByOwner(t *testing.T) {
	c := dmgr_setup()
	defer dmgr_teardown(c, t)

	mgr := DomainManager(&MongoDomainManager{c})
	uid1, uid2 := gouuid.New(), gouuid.New()
	mgr.ClaimDomain("gotest_user1_1.org", &uid1)
	mgr.ClaimDomain("gotest_user1_2.org", &uid1)
	mgr.ClaimDomain("gotest_user2_1.org", &uid2)
	mgr.ClaimDomain("gotest_user2_2.org", &uid2)

	got, err := mgr.DomainsByOwner(&uid1)
	if err != nil {
		t.Fatalf("Could find domains: %s", err)
	}

	expected := []*Domain{
		&Domain{
			Name:    "gotest_user1_1.org",
			Owners:  []*gouuid.UUID{&uid1},
			Aliases: []*Alias{},
		},
		&Domain{
			Name:    "gotest_user1_2.org",
			Owners:  []*gouuid.UUID{&uid1},
			Aliases: []*Alias{},
		},
	}

	if !reflect.DeepEqual(got, expected) {
		t.Fatalf("Unexpected domain data. Got %#v, expected %#v", got, expected)
	}
}
Exemplo n.º 2
0
func (mum *MongoUserManager) New(authenticator, id string) (*User, error) {
	uid, apikey := gouuid.New(), gouuid.New()
	user := &User{
		UID:    &uid,
		APIKey: &apikey,
		Authenticators: map[string]string{
			authenticator: id,
		},
	}
	return user, mum.UpdateUser(user)
}
Exemplo n.º 3
0
func TestClaim(t *testing.T) {
	c := dmgr_setup()
	defer dmgr_teardown(c, t)

	mgr := DomainManager(&MongoDomainManager{c})
	uid := gouuid.New()
	err := mgr.ClaimDomain("gotest.org", &uid)
	if err != nil {
		t.Fatalf("ClaimDomain failed: %s", err)
	}

	qry := c.Find(bson.M{})
	if count, _ := qry.Count(); count != 1 {
		t.Fatalf("Unexpected number of results: %d", count)
	}

	got, expected := Domain{}, Domain{
		Name:    "gotest.org",
		Owners:  []*gouuid.UUID{&uid},
		Aliases: []*Alias{},
	}
	err = qry.One(&got)
	if err != nil {
		t.Fatalf("Could not get domain: %s", err)
	}
	if !reflect.DeepEqual(got, expected) {
		t.Fatalf("Unexpected domain data. Got %#v, expected %#v", got, expected)
	}
}
Exemplo n.º 4
0
func TestFindAlias(t *testing.T) {
	c := dmgr_setup()
	defer dmgr_teardown(c, t)

	mgr := DomainManager(&MongoDomainManager{c})
	uid := gouuid.New()
	mgr.ClaimDomain("gotest.org", &uid)

	expected := []*Alias{
		&Alias{
			RepoURL:    "repo1",
			RepoType:   "git",
			ForwardURL: "homepage1",
			Alias:      "alias1",
		},
		&Alias{
			RepoURL:    "repo2",
			RepoType:   "git",
			ForwardURL: "homepage2",
			Alias:      "alias2",
		},
	}
	mgr.SetAlias("gotest.org", expected[0], &uid)
	mgr.SetAlias("gotest.org", expected[1], &uid)

	alias, err := mgr.FindAlias("gotest.org", "alias1")
	if err != nil {
		t.Fatalf("Could not find domain: %s", err)
	}

	if !reflect.DeepEqual(alias, expected[0]) {
		t.Fatalf("Unexpected domain data. Got %#v, expected %#v", alias, expected)
	}
}
Exemplo n.º 5
0
func TestDeleteAlias(t *testing.T) {
	c := dmgr_setup()
	defer dmgr_teardown(c, t)

	mgr := DomainManager(&MongoDomainManager{c})
	uid := gouuid.New()
	mgr.ClaimDomain("gotest.org", &uid)

	expected := []*Alias{
		&Alias{
			RepoURL:    "repo1",
			RepoType:   "git",
			ForwardURL: "homepage1",
			Alias:      "alias1",
		},
		&Alias{
			RepoURL:    "repo2",
			RepoType:   "git",
			ForwardURL: "homepage2",
			Alias:      "alias2",
		},
	}
	mgr.SetAlias("gotest.org", expected[0], &uid)
	mgr.SetAlias("gotest.org", expected[1], &uid)

	mgr.DeleteAlias(expected[1].ID, &uid)
	expected = expected[0:1]

	domain := Domain{}
	c.Find(bson.M{}).One(&domain)
	if !reflect.DeepEqual(domain.Aliases, expected) {
		t.Fatalf("Unexpected domain data. Got %#v, expected %#v", domain.Aliases, expected)
	}
}
Exemplo n.º 6
0
func TestDeleteDomain(t *testing.T) {
	c := dmgr_setup()
	defer dmgr_teardown(c, t)

	mgr := DomainManager(&MongoDomainManager{c})
	uid := gouuid.New()
	mgr.ClaimDomain("gotest.org", &uid)
	mgr.ClaimDomain("gotest2.org", &uid)

	err := mgr.DeleteDomain("gotest.org", &uid)
	if err != nil {
		t.Fatalf("Could not delete domain: %s", err)
	}

	got, err := mgr.DomainsByOwner(&uid)
	if err != nil {
		t.Fatalf("Could not get remaining domains: %s", err)
	}
	expected := []*Domain{
		&Domain{
			Name:    "gotest2.org",
			Owners:  []*gouuid.UUID{&uid},
			Aliases: []*Alias{},
		},
	}
	if !reflect.DeepEqual(got, expected) {
		t.Fatalf("Unexpected domain data. Got %#v, expected %#v", got, expected)
	}
}
Exemplo n.º 7
0
func init() {
	// Register *gouuid.UUID as a type with gob
	// as gob is being used bei gorilla’s cookie session store which
	// is used by us to save the user’s uuid.
	uuid := gouuid.New()
	gob.Register(&uuid)
}
Exemplo n.º 8
0
func TestFindUser(t *testing.T) {
	c := umgr_setup()
	defer umgr_teardown(c, t)

	mgr := UserManager(&MongoUserManager{c})
	uuid := gouuid.New()
	err := c.Insert(
		&User{
			APIKey: &uuid,
			Authenticators: map[string]string{
				"gotest": TOKEN,
			},
		})
	if err != nil {
		panic(err)
	}

	user, err := mgr.FindByAuthenticator("gotest", TOKEN)
	if err != nil {
		t.Fatalf("Could not get user: %s", err)
	}
	if !reflect.DeepEqual(user.APIKey, &uuid) {
		t.Fatalf("Unexpected APIKey for user")
	}

	user2, err := mgr.FindByAPIKey(&uuid)
	if err != nil {
		t.Fatalf("Could not get user: %s", err)
	}
	if !reflect.DeepEqual(user, user2) {
		t.Fatalf("Unexpected user data")
	}
}
Exemplo n.º 9
0
func TestDeleteUnknownDomain(t *testing.T) {
	c := dmgr_setup()
	defer dmgr_teardown(c, t)

	mgr := DomainManager(&MongoDomainManager{c})
	uid := gouuid.New()
	other_uid := gouuid.New()
	mgr.ClaimDomain("gotest.org", &uid)
	mgr.ClaimDomain("gotest2.org", &uid)

	err := mgr.DeleteDomain("gotest.org", &other_uid)
	if err == nil {
		t.Fatalf("Could delete domain: %s", err)
	}

	got, err := mgr.DomainsByOwner(&uid)
	if err != nil {
		t.Fatalf("Could not get remaining domains: %s", err)
	}
	if len(got) != 2 {
		t.Fatalf("Got unexpected number of remaining domains")
	}
}
Exemplo n.º 10
0
func TestFindDomain(t *testing.T) {
	c := dmgr_setup()
	defer dmgr_teardown(c, t)

	mgr := DomainManager(&MongoDomainManager{c})
	uid, other_uid := gouuid.New(), gouuid.New()
	mgr.ClaimDomain("gotest.org", &uid)
	mgr.ClaimDomain("gotest2.org", &other_uid)

	got, err := mgr.FindDomain("gotest.org")
	if err != nil {
		t.Fatalf("Could find domain: %s", err)
	}

	expected := &Domain{
		Name:    "gotest.org",
		Owners:  []*gouuid.UUID{&uid},
		Aliases: []*Alias{},
	}
	if !reflect.DeepEqual(got, expected) {
		t.Fatalf("Unexpected domain data. Got %#v, expected %#v", got, expected)
	}
}
Exemplo n.º 11
0
func (mdm *MongoDomainManager) SetAlias(name string, alias *Alias, uid *gouuid.UUID) error {
	// Update
	if alias.ID != nil {
		mdm.Collection.EnsureIndexKey("aliases.id")
		return mdm.Collection.Update(bson.M{
			"aliases.id": alias.ID,
		}, bson.M{
			"$set": bson.M{
				"aliases.$": alias,
			},
		})
	}
	aid := gouuid.New()
	alias.ID = &aid
	mdm.Collection.EnsureIndexKey("name", "owners")
	return mdm.Collection.Update(bson.M{
		"name":   name,
		"owners": uid,
	}, bson.M{
		"$push": bson.M{
			"aliases": alias,
		},
	})
}
Exemplo n.º 12
0
func generateID() string {
	return gouuid.New().ShortString()
}
Exemplo n.º 13
0
func TestNumber(t *testing.T) {
	_ = gouuid.New()
	if Number() != 4 {
		t.Fatalf("AAAH!")
	}
}