Integration tests should time out of operation fails to complete in 5

seconds (aka if datastore emulator is not running or cannot be reached).

datastore-translator/translator_integration_test.go

@@ -11,14 +11,19 @@ import (
 	"gotest.tools/assert"
 	"log"
 	"testing"
+	"time"
 )
 
 func TestIntegration(t *testing.T) {
 	ctx := context.Background()
+
 	// 1. create a new datastore client
 	client, err := datastore.NewClient(ctx, "st2-saas-prototype-dev")
 	assert.NilError(t, err)
 
+	ctx, cancel := context.WithTimeout(ctx, 5*time.Second)
+	defer cancel()
+
 	// 2. create a key that we plan to save into
 	key := datastore.NameKey("Example_DB_Model", "complex_proto_2", nil)
 
@@ -120,6 +125,9 @@ func TestEmptyProtoMessage(t *testing.T) {
 	client, err := datastore.NewClient(ctx, "st2-saas-prototype-dev")
 	assert.NilError(t, err)
 
+	ctx, cancel := context.WithTimeout(ctx, 5*time.Second)
+	defer cancel()
+
 	// 2. create a key that we plan to save into
 	key := datastore.NameKey("Example_DB_Model", "complex_proto_empty", nil)
 
@@ -149,6 +157,9 @@ func TestProtoWithNilPointer(t *testing.T) {
 	client, err := datastore.NewClient(ctx, "st2-saas-prototype-dev")
 	assert.NilError(t, err)
 
+	ctx, cancel := context.WithTimeout(ctx, 5*time.Second)
+	defer cancel()
+
 	// 2. create a key that we plan to save into
 	key := datastore.NameKey("Example_DB_Model", "complex_proto_empty", nil)