From bfea181a559abdc6e317ac908076429979e78be1 Mon Sep 17 00:00:00 2001
From: Kevin Herro <kherro@wisc.edu>
Date: Tue, 29 May 2018 12:48:23 -0700
Subject: [PATCH] test persist release (#99)

* test persist release

* refactor fs
---
 pkg/specs/resolver.go      | 12 ++++++------
 pkg/specs/resolver_test.go | 29 +++++++++++++++++++++++++++++
 2 files changed, 35 insertions(+), 6 deletions(-)
 create mode 100644 pkg/specs/resolver_test.go

diff --git a/pkg/specs/resolver.go b/pkg/specs/resolver.go
index 1efc01608..57be63a98 100644
--- a/pkg/specs/resolver.go
+++ b/pkg/specs/resolver.go
@@ -3,15 +3,13 @@ package specs
 import (
 	"context"
 	"fmt"
-	"io/ioutil"
-
-	"os"
 	"path/filepath"
 
 	"github.com/go-kit/kit/log"
 	"github.com/go-kit/kit/log/level"
 	"github.com/pkg/errors"
 	"github.com/replicatedcom/ship/pkg/api"
+	"github.com/replicatedcom/ship/pkg/lifecycle/render/state"
 	"github.com/replicatedcom/ship/pkg/logger"
 	"github.com/spf13/viper"
 	"gopkg.in/yaml.v2"
@@ -36,6 +34,7 @@ type Selector struct {
 type Resolver struct {
 	Logger            log.Logger
 	Client            *GraphQLClient
+	StateManager      *state.StateManager
 	StudioFile        string
 	StudioChannelName string
 	StudioChannelIcon string
@@ -50,6 +49,7 @@ func ResolverFromViper(v *viper.Viper) (*Resolver, error) {
 	return &Resolver{
 		Logger:            logger.FromViper(v),
 		Client:            graphql,
+		StateManager:      state.ManagerFromViper(v),
 		StudioFile:        v.GetString("studio-file"),
 		StudioChannelName: v.GetString("studio-channel-name"),
 		StudioChannelIcon: v.GetString("studio-channel-icon"),
@@ -98,7 +98,7 @@ func (r *Resolver) resolveStudioRelease() (*ShipRelease, error) {
 	debug := level.Debug(log.With(r.Logger, "method", "resolveStudioSpec"))
 	debug.Log("phase", "load-specs", "from", "studio-file", "file", r.StudioFile)
 
-	specYAML, err := ioutil.ReadFile(r.StudioFile)
+	specYAML, err := r.StateManager.FS.ReadFile(r.StudioFile)
 	if err != nil {
 		return nil, errors.Wrapf(err, "read specs from %s", r.StudioFile)
 	}
@@ -136,11 +136,11 @@ func (r *Resolver) resolveCloudRelease(customerID, installationID string) (*Ship
 
 // persistSpec persists last-used YAML to disk at .ship/release.yml
 func (r *Resolver) persistSpec(specYAML []byte) error {
-	if err := os.MkdirAll(filepath.Dir(ReleasePath), 0700); err != nil {
+	if err := r.StateManager.FS.MkdirAll(filepath.Dir(ReleasePath), 0700); err != nil {
 		return errors.Wrap(err, "mkdir yaml")
 	}
 
-	if err := ioutil.WriteFile(ReleasePath, specYAML, 0644); err != nil {
+	if err := r.StateManager.FS.WriteFile(ReleasePath, specYAML, 0644); err != nil {
 		return errors.Wrap(err, "write yaml file")
 	}
 	return nil
diff --git a/pkg/specs/resolver_test.go b/pkg/specs/resolver_test.go
new file mode 100644
index 000000000..38cb9025a
--- /dev/null
+++ b/pkg/specs/resolver_test.go
@@ -0,0 +1,29 @@
+package specs
+
+import (
+	"testing"
+	"github.com/go-kit/kit/log"
+	"github.com/replicatedcom/ship/pkg/lifecycle/render/state"
+	"github.com/spf13/afero"
+	"github.com/stretchr/testify/require"
+	"reflect"
+)
+
+func TestPersistSpec(t *testing.T) {
+
+	r := &Resolver{
+		StateManager: &state.StateManager{
+			Logger: log.NewNopLogger(),
+			FS:     afero.Afero{Fs: afero.NewMemMapFs()},
+		},
+	}
+
+	req := require.New(t)
+
+	desiredSpec := []byte("my cool spec")
+	err := r.persistSpec(desiredSpec)
+	req.NoError(err)
+
+	persistedSpec, err := r.StateManager.FS.ReadFile(".ship/release.yml")
+	req.True(reflect.DeepEqual(desiredSpec, persistedSpec))
+}