Support different runtime environments out-of-the-box.

.github/workflows/docker-repo.yml

@@ -0,0 +1,49 @@
+name: Get ESDB Docker repo
+on:
+  workflow_call:
+    inputs:
+      runtime_env:
+        description: The runtime environment we want to run like release or staging
+        type: string
+        default: release
+
+    outputs:
+      docker_repo:
+        description: ESDB docker repository
+        value: ${{ jobs.provide_docker.outputs.docker_repo }}
+
+      docker_container:
+        description: ESDB docker container
+        value: ${{ jobs.provide_docker.outputs.docker_container }}
+
+jobs:
+  provide_docker:
+    runs-on: ubuntu-latest
+    outputs:
+      docker_repo: ${{ steps.set_docker.outputs.docker_repo }}
+      docker_container: ${{ steps.set_docker.outputs.docker_container }}
+    steps:
+      - name: Set ESDB docker repo
+        id: set_docker
+        run: |
+          case ${{ inputs.runtime_env }} in
+            "release")
+              echo "docker_repo=eventstore-ce" >> $GITHUB_OUTPUT
+              echo "docker_container=eventstoredb-ce" >> $GITHUB_OUTPUT
+            ;;
+
+            "staging")
+              echo "docker_repo=eventstore-staging-ce" >> $GITHUB_OUTPUT
+              echo "docker_container=eventstoredb-oss" >> $GITHUB_OUTPUT
+            ;;
+
+            "enterprise")
+              echo "docker_repo=eventstore-ee" >> $GITHUB_OUTPUT
+              echo "docker_container=eventstoredb-commercial" >> $GITHUB_OUTPUT
+            ;;
+
+            *)
+              echo "docker_repo=eventstore-ce" >> $GITHUB_OUTPUT
+              echo "docker_container=eventstoredb-ce" >> $GITHUB_OUTPUT
+            ;;
+          esac

.github/workflows/pull-request.yml

@@ -27,7 +27,9 @@ jobs:
         run: make misc
 
   tests:
-    needs: build
+    needs:
+      - go-version
+      - build
     name: Tests
 
     strategy:

.github/workflows/test-dispatch.yml

@@ -3,8 +3,13 @@ name: "Dispatch"
 on:
   workflow_dispatch:
     inputs:
+      runtime_env:
+        description: The runtime environment. release, staging or enterprise
+        type: string
+        default: release
+
       version:
-        description: "Version tag"
+        description: Docker version tag
         required: true
         type: string
 
@@ -18,4 +23,5 @@ jobs:
     uses: ./.github/workflows/tests.yml
     with:
       esdb_version: ${{ inputs.version }}
+      runtime_env: ${{ inputs.runtime_env }}
       go_version: ${{ needs.go-version.outputs.go_version }}

.github/workflows/tests.yml

@@ -3,6 +3,10 @@ name: tests workflow
 on:
   workflow_call:
     inputs:
+      runtime_env:
+        type: string
+        default: release
+
       esdb_version:
         required: true
         type: string
@@ -12,13 +16,19 @@ on:
         type: string
 
 jobs:
+  provide_docker:
+    uses: ./.github/workflows/docker-repo.yml
+    with:
+      runtime_env: ${{ inputs.runtime_env }}
+
   single_node:
+    needs: provide_docker
     name: Single node
 
     strategy:
       fail-fast: false
       matrix:
-        test: [Streams, PersistentSubscriptions, Expectations, Projections]
+        test: [Streams, PersistentSubscriptions, Projections]
 
     runs-on: ubuntu-latest
     steps:
@@ -33,11 +43,14 @@ jobs:
       - name: Run Go Tests
         run: make ci CI_TARGET=Test${{ matrix.test }}
         env:
