Merged in feature/splittests (pull request #140)

Split Tests * split
2025-05-12 20:54:16 +00:00
parent f5061f53db
commit 9dfd867906
6 changed files with 614 additions and 215 deletions
@@ -13,153 +13,334 @@ import (
 )

 func TestDocument(t *testing.T) {
-	t.Parallel()
 	if testing.Short() {
 		t.SkipNow()
 	}
-	ctx := context.Background()
+	t.Run("List docs for a client", func(t *testing.T) {
+		t.Parallel()
+		ctx := context.Background()

-	cfg := &serviceconfig.BaseConfig{}
-	net := test.GetNetwork(t)
-	test.CreateDB(t, cfg, net, &test.CreateDatabaseConfig{})
+		cfg := &serviceconfig.BaseConfig{}
+		net := test.GetNetwork(t)
+		test.CreateDB(t, cfg, net, &test.CreateDatabaseConfig{})

-	queries := cfg.GetDBQueries()
+		queries := cfg.GetDBQueries()

-	clientId := "EXAMPLE"
-	err := queries.CreateClient(ctx, &repository.CreateClientParams{
-		Name:     "example_client",
-		Clientid: clientId,
+		clientId := "EXAMPLE"
+		err := queries.CreateClient(ctx, &repository.CreateClientParams{
+			Name:     "example_client",
+			Clientid: clientId,
+		})
+		require.NoError(t, err)
+
+		docs, err := queries.ListDocumentsByClient(ctx, clientId)
+		require.NoError(t, err)
+		assert.Len(t, docs, 0)
 	})
-	require.NoError(t, err)
+	t.Run("Create document", func(t *testing.T) {
+		t.Parallel()
+		ctx := context.Background()

-	docs, err := queries.ListDocumentsByClient(ctx, clientId)
-	require.NoError(t, err)
-	assert.Len(t, docs, 0)
+		cfg := &serviceconfig.BaseConfig{}
+		net := test.GetNetwork(t)
+		test.CreateDB(t, cfg, net, &test.CreateDatabaseConfig{})

-	hash := "example_hash"
-	id, err := queries.CreateDocument(ctx, &repository.CreateDocumentParams{
-		Clientid: clientId,
-		Hash:     hash,
+		queries := cfg.GetDBQueries()
+
+		clientId := "EXAMPLE"
+		err := queries.CreateClient(ctx, &repository.CreateClientParams{
+			Name:     "example_client",
+			Clientid: clientId,
+		})
+		require.NoError(t, err)
+
+		hash := "example_hash"
+		id, err := queries.CreateDocument(ctx, &repository.CreateDocumentParams{
+			Clientid: clientId,
+			Hash:     hash,
+		})
+		require.NoError(t, err)
+		assert.NotEmpty(t, id)
+
+		docs, err := queries.ListDocumentsByClient(ctx, clientId)
+		require.NoError(t, err)
+		assert.Len(t, docs, 1)
+		assert.Equal(t, hash, docs[0].Hash)
 	})
-	require.NoError(t, err)
-	assert.NotEmpty(t, id)

-	bucketone := "bucketone"
-	keyone := "keyone"
-	err = queries.AddDocumentEntry(ctx, &repository.AddDocumentEntryParams{
-		Documentid: id,
-		Bucket:     bucketone,
-		Key:        keyone,
+	t.Run("Create two documents", func(t *testing.T) {
+		t.Parallel()
+		ctx := context.Background()
+
+		cfg := &serviceconfig.BaseConfig{}
+		net := test.GetNetwork(t)
+		test.CreateDB(t, cfg, net, &test.CreateDatabaseConfig{})
+
+		queries := cfg.GetDBQueries()
+
+		clientId := "EXAMPLE"
+		err := queries.CreateClient(ctx, &repository.CreateClientParams{
+			Name:     "example_client",
+			Clientid: clientId,
+		})
+		require.NoError(t, err)
+
+		hash := "example_hash"
+		id, err := queries.CreateDocument(ctx, &repository.CreateDocumentParams{
+			Clientid: clientId,
+			Hash:     hash,
+		})
+		require.NoError(t, err)
+		assert.NotEmpty(t, id)
+
+		documentTwoID, err := queries.CreateDocument(ctx, &repository.CreateDocumentParams{
+			Clientid: clientId,
+			Hash:     "example_hash_two",
+		})
+		require.NoError(t, err)
+		assert.NotEmpty(t, documentTwoID)
+
+		docs, err := queries.ListDocumentsByClient(ctx, clientId)
+		require.NoError(t, err)
+		assert.Len(t, docs, 2)
 	})
-	require.NoError(t, err)
+	t.Run("Two clients", func(t *testing.T) {
+		t.Parallel()
+		ctx := context.Background()

-	docs, err = queries.ListDocumentsByClient(ctx, clientId)
-	require.NoError(t, err)
-	assert.Len(t, docs, 1)
-	assert.Equal(t, hash, docs[0].Hash)
+		cfg := &serviceconfig.BaseConfig{}
+		net := test.GetNetwork(t)
+		test.CreateDB(t, cfg, net, &test.CreateDatabaseConfig{})

-	documentTwoID, err := queries.CreateDocument(ctx, &repository.CreateDocumentParams{
-		Clientid: clientId,
-		Hash:     "example_hash_two",
+		queries := cfg.GetDBQueries()
+
+		clientId := "EXAMPLE"
+		err := queries.CreateClient(ctx, &repository.CreateClientParams{
+			Name:     "example_client",
+			Clientid: clientId,
+		})
+		require.NoError(t, err)
+
+		hash := "example_hash"
+		id, err := queries.CreateDocument(ctx, &repository.CreateDocumentParams{
+			Clientid: clientId,
+			Hash:     hash,
+		})
+		require.NoError(t, err)
+		assert.NotEmpty(t, id)
+
+		clientTwoId := "EXAMPLE TWO"
+		err = queries.CreateClient(ctx, &repository.CreateClientParams{
+			Name:     "example_name_two",
+			Clientid: clientTwoId,
+		})
+		require.NoError(t, err)
+
+		hashTwo := "example_hash_two"
+		idTwo, err := queries.CreateDocument(ctx, &repository.CreateDocumentParams{
+			Clientid: clientTwoId,
+			Hash:     hashTwo,
+		})
+		require.NoError(t, err)
+		assert.NotEmpty(t, idTwo)
+
+		docs, err := queries.ListDocumentsByClient(ctx, clientId)
+		require.NoError(t, err)
+		assert.Len(t, docs, 1)
+		docs, err = queries.ListDocumentsByClient(ctx, clientTwoId)
+		require.NoError(t, err)
+		assert.Len(t, docs, 1)
 	})
-	require.NoError(t, err)
-	assert.NotEmpty(t, documentTwoID)
+	t.Run("document summary", func(t *testing.T) {
+		t.Parallel()
+		ctx := context.Background()

-	buckettwo := "buckettwo"
-	keytwo := "keytwo"
-	err = queries.AddDocumentEntry(ctx, &repository.AddDocumentEntryParams{
-		Documentid: documentTwoID,
-		Bucket:     buckettwo,
-		Key:        keytwo,
+		cfg := &serviceconfig.BaseConfig{}
+		net := test.GetNetwork(t)
+		test.CreateDB(t, cfg, net, &test.CreateDatabaseConfig{})
+
+		queries := cfg.GetDBQueries()
+
+		clientId := "EXAMPLE"
+		err := queries.CreateClient(ctx, &repository.CreateClientParams{
+			Name:     "example_client",
+			Clientid: clientId,
+		})
+		require.NoError(t, err)
+
+		hash := "example_hash"
+		id, err := queries.CreateDocument(ctx, &repository.CreateDocumentParams{
+			Clientid: clientId,
+			Hash:     hash,
+		})
+		require.NoError(t, err)
+		assert.NotEmpty(t, id)
+
+		doc, err := queries.GetDocumentSummary(ctx, id)
+		require.NoError(t, err)
+		assert.EqualExportedValues(t, &repository.Document{
+			ID:       id,
+			Clientid: clientId,
+			Hash:     hash,
+		}, doc)
 	})
-	require.NoError(t, err)
+	t.Run("document external", func(t *testing.T) {
+		t.Parallel()
+		ctx := context.Background()

-	docs, err = queries.ListDocumentsByClient(ctx, clientId)
-	require.NoError(t, err)
-	assert.Len(t, docs, 2)
+		cfg := &serviceconfig.BaseConfig{}
+		net := test.GetNetwork(t)
+		test.CreateDB(t, cfg, net, &test.CreateDatabaseConfig{})

-	clientTwoId := "EXAMPLE TWO"
-	err = queries.CreateClient(ctx, &repository.CreateClientParams{
-		Name:     "example_name_two",
-		Clientid: clientTwoId,
+		queries := cfg.GetDBQueries()
+
+		clientId := "EXAMPLE"
+		err := queries.CreateClient(ctx, &repository.CreateClientParams{
+			Name:     "example_client",
+			Clientid: clientId,
+		})
+		require.NoError(t, err)
+
+		hash := "example_hash"
+		id, err := queries.CreateDocument(ctx, &repository.CreateDocumentParams{
+			Clientid: clientId,
+			Hash:     hash,
+		})
+		require.NoError(t, err)
+		assert.NotEmpty(t, id)
+
+		docext, err := queries.GetDocumentExternal(ctx, id)
+		require.NoError(t, err)
+		assert.EqualExportedValues(t, &repository.GetDocumentExternalRow{
+			ID:       id,
+			Clientid: clientId,
+			Hash:     hash,
+			Fields:   []byte("{}"),
+		}, docext)
 	})
-	require.NoError(t, err)
+	t.Run("doc id by hash", func(t *testing.T) {
+		t.Parallel()
+		ctx := context.Background()

-	docs, err = queries.ListDocumentsByClient(ctx, clientId)
-	require.NoError(t, err)
-	assert.Len(t, docs, 2)
-	docs, err = queries.ListDocumentsByClient(ctx, clientTwoId)
-	require.NoError(t, err)
-	assert.Len(t, docs, 0)
+		cfg := &serviceconfig.BaseConfig{}
+		net := test.GetNetwork(t)
+		test.CreateDB(t, cfg, net, &test.CreateDatabaseConfig{})

-	documentThreeId, err := queries.CreateDocument(ctx, &repository.CreateDocumentParams{
-		Clientid: clientTwoId,
-		Hash:     hash,
+		queries := cfg.GetDBQueries()
+
+		clientId := "EXAMPLE"
+		err := queries.CreateClient(ctx, &repository.CreateClientParams{
+			Name:     "example_client",
+			Clientid: clientId,
+		})
+		require.NoError(t, err)
+
+		hash := "example_hash"
+		id, err := queries.CreateDocument(ctx, &repository.CreateDocumentParams{
+			Clientid: clientId,
+			Hash:     hash,
+		})
+		require.NoError(t, err)
+		assert.NotEmpty(t, id)
+
+		docid, err := queries.GetDocumentIDByHash(ctx, &repository.GetDocumentIDByHashParams{
+			Hash:     hash,
+			Clientid: clientId,
+		})
+		require.NoError(t, err)
+		assert.EqualExportedValues(t, id, docid)
 	})
-	require.NoError(t, err)
+	t.Run("doc entry", func(t *testing.T) {
+		t.Parallel()
+		ctx := context.Background()

-	bucketthree := "buckettwo"
-	keythree := "keytwo"
-	err = queries.AddDocumentEntry(ctx, &repository.AddDocumentEntryParams{
-		Documentid: documentThreeId,
-		Bucket:     bucketthree,
-		Key:        keythree,
+		cfg := &serviceconfig.BaseConfig{}
+		net := test.GetNetwork(t)
+		test.CreateDB(t, cfg, net, &test.CreateDatabaseConfig{})
+
+		queries := cfg.GetDBQueries()
+
+		clientId := "EXAMPLE"
+		err := queries.CreateClient(ctx, &repository.CreateClientParams{
+			Name:     "example_client",
+			Clientid: clientId,
+		})
+		require.NoError(t, err)
+
+		hash := "example_hash"
+		id, err := queries.CreateDocument(ctx, &repository.CreateDocumentParams{
+			Clientid: clientId,
+			Hash:     hash,
+		})
+		require.NoError(t, err)
+		assert.NotEmpty(t, id)
+
+		bucketone := "bucketone"
+		keyone := "keyone"
+		err = queries.AddDocumentEntry(ctx, &repository.AddDocumentEntryParams{
+			Documentid: id,
+			Bucket:     bucketone,
+			Key:        keyone,
+		})
+		require.NoError(t, err)
+
+		entry, err := queries.GetDocumentEntry(ctx, id)
+		require.NoError(t, err)
+		assert.EqualExportedValues(t, &repository.GetDocumentEntryRow{
+			Documentid: id,
+			Bucket:     bucketone,
+			Key:        keyone,
+		}, entry)
 	})
-	require.NoError(t, err)
+	t.Run("doc latest entry", func(t *testing.T) {
+		t.Parallel()
+		ctx := context.Background()

-	docs, err = queries.ListDocumentsByClient(ctx, clientId)
-	require.NoError(t, err)
-	assert.Len(t, docs, 2)
-	docs, err = queries.ListDocumentsByClient(ctx, clientTwoId)
-	require.NoError(t, err)
-	assert.Len(t, docs, 1)
-	assert.Equal(t, hash, docs[0].Hash)
+		cfg := &serviceconfig.BaseConfig{}
+		net := test.GetNetwork(t)
+		test.CreateDB(t, cfg, net, &test.CreateDatabaseConfig{})

-	doc, err := queries.GetDocumentSummary(ctx, id)
-	require.NoError(t, err)
-	assert.EqualExportedValues(t, &repository.Document{
-		ID:       id,
-		Clientid: clientId,
-		Hash:     hash,
-	}, doc)
+		queries := cfg.GetDBQueries()

-	docext, err := queries.GetDocumentExternal(ctx, id)
-	require.NoError(t, err)
-	assert.EqualExportedValues(t, &repository.GetDocumentExternalRow{
-		ID:       id,
-		Clientid: clientId,
-		Hash:     hash,
-		Fields:   []byte("{}"),
-	}, docext)
+		clientId := "EXAMPLE"
+		err := queries.CreateClient(ctx, &repository.CreateClientParams{
+			Name:     "example_client",
+			Clientid: clientId,
+		})
+		require.NoError(t, err)

-	docid, err := queries.GetDocumentIDByHash(ctx, &repository.GetDocumentIDByHashParams{
-		Hash:     hash,
-		Clientid: clientId,
+		hash := "example_hash"
+		id, err := queries.CreateDocument(ctx, &repository.CreateDocumentParams{
+			Clientid: clientId,
+			Hash:     hash,
+		})
+		require.NoError(t, err)
+		assert.NotEmpty(t, id)
+
+		bucketone := "bucketone"
+		keyone := "keyone"
+		err = queries.AddDocumentEntry(ctx, &repository.AddDocumentEntryParams{
+			Documentid: id,
+			Bucket:     bucketone,
+			Key:        keyone,
+		})
+		require.NoError(t, err)
+
+		buckettwo := "buckettwo"
+		keytwo := "keytwo"
+		err = queries.AddDocumentEntry(ctx, &repository.AddDocumentEntryParams{
+			Documentid: id,
+			Bucket:     buckettwo,
+			Key:        keytwo,
+		})
+		require.NoError(t, err)
+
+		entry, err := queries.GetDocumentEntry(ctx, id)
+		require.NoError(t, err)
+		assert.EqualExportedValues(t, &repository.GetDocumentEntryRow{
+			Documentid: id,
+			Bucket:     buckettwo,
+			Key:        keytwo,
+		}, entry)
 	})
-	require.NoError(t, err)
-	assert.EqualExportedValues(t, id, docid)
-
-	entry, err := queries.GetDocumentEntry(ctx, id)
-	require.NoError(t, err)
-	assert.EqualExportedValues(t, &repository.GetDocumentEntryRow{
-		Documentid: id,
-		Bucket:     bucketone,
-		Key:        keyone,
-	}, entry)
-
-	entry, err = queries.GetDocumentEntry(ctx, documentTwoID)
-	require.NoError(t, err)
-	assert.EqualExportedValues(t, &repository.GetDocumentEntryRow{
-		Documentid: documentTwoID,
-		Bucket:     buckettwo,
-		Key:        keytwo,
-	}, entry)
-
-	entry, err = queries.GetDocumentEntry(ctx, documentThreeId)
-	require.NoError(t, err)
-	assert.EqualExportedValues(t, &repository.GetDocumentEntryRow{
-		Documentid: documentThreeId,
-		Bucket:     bucketthree,
-		Key:        keythree,
-	}, entry)
 }