Function TestFlyDbStore_GetUint64

lib/datastore/flydb_test.go:330–389

(t *testing.T)

}
func TestFlyDbStore_GetUint64(t *testing.T) {
	type kv struct {
		key string
		val uint64
	}
	type test struct {
		name        string
		input       []kv
		query       []kv
		expectError bool
	}
	tests := []test{
		{
			name: "set three",
			input: []kv{
				{key: "1", val: 11},
				{key: "foo", val: 55},
				{key: "hello", val: 336},
			},
			query: []kv{
				{key: "1", val: 11},
				{key: "foo", val: 55},
				{key: "hello", val: 336},
			},
			expectError: false,
		},
		{
			name: "non existence",
			input: []kv{
				{key: "4", val: 2},
			},
			query: []kv{
				{key: "1", val: 2},
				{key: "2", val: 4},
			},
			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), uint64ToBytes(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, bytesToUint64(val))
			}
		}

	}