1 files changed, 34 insertions, 0 deletions
diff --git a/internal/db/bundb/basic_test.go b/internal/db/bundb/basic_test.go
index e5f7e159a..acdfb6640 100644
--- a/internal/db/bundb/basic_test.go
+++ b/internal/db/bundb/basic_test.go
@@ -64,6 +64,40 @@ func (suite *BasicTestSuite) TestGetAllNotNull() {
 	}
 }
 
+func (suite *BasicTestSuite) TestUpdateOneByPrimaryKeySetEmpty() {
+	testAccount := suite.testAccounts["local_account_1"]
+
+	// try removing the note from zork
+	err := suite.db.UpdateOneByPrimaryKey(context.Background(), "note", "", testAccount)
+	suite.NoError(err)
+
+	// get zork out of the database
+	dbAccount, err := suite.db.GetAccountByID(context.Background(), testAccount.ID)
+	suite.NoError(err)
+	suite.NotNil(dbAccount)
+
+	// note should be empty now
+	suite.Empty(dbAccount.Note)
+}
+
+func (suite *BasicTestSuite) TestUpdateOneByPrimaryKeySetValue() {
+	testAccount := suite.testAccounts["local_account_1"]
+
+	note := "this is my new note :)"
+
+	// try updating the note on zork
+	err := suite.db.UpdateOneByPrimaryKey(context.Background(), "note", note, testAccount)
+	suite.NoError(err)
+
+	// get zork out of the database
+	dbAccount, err := suite.db.GetAccountByID(context.Background(), testAccount.ID)
+	suite.NoError(err)
+	suite.NotNil(dbAccount)
+
+	// note should be set now
+	suite.Equal(note, dbAccount.Note)
+}
+
 func TestBasicTestSuite(t *testing.T) {
 	suite.Run(t, new(BasicTestSuite))
 }