Multitenant streams api (#72)

groups/handler/handler_test.go

@@ -2,6 +2,7 @@ package handler_test
 
 import (
 	"context"
+	"os"
 	"sort"
 	"testing"
 
@@ -15,21 +16,25 @@ import (
 
 func testHandler(t *testing.T) *handler.Groups {
 	// connect to the database
-	db, err := gorm.Open(postgres.Open("postgresql://postgres@localhost:5432/groups?sslmode=disable"), &gorm.Config{})
+	addr := os.Getenv("POSTGRES_URL")
+	if len(addr) == 0 {
+		addr = "postgresql://postgres@localhost:5432/postgres?sslmode=disable"
+	}
+	db, err := gorm.Open(postgres.Open(addr), &gorm.Config{})
 	if err != nil {
 		t.Fatalf("Error connecting to database: %v", err)
 	}
 
+	// clean any data from a previous run
+	if err := db.Exec("DROP TABLE IF EXISTS groups, memberships CASCADE").Error; err != nil {
+		t.Fatalf("Error cleaning database: %v", err)
+	}
+
 	// migrate the database
 	if err := db.AutoMigrate(&handler.Group{}, &handler.Membership{}); err != nil {
 		t.Fatalf("Error migrating database: %v", err)
 	}
 
-	// clean any data from a previous run
-	if err := db.Exec("TRUNCATE TABLE groups CASCADE").Error; err != nil {
-		t.Fatalf("Error cleaning database: %v", err)
-	}
-
 	return &handler.Groups{DB: db}
 }
 func TestCreate(t *testing.T) {

groups/proto/groups.pb.go

@@ -1,7 +1,7 @@
 // Code generated by protoc-gen-go. DO NOT EDIT.
 // versions:
 // 	protoc-gen-go v1.25.0
-// 	protoc        v3.6.1
+// 	protoc        v3.15.5
 // source: proto/groups.proto
 
 package groups

groups/proto/groups.proto

@@ -1,7 +1,7 @@
 syntax = "proto3";
 
 package groups;
-option go_package = "proto;groups";
+option go_package = "./proto;groups";
 
 service Groups {
 	// Create a group
@@ -78,4 +78,4 @@ message RemoveMemberRequest {
 	string member_id = 2;
 }
 
-message RemoveMemberResponse {}
+message RemoveMemberResponse {}