Enable log instrumentation via SetLogger (#36)

hashicorp · Nov 11, 2021 · 38e1640 · 38e1640
1 parent e677aa4
commit 38e1640
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/installer.go b/installer.go
@@ -26,6 +26,10 @@ func NewInstaller() *Installer {
  }
 }
 
+func (i *Installer) SetLogger(logger *log.Logger) {
+ i.logger = logger
+}
+
 func (i *Installer) Ensure(ctx context.Context, sources []src.Source) (string, error) {
  var errs *multierror.Error
 

diff --git a/installer_test.go b/installer_test.go
@@ -18,12 +18,18 @@ func TestInstaller_Ensure(t *testing.T) {
  // also works and continues working
 
  i := NewInstaller()
- _, err := i.Ensure(context.Background(), []src.Source{
+ i.SetLogger(testutil.TestLogger())
+ ctx := context.Background()
+ _, err := i.Ensure(ctx, []src.Source{
  &releases.LatestVersion{
  Product: product.Terraform,
  },
  })
  if err != nil {
  t.Fatal(err)
  }
+ err = i.Remove(ctx)
+ if err != nil {
+ t.Fatal(err)
+ }
 }