func TestGobEncodeDecode(t *testing.T) {
indexer.Initialize(config.Config)
c, _ := New("foo")
saved := new(bytes.Buffer)
var err error
enc := gob.NewEncoder(saved)
defer func() {
if x := recover(); x != nil {
err = fmt.Errorf("Something went wrong encoding the data store with Gob")
}
}()
err = enc.Encode(c)
if err != nil {
t.Errorf(err.Error())
}
dec := gob.NewDecoder(saved)
c2 := new(Client)
err = dec.Decode(&c2)
if err != nil {
t.Errorf(err.Error())
}
if c2.Name != c.Name {
t.Errorf("saved user doesn't seem to be equal to original: %v vs %v", c2, c)
}
}
func TestActorClient(t *testing.T) {
config.Config.UseAuth = true
indexer.Initialize(config.Config)
c, _ := client.New("fooclient")
gob.Register(c)
c.Save()
c1, err := GetReqUser("fooclient")
if err != nil {
t.Errorf(err.Error())
}
y := c1.IsSelf(c)
if y == false {
t.Errorf("self not equal to self")
}
c2, _ := client.New("foo2client")
y = c1.IsSelf(c2)
if y != false {
t.Errorf("client %s was equal to client %s, but should not have been", c1.GetName(), c2.Name)
}
u, _ := user.New("foouser")
gob.Register(u)
u.Save()
y = c1.IsSelf(u)
if y != false {
t.Errorf("client %s was equal to user %s, but should not have been", c1.GetName(), u.Username)
}
c.Delete()
c2.Delete()
u.Delete()
}
func makeSearchItems() int {
indexer.Initialize(config.Config)
/* Gotta populate the search index */
nodes := make([]*node.Node, 4)
roles := make([]*role.Role, 4)
envs := make([]*environment.ChefEnvironment, 4)
clients := make([]*client.Client, 4)
dbags := make([]*databag.DataBag, 4)
gob.Register(new(node.Node))
gob.Register(new(role.Role))
gob.Register(new(environment.ChefEnvironment))
gob.Register(new(client.Client))
gob.Register(new(databag.DataBag))
for i := 0; i < 4; i++ {
nodes[i], _ = node.New(fmt.Sprintf("node%d", i))
nodes[i].Default["baz"] = fmt.Sprintf("borb")
nodes[i].Default["blurg"] = fmt.Sprintf("b%d", i)
nodes[i].Save()
roles[i], _ = role.New(fmt.Sprintf("role%d", i))
roles[i].Save()
envs[i], _ = environment.New(fmt.Sprintf("env%d", i))
envs[i].Save()
clients[i], _ = client.New(fmt.Sprintf("client%d", i))
clients[i].Save()
dbags[i], _ = databag.New(fmt.Sprintf("databag%d", i))
dbags[i].Save()
dbi := make(map[string]interface{})
dbi["id"] = fmt.Sprintf("dbi%d", i)
dbi["foo"] = fmt.Sprintf("dbag_item_%d", i)
dbags[i].NewDBItem(dbi)
}
node1 = nodes[0]
node2 = nodes[1]
node3 = nodes[2]
node4 = nodes[3]
role1 = roles[0]
role2 = roles[1]
role3 = roles[2]
role4 = roles[3]
env1 = envs[0]
env2 = envs[1]
env3 = envs[2]
env4 = envs[3]
client1 = clients[0]
client2 = clients[1]
client3 = clients[2]
client4 = clients[3]
dbag1 = dbags[0]
dbag2 = dbags[1]
dbag3 = dbags[2]
dbag4 = dbags[3]
/* Make this function return something so the compiler's happy building
* the tests. */
return 1
}
func TestShoveyCreation(t *testing.T) {
indexer.Initialize(config.Config)
nn := new(node.Node)
ns := new(node.NodeStatus)
gob.Register(nn)
gob.Register(ns)
nodes := make([]*node.Node, 5)
nodeNames := make([]string, 5)
for i := 0; i < 5; i++ {
n, _ := node.New(fmt.Sprintf("node-shove-%d", i))
n.Save()
err := n.UpdateStatus("up")
if err != nil {
t.Errorf(err.Error())
}
n.Save()
nodes[i] = n
nodeNames[i] = n.Name
}
z := new(Shovey)
zz := new(ShoveyRun)
gob.Register(z)
gob.Register(zz)
s, err := New("/bin/ls", 300, "100%", nodeNames)
if err != nil {
t.Errorf(err.Error())
}
s2, err := Get(s.RunID)
if err != nil {
t.Errorf(err.Error())
}
if s.RunID != s2.RunID {
t.Errorf("Run IDs should have been equal, but weren't. Got %s and %s", s.RunID, s2.RunID)
}
//err = s.Cancel()
//if err != nil {
// t.Errorf(err.Error())
//}
}
func makeSearchItems() int {
indexer.Initialize(config.Config)
/* Gotta populate the search index */
nodes := make([]*node.Node, 4)
roles := make([]*role.Role, 4)
envs := make([]*environment.ChefEnvironment, 4)
clients := make([]*client.Client, 4)
dbags := make([]*databag.DataBag, 4)
gob.Register(new(node.Node))
gob.Register(new(role.Role))
gob.Register(new(environment.ChefEnvironment))
gob.Register(new(client.Client))
gob.Register(new(databag.DataBag))
// circleci is sometimes weird about the index having everything. This
// *never* comes up locally. ??? Could possibly be because the indexer
// hasn't had a chance to finish indexing?
reindexObjs := make([]indexer.Indexable, 0, 4*5)
for i := 0; i < 4; i++ {
nodes[i], _ = node.New(fmt.Sprintf("node%d", i))
nodes[i].Default["baz"] = fmt.Sprintf("borb")
nodes[i].Default["blurg"] = fmt.Sprintf("b%d", i)
nodes[i].Save()
roles[i], _ = role.New(fmt.Sprintf("role%d", i))
roles[i].Save()
envs[i], _ = environment.New(fmt.Sprintf("env%d", i))
envs[i].Save()
clients[i], _ = client.New(fmt.Sprintf("client%d", i))
clients[i].Save()
dbags[i], _ = databag.New(fmt.Sprintf("databag%d", i))
dbags[i].Save()
dbi := make(map[string]interface{})
dbi["id"] = fmt.Sprintf("dbi%d", i)
dbi["foo"] = fmt.Sprintf("dbag_item_%d", i)
dbags[i].NewDBItem(dbi)
reindexObjs = append(reindexObjs, nodes[i])
reindexObjs = append(reindexObjs, roles[i])
reindexObjs = append(reindexObjs, envs[i])
reindexObjs = append(reindexObjs, clients[i])
dbis, _ := dbags[i].AllDBItems()
for _, d := range dbis {
reindexObjs = append(reindexObjs, d)
}
}
node1 = nodes[0]
node2 = nodes[1]
node3 = nodes[2]
node4 = nodes[3]
role1 = roles[0]
role2 = roles[1]
role3 = roles[2]
role4 = roles[3]
env1 = envs[0]
env2 = envs[1]
env3 = envs[2]
env4 = envs[3]
client1 = clients[0]
client2 = clients[1]
client3 = clients[2]
client4 = clients[3]
dbag1 = dbags[0]
dbag2 = dbags[1]
dbag3 = dbags[2]
dbag4 = dbags[3]
indexer.ClearIndex()
indexer.ReIndex(reindexObjs)
/* Make this function return something so the compiler's happy building
* the tests. */
return 1
}