test: fixes backup unit test

googleapis · Feb 1, 2021 · 8c34fc3 · 8c34fc3
1 parent bc7fb50
commit 8c34fc3
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/google-cloud-spanner/src/test/java/com/google/cloud/spanner/DatabaseTest.java b/google-cloud-spanner/src/test/java/com/google/cloud/spanner/DatabaseTest.java
@@ -64,12 +64,13 @@ public Backup.Builder answer(InvocationOnMock invocation) {
   public void backup() {
     Timestamp expireTime = Timestamp.now();
     Database db = createDatabase();
-    db.backup(
+    Backup backup =
         dbClient
             .newBackupBuilder(BackupId.of("test-project", "test-instance", "test-backup"))
             .setExpireTime(expireTime)
-            .build());
-    verify(dbClient).createBackup("test-instance", "test-backup", "database-1", expireTime);
+            .build();
+    db.backup(backup);
+    verify(dbClient).createBackup(backup.toBuilder().setDatabase(db.getId()).build());
   }
 
   @Test