-          EVENTSTORE_DOCKER_TAG_ENV: ${{ inputs.esdb_version }}
+          ESDB_DOCKER_REPO: ${{ needs.provide_docker.outputs.docker_repo }}
+          ESDB_DOCKER_CONTAINER: ${{ needs.provide_docker.outputs.docker_container }}
+          ESDB_DOCKER_CONTAINER_VERSION: ${{ inputs.esdb_version }}
           EVENTSTORE_INSECURE: true
 
   secure:
     name: Secure
+    needs: provide_docker
 
     strategy:
       fail-fast: false
@@ -58,11 +71,14 @@ jobs:
         run: make ci CI_TARGET=Test${{ matrix.test }}
 
         env:
-          EVENTSTORE_DOCKER_TAG_ENV: ${{ inputs.esdb_version }}
+          ESDB_DOCKER_REPO: ${{ needs.provide_docker.outputs.docker_repo }}
+          ESDB_DOCKER_CONTAINER: ${{ needs.provide_docker.outputs.docker_container }}
+          ESDB_DOCKER_CONTAINER_VERSION: ${{ inputs.esdb_version }}
           EVENTSTORE_INSECURE: false
 
   cluster:
     name: Cluster
+    needs: provide_docker
 
     strategy:
       fail-fast: false
@@ -80,7 +96,9 @@ jobs:
         run: |
           docker compose -f cluster-docker-compose.yml up -d
         env:
-          EVENTSTORE_DOCKER_TAG_ENV: ${{ inputs.esdb_version }}
+          ESDB_DOCKER_REPO: ${{ needs.provide_docker.outputs.docker_repo }}
+          ESDB_DOCKER_CONTAINER: ${{ needs.provide_docker.outputs.docker_container }}
+          ESDB_DOCKER_CONTAINER_VERSION: ${{ inputs.esdb_version }}
 
       - name: Run Go Tests
         run: make ci CI_TARGET=Test${{ matrix.test }}

Makefile

@@ -35,7 +35,7 @@ endif
 
 .PHONY: singleNode
 singleNode: ## Run tests against a single node.
-	@EVENTSTORE_INSECURE=true go test -count=1 -v ./esdb -run 'TestStreams|TestPersistentSubscriptions|TestExpectations|TestProjections'
+	@EVENTSTORE_INSECURE=true go test -count=1 -v ./esdb -run 'TestStreams|TestPersistentSubscriptions|TestProjections'
 
 .PHONY: secureNode
 secureNode: ## Run tests against a secure node.

cluster-docker-compose.yml

@@ -1,5 +1,3 @@
-version: "3.5"
-
 services:
   volumes-provisioner:
     image: hasnat/volumes-provisioner
@@ -25,8 +23,8 @@ services:
     depends_on:
       - volumes-provisioner
 
-  esdb-node1:
-    image: docker.eventstore.com/eventstore-ce/eventstoredb-ce:${EVENTSTORE_DOCKER_TAG_ENV:-latest}
+  esdb-node1: &template
+    image: docker.eventstore.com/${ESDB_DOCKER_REPO:-eventstore-ce}/${ESDB_DOCKER_CONTAINER:-eventstoredb-ce}:${ESDB_DOCKER_CONTAINER_VERSION:-latest}
     env_file:
       - shared.env
     environment:
@@ -47,9 +45,7 @@ services:
       - cert-gen
 
   esdb-node2:
-    image: docker.eventstore.com/eventstore-ce/eventstoredb-ce:${EVENTSTORE_DOCKER_TAG_ENV:-latest}
-    env_file:
-      - shared.env
+    <<: *template
     environment:
       - EVENTSTORE_GOSSIP_SEED=172.30.240.11:2113,172.30.240.13:2113
       - EVENTSTORE_INT_IP=172.30.240.12
@@ -61,16 +57,9 @@ services:
     networks:
       clusternetwork:
         ipv4_address: 172.30.240.12
-    volumes:
-      - ./certs:/etc/eventstore/certs
-    restart: unless-stopped
-    depends_on:
-      - cert-gen
 
   esdb-node3:
