fix: read Database and Schema name during Stream import (#732)

docs/resources/stream.md

@@ -55,6 +55,6 @@ resource snowflake_stream stream {
 Import is supported using the following syntax:
 
 ```shell
-# format is database name | schema name | on table name
-terraform import snowflake_stream.example 'dbName|schemaName|tableName'
+# format is database name | schema name | stream name
+terraform import snowflake_stream.example 'dbName|schemaName|streamName'
 ```

examples/resources/snowflake_stream/import.sh

@@ -1,2 +1,2 @@
-# format is database name | schema name | on table name
-terraform import snowflake_stream.example 'dbName|schemaName|tableName'
+# format is database name | schema name | stream name
+terraform import snowflake_stream.example 'dbName|schemaName|streamName'

pkg/resources/stream.go

@@ -246,6 +246,16 @@ func ReadStream(d *schema.ResourceData, meta interface{}) error {
 		return err
 	}
 
+	err = d.Set("database", stream.DatabaseName.String)
+	if err != nil {
+		return err
+	}
+
+	err = d.Set("schema", stream.SchemaName.String)
+	if err != nil {
+		return err
+	}
+
 	err = d.Set("on_table", stream.TableName.String)
 	if err != nil {
 		return err

pkg/resources/stream_acceptance_test.go

@@ -40,6 +40,11 @@ func TestAcc_Stream(t *testing.T) {
 					checkBool("snowflake_stream.test_stream", "show_initial_rows", false),
 				),
 			},
+			{
+				ResourceName:      "snowflake_stream.test_stream",
+				ImportState:       true,
+				ImportStateVerify: true,
+			},
 		},
 	})
 }

pkg/resources/stream_test.go

@@ -57,6 +57,8 @@ func TestStreamRead(t *testing.T) {
 		err := resources.ReadStream(d, db)
 		r.NoError(err)
 		r.Equal("stream_name", d.Get("name").(string))
+		r.Equal("database_name", d.Get("database").(string))
+		r.Equal("schema_name", d.Get("schema").(string))
 		r.Equal("grand comment", d.Get("comment").(string))
 
 		// Test when resource is not found, checking if state will be empty