More snapshot ID tests

radekg · Feb 7, 2022 · 654296b · 654296b
1 parent 07dd3af
commit 654296b
Showing 1 changed file with 34 additions and 1 deletion.
diff --git a/utils/ybdbid/snapshot_id_test.go b/utils/ybdbid/snapshot_id_test.go
@@ -25,6 +25,40 @@ func TestSnapshotIDParsing(t *testing.T) {
 
 	})
 
+	t.Run("it=parses UUID formatted wrapped input and back", func(tt *testing.T) {
+
+		validYBDBID := "{dfec75ee-290e-4f3b-b965-469a0246c133}"
+		parsed, err := TryParseSnapshotIDFromString(validYBDBID)
+		assert.Nil(tt, err)
+		assert.Equal(tt, len(parsed.Bytes()), 16)
+
+		parsedBackViaBytes, err := TryParseSnapshotIDFromBytes(parsed.Bytes())
+		assert.Nil(tt, err)
+		assert.Equal(tt, parsed.String(), parsedBackViaBytes.String())
+
+		parsedBackViaString, err := TryParseSnapshotIDFromString(parsed.String())
+		assert.Nil(tt, err)
+		assert.Equal(tt, parsed.String(), parsedBackViaString.String())
+
+	})
+
+	t.Run("it=parses urn prefixed UUID formatted input and back", func(tt *testing.T) {
+
+		validYBDBID := "urn:uuid:dfec75ee-290e-4f3b-b965-469a0246c133"
+		parsed, err := TryParseSnapshotIDFromString(validYBDBID)
+		assert.Nil(tt, err)
+		assert.Equal(tt, len(parsed.Bytes()), 16)
+
+		parsedBackViaBytes, err := TryParseSnapshotIDFromBytes(parsed.Bytes())
+		assert.Nil(tt, err)
+		assert.Equal(tt, parsed.String(), parsedBackViaBytes.String())
+
+		parsedBackViaString, err := TryParseSnapshotIDFromString(parsed.String())
+		assert.Nil(tt, err)
+		assert.Equal(tt, parsed.String(), parsedBackViaString.String())
+
+	})
+
 	t.Run("it=handles non-UUID input", func(tt *testing.T) {
 		invalidYBDBID := "dfec75ee-290e-4f3b---b965-469a0246c133"
 		parsed, err := TryParseSnapshotIDFromString(invalidYBDBID)
@@ -39,4 +73,3 @@ func TestSnapshotIDParsing(t *testing.T) {
 	})
 
 }
-