func TestUninstall(t *testing.T) {
fakeSB := runit.FakeServiceBuilder()
defer fakeSB.Cleanup()
serviceBuilder := &fakeSB.ServiceBuilder
testPodDir, err := ioutil.TempDir("", "testPodDir")
Assert(t).IsNil(err, "Got an unexpected error creating a temp directory")
pod := Pod{
Id: "testPod",
path: testPodDir,
ServiceBuilder: serviceBuilder,
}
manifest := getTestPodManifest(t)
manifestContent, err := manifest.Marshal()
Assert(t).IsNil(err, "couldn't get manifest bytes")
err = ioutil.WriteFile(pod.currentPodManifestPath(), manifestContent, 0744)
Assert(t).IsNil(err, "should have written current manifest")
serviceBuilderFilePath := filepath.Join(serviceBuilder.ConfigRoot, "testPod.yaml")
err = ioutil.WriteFile(serviceBuilderFilePath, []byte("stuff"), 0744)
Assert(t).IsNil(err, "Error writing fake servicebuilder file")
err = pod.Uninstall()
Assert(t).IsNil(err, "Error uninstalling pod")
_, err = os.Stat(serviceBuilderFilePath)
Assert(t).IsTrue(os.IsNotExist(err), "Expected file to not exist after uninstall")
_, err = os.Stat(pod.currentPodManifestPath())
Assert(t).IsTrue(os.IsNotExist(err), "Expected file to not exist after uninstall")
}
func TestBuildRunitServices(t *testing.T) {
fakeSB := runit.FakeServiceBuilder()
defer fakeSB.Cleanup()
serviceBuilder := &fakeSB.ServiceBuilder
pod := Pod{
P2Exec: "/usr/bin/p2-exec",
Id: "testPod",
home: "/data/pods/testPod",
ServiceBuilder: serviceBuilder,
LogExec: runit.DefaultLogExec(),
FinishExec: DefaultFinishExec,
}
hl, sb := hoist.FakeHoistLaunchableForDir("multiple_script_test_hoist_launchable")
defer hoist.CleanupFakeLaunchable(hl, sb)
hl.RunAs = "testPod"
executables, err := hl.Executables(serviceBuilder)
outFilePath := filepath.Join(serviceBuilder.ConfigRoot, "testPod.yaml")
Assert(t).IsNil(err, "Got an unexpected error when attempting to start runit services")
testManifest := manifest.NewBuilder()
testManifest.SetRestartPolicy(runit.RestartPolicyAlways)
testLaunchable := hl.If()
pod.buildRunitServices([]launch.Launchable{testLaunchable}, testManifest.GetManifest())
f, err := os.Open(outFilePath)
defer f.Close()
bytes, err := ioutil.ReadAll(f)
Assert(t).IsNil(err, "Got an unexpected error reading the servicebuilder yaml file")
expectedMap := map[string]runit.ServiceTemplate{
executables[0].Service.Name: {
Run: executables[0].Exec,
Log: runit.DefaultLogExec(),
Finish: pod.FinishExecForLaunchable(testLaunchable),
},
executables[1].Service.Name: {
Run: executables[1].Exec,
Log: runit.DefaultLogExec(),
Finish: pod.FinishExecForLaunchable(testLaunchable),
},
}
expected, err := yaml.Marshal(expectedMap)
Assert(t).IsNil(err, "Got error marshalling expected map to yaml")
Assert(t).AreEqual(string(bytes), string(expected), "Servicebuilder yaml file didn't have expected contents")
}
func TestBuildRunitServices(t *testing.T) {
fakeSB := runit.FakeServiceBuilder()
defer fakeSB.Cleanup()
serviceBuilder := &fakeSB.ServiceBuilder
pod := Pod{
Id: "testPod",
path: "/data/pods/testPod",
ServiceBuilder: serviceBuilder,
}
hl, sb := hoist.FakeHoistLaunchableForDir("multiple_script_test_hoist_launchable")
defer hoist.CleanupFakeLaunchable(hl, sb)
hl.RunAs = "testPod"
executables, err := hl.Executables(serviceBuilder)
outFilePath := filepath.Join(serviceBuilder.ConfigRoot, "testPod.yaml")
Assert(t).IsNil(err, "Got an unexpected error when attempting to start runit services")
pod.buildRunitServices([]launch.Launchable{hl.If()}, runit.RestartPolicyAlways)
f, err := os.Open(outFilePath)
defer f.Close()
bytes, err := ioutil.ReadAll(f)
Assert(t).IsNil(err, "Got an unexpected error reading the servicebuilder yaml file")
expectedMap := map[string]interface{}{
executables[0].Service.Name: map[string]interface{}{
"run": executables[0].Exec,
},
executables[1].Service.Name: map[string]interface{}{
"run": executables[1].Exec,
},
}
expected, err := yaml.Marshal(expectedMap)
Assert(t).IsNil(err, "Got error marshalling expected map to yaml")
Assert(t).AreEqual(string(bytes), string(expected), "Servicebuilder yaml file didn't have expected contents")
}