Function TestAuditDocumentCreateUpdateEvents

rest/audit_test.go:1208–1293

(t *testing.T)

}

func TestAuditDocumentCreateUpdateEvents(t *testing.T) {
	rt := createAuditLoggingRestTester(t)
	defer rt.Close()

	dbConfig := rt.NewDbConfig()
	dbConfig.Logging = &DbLoggingConfig{
		Audit: &DbAuditLoggingConfig{
			Enabled: base.Ptr(true),
			EnabledEvents: base.Ptr([]uint{
				uint(base.AuditIDDocumentCreate),
				uint(base.AuditIDDocumentUpdate),
			}),
		},
	}
	if base.TestUseXattrs() {
		// this is not set automatically for CE
		dbConfig.AutoImport = base.Ptr(true)
	}

	RequireStatus(t, rt.CreateDatabase("db", dbConfig), http.StatusCreated)
	type testCase struct {
		name                string
		setupCode           func(t testing.TB, docID string) DocVersion
		auditableCode       func(t testing.TB, docID string, docVersion DocVersion)
		documentCreateCount int
		documentUpdateCount int
	}
	testCases := []testCase{
		{
			name: "create doc",
			auditableCode: func(t testing.TB, docID string, docVersion DocVersion) {
				RequireStatus(t, rt.SendAdminRequest(http.MethodPut, "/{{.keyspace}}/"+docID, `{"foo": "bar"}`), http.StatusCreated)
			},
			documentCreateCount: 1,
		},
		{
			name: "update doc",
			setupCode: func(_ testing.TB, docID string) DocVersion {
				return rt.CreateTestDoc(docID)
			},
			auditableCode: func(t testing.TB, docID string, docVersion DocVersion) {
				RequireStatus(t, rt.SendAdminRequest(http.MethodPut, "/{{.keyspace}}/"+docID+"?rev="+docVersion.RevTreeID, `{"foo": "bar"}`), http.StatusCreated)
			},
			documentUpdateCount: 1,
		},
	}
	if base.TestUseXattrs() {
		testCases = append(testCases, []testCase{
			{
				name: "import doc",
				auditableCode: func(t testing.TB, docID string, docVersion DocVersion) {
					importCount := rt.GetDatabase().DbStats.SharedBucketImport().ImportCount.Value()
					_, err := rt.GetSingleDataStore().Add(docID, 0, db.Body{"foo": "bar"})
					require.NoError(t, err)
					base.RequireWaitForStat(t, func() int64 {
						return rt.GetDatabase().DbStats.SharedBucketImport().ImportCount.Value()
					}, importCount+1)
				},