Function TestFlyDbStore_Get

lib/datastore/flydb_test.go:220–279

(t *testing.T)

}
func TestFlyDbStore_Get(t *testing.T) {
	type kv struct {
		key string
		val string
	}
	type test struct {
		name        string
		input       []kv
		query       []kv
		expectError bool
	}
	tests := []test{
		{
			name: "set three",
			input: []kv{
				{key: "1", val: "2"},
				{key: "foo", val: "bar"},
				{key: "hello", val: "world"},
			},
			query: []kv{
				{key: "1", val: "2"},
				{key: "foo", val: "bar"},
				{key: "hello", val: "world"},
			},
			expectError: false,
		},
		{
			name: "non existence",
			input: []kv{
				{key: "4", val: "bar"},
			},
			query: []kv{
				{key: "1", val: ""},
				{key: "2", val: ""},
			},
			expectError: true,
		},
	}
	for _, tc := range tests {
		ds, err := testFlyDbDatastore()
		assert.NoError(t, err)
		// set all inputs
		for _, v := range tc.input {
			_ = ds.Set(stringToBytes(v.key), stringToBytes(v.val))

		}
		// recall all inputs
		for _, v := range tc.query {
			val, err := ds.Get(stringToBytes(v.key))
			if tc.expectError {
				assert.NotNil(t, err)
			} else {
				assert.NoError(t, err)
				assert.Equal(t, v.val, string(val))
			}
		}

	}