From b5b2718e143bb7dfc3dee8b0887dcf77a1fa86c4 Mon Sep 17 00:00:00 2001
From: Michael Schurter <mschurter@hashicorp.com>
Date: Mon, 4 Mar 2019 13:37:20 -0800
Subject: [PATCH] test: fix NewMemDB API change

---
 client/devicemanager/manager_test.go | 5 +++--
 client/state/db_test.go              | 2 +-
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/client/devicemanager/manager_test.go b/client/devicemanager/manager_test.go
index 044fb6f26..2cc78da5c 100644
--- a/client/devicemanager/manager_test.go
+++ b/client/devicemanager/manager_test.go
@@ -119,11 +119,12 @@ func baseTestConfig(t *testing.T) (
 	mc := &loader.MockCatalog{}
 
 	// Create the config
+	logger := testlog.HCLogger(t)
 	config = &Config{
-		Logger:        testlog.HCLogger(t),
+		Logger:        logger,
 		PluginConfig:  &base.AgentConfig{},
 		StatsInterval: 100 * time.Millisecond,
-		State:         state.NewMemDB(),
+		State:         state.NewMemDB(logger),
 		Updater:       updateFn,
 		Loader:        mc,
 	}
diff --git a/client/state/db_test.go b/client/state/db_test.go
index 87431d980..230c7ad86 100644
--- a/client/state/db_test.go
+++ b/client/state/db_test.go
@@ -44,7 +44,7 @@ func testDB(t *testing.T, f func(*testing.T, StateDB)) {
 	boltdb, cleanup := setupBoltStateDB(t)
 	defer cleanup()
 
-	memdb := NewMemDB()
+	memdb := NewMemDB(testlog.HCLogger(t))
 
 	impls := []StateDB{boltdb, memdb}