func (s *PlatformSuite) TestPlatformRemoveAlwaysRemoveFromDB(c *check.C) {
provisioner := provisiontest.ExtensibleFakeProvisioner{
FakeProvisioner: provisiontest.NewFakeProvisioner(),
}
Provisioner = &provisioner
defer func() {
Provisioner = s.provisioner
}()
conn, err := db.Conn()
c.Assert(err, check.IsNil)
defer conn.Close()
err = PlatformRemove("platform_dont_exists")
c.Assert(err, check.NotNil)
name := "test_platform_update"
args := make(map[string]string)
args["dockerfile"] = "http://localhost/Dockerfile"
err = PlatformAdd(provision.PlatformOptions{Name: name})
c.Assert(err, check.IsNil)
provisioner.PlatformRemove(name)
defer conn.Platforms().Remove(bson.M{"_id": name})
err = PlatformRemove(name)
c.Assert(err, check.IsNil)
count, err := conn.Platforms().Find(bson.M{"_id": name}).Count()
c.Assert(err, check.IsNil)
c.Assert(count, check.Equals, 0)
}
func (s *PlatformSuite) TestPlatformAddDuplicate(c *check.C) {
provisioner := provisiontest.ExtensibleFakeProvisioner{
FakeProvisioner: provisiontest.NewFakeProvisioner(),
}
Provisioner = &provisioner
defer func() {
Provisioner = s.provisioner
}()
conn, err := db.Conn()
c.Assert(err, check.IsNil)
defer conn.Close()
name := "test_platform_add"
args := make(map[string]string)
args["dockerfile"] = "http://localhost/Dockerfile"
err = PlatformAdd(provision.PlatformOptions{Name: name, Args: args})
defer conn.Platforms().Remove(bson.M{"_id": name})
c.Assert(err, check.IsNil)
provisioner.PlatformRemove(name)
err = PlatformAdd(provision.PlatformOptions{Name: name, Args: args})
c.Assert(err, check.Equals, DuplicatePlatformError)
}