Handle BigFilesTemproaryDir correctly

Signed-off-by: Daniel J Walsh <[email protected]>
containers · Jun 16, 2021 · 92bd63e · 92bd63e
1 parent f10e62c
commit 92bd63e
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 3 deletions.
diff --git a/libimage/download.go b/libimage/download.go
@@ -11,7 +11,7 @@ import (
 )
 
 // tmpdir returns a path to a temporary directory.
-func (r *Runtime) tmpdir() string {
+func tmpdir() string {
 	tmpdir := os.Getenv("TMPDIR")
 	if tmpdir == "" {
 		tmpdir = "/var/tmp"
@@ -25,7 +25,7 @@ func (r *Runtime) tmpdir() string {
 func (r *Runtime) downloadFromURL(source string) (string, error) {
 	fmt.Printf("Downloading from %q\n", source)
 
-	outFile, err := ioutil.TempFile(r.tmpdir(), "import")
+	outFile, err := ioutil.TempFile(r.systemContext.BigFilesTemporaryDir, "import")
 	if err != nil {
 		return "", errors.Wrap(err, "error creating file")
 	}

diff --git a/libimage/runtime.go b/libimage/runtime.go
@@ -81,8 +81,13 @@ func RuntimeFromStore(store storage.Store, options *RuntimeOptions) (*Runtime, e
 	var systemContext types.SystemContext
 	if options.SystemContext != nil {
 		systemContext = *options.SystemContext
+		if systemContext.BigFilesTemporaryDir == "" {
+			systemContext.BigFilesTemporaryDir = tmpdir()
+		}
 	} else {
-		systemContext = types.SystemContext{}
+		systemContext = types.SystemContext{
+			BigFilesTemporaryDir: tmpdir(),
+		}
 	}
 
 	setRegistriesConfPath(&systemContext)

diff --git a/libimage/runtime_test.go b/libimage/runtime_test.go
@@ -38,6 +38,7 @@ func testNewRuntime(t *testing.T) (runtime *Runtime, cleanup func()) {
 
 	runtime, err = RuntimeFromStoreOptions(&RuntimeOptions{SystemContext: systemContext}, storeOptions)
 	require.NoError(t, err)
+	require.Equal(t, runtime.systemContext.BigFilesTemporaryDir, tmpdir())
 
 	cleanup = func() {
 		runtime.Shutdown(true)