aboutsummaryrefslogtreecommitdiff
path: root/forwarder_postgresql_test.go
diff options
context:
space:
mode:
authorShulhan <ms@kilabit.info>2024-03-18 01:32:59 +0700
committerShulhan <ms@kilabit.info>2024-03-18 21:55:35 +0700
commite6fed3ef602c587602a7e1eb1de303a0aafdc527 (patch)
treeed32fd9c50d82f63c58600a12b23b9daeb2af3a9 /forwarder_postgresql_test.go
parent0b791515c819b4742c95f98f970e838b3373b0ef (diff)
downloadhaminer-e6fed3ef602c587602a7e1eb1de303a0aafdc527.tar.xz
all: implement forwarder for Postgresql
The Postgresql forwarder accept single option "URL", [forwarder "postgresql"] url = postgres://<user>:<pass>@<host>/<database>?sslmode=<> The user and database must already created first, manually.
Diffstat (limited to 'forwarder_postgresql_test.go')
-rw-r--r--forwarder_postgresql_test.go77
1 files changed, 77 insertions, 0 deletions
diff --git a/forwarder_postgresql_test.go b/forwarder_postgresql_test.go
new file mode 100644
index 0000000..1ada5d7
--- /dev/null
+++ b/forwarder_postgresql_test.go
@@ -0,0 +1,77 @@
+// SPDX-FileCopyrightText: 2024 M. Shulhan <ms@kilabit.info>
+// SPDX-License-Identifier: GPL-3.0-or-later
+
+package haminer
+
+import (
+ "encoding/json"
+ "testing"
+
+ "git.sr.ht/~shulhan/pakakeh.go/lib/test"
+)
+
+func TestForwarderPostgresql_Forwards(t *testing.T) {
+ if !testIntegration {
+ t.Skip()
+ }
+
+ var (
+ logp = `TestForwarderPostgresql_Forwards`
+
+ tdata *test.Data
+ err error
+ )
+
+ tdata, err = test.LoadData(`testdata/forwarderPostgresql_Forwards_test.txt`)
+ if err != nil {
+ t.Fatal(logp, err)
+ }
+
+ var (
+ fwdConfig = ConfigForwarder{
+ URL: `postgres://haminer:haminer@169.254.194.180/haminer?sslmode=disable`,
+ }
+
+ fwdpg *forwarderPostgresql
+ )
+
+ fwdpg, err = newForwarderPostgresql(fwdConfig)
+ if err != nil {
+ t.Fatal(logp, err)
+ }
+
+ err = fwdpg.conn.TruncateTable(tableNameHTTPLog)
+ if err != nil {
+ t.Fatal(logp, err)
+ }
+
+ var (
+ tag = `http_log.json`
+ rawb = tdata.Input[tag]
+
+ logs []*HTTPLog
+ )
+
+ err = json.Unmarshal(rawb, &logs)
+ if err != nil {
+ t.Fatal(logp, err)
+ }
+
+ fwdpg.Forwards(logs)
+
+ var listLog []HTTPLog
+
+ listLog, err = listHTTPLog(fwdpg.conn)
+ if err != nil {
+ t.Fatal(logp, err)
+ }
+
+ rawb, err = json.MarshalIndent(listLog, ``, ` `)
+ if err != nil {
+ t.Fatal(logp, err)
+ }
+
+ var exp = tdata.Output[tag]
+
+ test.Assert(t, `listHTTPLog`, string(exp), string(rawb))
+}