File: reql_transform_table_test.go

package info (click to toggle)
golang-gopkg-rethinkdb-rethinkdb-go.v6 6.2.1-5
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 2,736 kB
  • sloc: python: 1,382; makefile: 16; sh: 9
file content (129 lines) | stat: -rw-r--r-- 4,404 bytes parent folder | download | duplicates (3)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
// Code generated by gen_tests.py and process_polyglot.py.
// Do not edit this file directly.
// The template for this file is located at:
// ../template.go.tpl
package reql_tests

import (
	"testing"
	"time"

	"github.com/stretchr/testify/suite"
	r "gopkg.in/rethinkdb/rethinkdb-go.v6"
	"gopkg.in/rethinkdb/rethinkdb-go.v6/internal/compare"
)

// Tests manipulation operations on tables
func TestTransformTableSuite(t *testing.T) {
	suite.Run(t, new(TransformTableSuite))
}

type TransformTableSuite struct {
	suite.Suite

	session *r.Session
}

func (suite *TransformTableSuite) SetupTest() {
	suite.T().Log("Setting up TransformTableSuite")
	// Use imports to prevent errors
	_ = time.Time{}
	_ = compare.AnythingIsFine

	session, err := r.Connect(r.ConnectOpts{
		Address: url,
	})
	suite.Require().NoError(err, "Error returned when connecting to server")
	suite.session = session

	r.DBDrop("db_tfm_table").Exec(suite.session)
	err = r.DBCreate("db_tfm_table").Exec(suite.session)
	suite.Require().NoError(err)
	err = r.DB("db_tfm_table").Wait().Exec(suite.session)
	suite.Require().NoError(err)

	r.DB("db_tfm_table").TableDrop("table_test_transform_table").Exec(suite.session)
	err = r.DB("db_tfm_table").TableCreate("table_test_transform_table").Exec(suite.session)
	suite.Require().NoError(err)
	err = r.DB("db_tfm_table").Table("table_test_transform_table").Wait().Exec(suite.session)
	suite.Require().NoError(err)
}

func (suite *TransformTableSuite) TearDownSuite() {
	suite.T().Log("Tearing down TransformTableSuite")

	if suite.session != nil {
		r.DB("rethinkdb").Table("_debug_scratch").Delete().Exec(suite.session)
		r.DB("db_tfm_table").TableDrop("table_test_transform_table").Exec(suite.session)
		r.DBDrop("db_tfm_table").Exec(suite.session)

		suite.session.Close()
	}
}

func (suite *TransformTableSuite) TestCases() {
	suite.T().Log("Running TransformTableSuite: Tests manipulation operations on tables")

	table_test_transform_table := r.DB("db_tfm_table").Table("table_test_transform_table")
	_ = table_test_transform_table // Prevent any noused variable errors

	{
		// transform/table.yaml line #5
		/* AnythingIsFine */
		var expected_ string = compare.AnythingIsFine
		/* table_test_transform_table.insert([{"a":["k1","v1"]},{"a":["k2","v2"]}]) */

		suite.T().Log("About to run line #5: table_test_transform_table.Insert([]interface{}{map[interface{}]interface{}{'a': []interface{}{'k1', 'v1'}, }, map[interface{}]interface{}{'a': []interface{}{'k2', 'v2'}, }})")

		runAndAssert(suite.Suite, expected_, table_test_transform_table.Insert([]interface{}{map[interface{}]interface{}{"a": []interface{}{"k1", "v1"}}, map[interface{}]interface{}{"a": []interface{}{"k2", "v2"}}}), suite.session, r.RunOpts{
			GeometryFormat: "raw",
			GroupFormat:    "map",
		})
		suite.T().Log("Finished running line #5")
	}

	{
		// transform/table.yaml line #10
		/* {"k1":"v1","k2":"v2"} */
		var expected_ map[interface{}]interface{} = map[interface{}]interface{}{"k1": "v1", "k2": "v2"}
		/* table_test_transform_table.map(r.row["a"]).coerce_to("object") */

		suite.T().Log("About to run line #10: table_test_transform_table.Map(r.Row.AtIndex('a')).CoerceTo('object')")

		runAndAssert(suite.Suite, expected_, table_test_transform_table.Map(r.Row.AtIndex("a")).CoerceTo("object"), suite.session, r.RunOpts{
			GeometryFormat: "raw",
			GroupFormat:    "map",
		})
		suite.T().Log("Finished running line #10")
	}

	{
		// transform/table.yaml line #14
		/* "SELECTION<STREAM>" */
		var expected_ string = "SELECTION<STREAM>"
		/* table_test_transform_table.limit(1).type_of() */

		suite.T().Log("About to run line #14: table_test_transform_table.Limit(1).TypeOf()")

		runAndAssert(suite.Suite, expected_, table_test_transform_table.Limit(1).TypeOf(), suite.session, r.RunOpts{
			GeometryFormat: "raw",
			GroupFormat:    "map",
		})
		suite.T().Log("Finished running line #14")
	}

	{
		// transform/table.yaml line #17
		/* "ARRAY" */
		var expected_ string = "ARRAY"
		/* table_test_transform_table.limit(1).coerce_to('array').type_of() */

		suite.T().Log("About to run line #17: table_test_transform_table.Limit(1).CoerceTo('array').TypeOf()")

		runAndAssert(suite.Suite, expected_, table_test_transform_table.Limit(1).CoerceTo("array").TypeOf(), suite.session, r.RunOpts{
			GeometryFormat: "raw",
			GroupFormat:    "map",
		})
		suite.T().Log("Finished running line #17")
	}
}