-    image: docker.eventstore.com/eventstore-ce/eventstoredb-ce:${EVENTSTORE_DOCKER_TAG_ENV:-latest}
-    env_file:
-      - shared.env
+    <<: *template
     environment:
       - EVENTSTORE_GOSSIP_SEED=172.30.240.11:2113,172.30.240.12:2113
       - EVENTSTORE_INT_IP=172.30.240.13
@@ -82,11 +71,6 @@ services:
     networks:
       clusternetwork:
         ipv4_address: 172.30.240.13
-    volumes:
-      - ./certs:/etc/eventstore/certs
-    restart: unless-stopped
-    depends_on:
-      - cert-gen
 
 networks:
   clusternetwork:

esdb/append_test.go

@@ -24,6 +24,16 @@ func createTestEvent() esdb.EventData {
 	return event
 }
 
+func createTestEvents(count int) []esdb.EventData {
+	var events []esdb.EventData
+
+	for i := 0; i < count; i++ {
+		events = append(events, createTestEvent())
+	}
+
+	return events
+}
+
 func collectStreamEvents(stream *esdb.ReadStream) ([]*esdb.ResolvedEvent, error) {
 	events := []*esdb.ResolvedEvent{}
 

esdb/client.go

@@ -32,6 +32,11 @@ func NewClient(configuration *Configuration) (*Client, error) {
 	}, nil
 }
 
+// Config Returns client's current configuration.
+func (client *Client) Config() *Configuration {
+	return client.config
+}
+
 // Close Closes a connection and cleans all its allocated resources.
 func (client *Client) Close() error {
 	client.grpcClient.close()

esdb/client_test.go

@@ -15,16 +15,6 @@ func TestStreams(t *testing.T) {
 		defer emptyClient.Close()
 	}
 
-	populatedContainer, populatedClient := CreatePopulatedDatabase(t)
-
-	if populatedContainer != nil {
-		defer populatedContainer.Close()
-	}
-
-	if populatedClient != nil {
-		defer populatedClient.Close()
-	}
-
 	isCluster := GetEnvOrDefault("CLUSTER", "false") == "true"
 	isInsecure := GetEnvOrDefault("EVENTSTORE_INSECURE", "true") == "true"
 
@@ -33,8 +23,8 @@ func TestStreams(t *testing.T) {
 	}
 
 	AppendTests(t, emptyContainer, emptyClient)
-	ReadStreamTests(t, emptyClient, populatedClient)
-	SubscriptionTests(t, emptyClient, populatedClient)
+	ReadStreamTests(t, emptyClient)
+	SubscriptionTests(t, emptyClient)
 	DeleteTests(t, emptyClient)
 	ConnectionTests(t, emptyContainer)
 
@@ -59,18 +49,8 @@ func TestPersistentSubscriptions(t *testing.T) {
 		defer emptyClient.Close()
 	}
 
-	populatedContainer, populatedClient := CreatePopulatedDatabase(t)
-
-	if populatedContainer != nil {
-		defer populatedContainer.Close()
-	}
-
-	if populatedClient != nil {
-		defer populatedClient.Close()
-	}
-
 	PersistentSubReadTests(t, emptyClient)
-	PersistentSubTests(t, emptyClient, populatedClient)
+	PersistentSubTests(t, emptyClient)
 }
 
 func TestProjections(t *testing.T) {
@@ -87,20 +67,6 @@ func TestProjections(t *testing.T) {
 	ProjectionTests(t, emptyClient)
 }
 
-func TestExpectations(t *testing.T) {
-	populatedContainer, populatedClient := CreatePopulatedDatabase(t)
-
-	if populatedContainer != nil {
-		defer populatedContainer.Close()
-	}
-
-	if populatedClient != nil {
-		defer populatedClient.Close()
-	}
-
-	ReadAllTests(t, populatedClient)
-}
-
 func TestMisc(t *testing.T) {
 	ConnectionStringTests(t)
 	TestPositionParsing(t)