golang/libnbd_255_opt_set_meta_queries_test.go

   1 /* libnbd golang tests
   2  * Copyright Red Hat
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Lesser General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2 of the License, or (at your option) any later version.
   8  *
   9  * This library is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Lesser General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Lesser General Public
  15  * License along with this library; if not, write to the Free Software
  16  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  17  */
  18
  19 package libnbd
  20
  21 import "testing"
  22
  23 var setq_count uint
  24 var setq_seen bool
  25
  26 func setmetaqf(user_data int, name string) int {
  27         if user_data != 42 {
  28                 panic("expected user_data == 42")
  29         }
  30         setq_count++
  31         if (name == context_base_allocation) {
  32                 setq_seen = true
  33         }
  34         return 0
  35 }
  36
  37 func Test255OptSetMetaQueries(t *testing.T) {
  38         /* Get into negotiating state. */
  39         h, err := Create()
  40         if err != nil {
  41                 t.Fatalf("could not create handle: %s", err)
  42         }
  43         defer h.Close()
  44
  45         err = h.SetOptMode(true)
  46         if err != nil {
  47                 t.Fatalf("could not set opt mode: %s", err)
  48         }
  49
  50         err = h.ConnectCommand([]string{
  51                 "nbdkit", "-s", "--exit-with-parent", "-v",
  52                 "memory", "size=1M",
  53         })
  54         if err != nil {
  55                 t.Fatalf("could not connect: %s", err)
  56         }
  57
  58         /* nbdkit does not match wildcard for SET, even though it does for LIST */
  59         setq_count = 0
  60         setq_seen = false
  61         r, err := h.OptSetMetaContextQueries([]string{"base:"},
  62                 func(name string) int {
  63                         return setmetaqf(42, name)
  64                 })
  65         if err != nil {
  66                 t.Fatalf("could not request opt_set_meta_context_queries: %s", err)
  67         }
  68         if r != setq_count || r != 0 || setq_seen {
  69                 t.Fatalf("unexpected count after opt_set_meta_context_queries")
  70         }
  71
  72         /* Negotiating with no contexts is not an error, but selects nothing.
  73          * An explicit empty list overrides a non-empty implicit list.
  74          */
  75         setq_count = 0
  76         setq_seen = false
  77         err = h.AddMetaContext(context_base_allocation)
  78         if err != nil {
  79                 t.Fatalf("could not request add_meta_context: %s", err)
  80         }
  81         r, err = h.OptSetMetaContextQueries([]string{}, func(name string) int {
  82                 return setmetaqf(42, name)
  83         })
  84         if err != nil {
  85                 t.Fatalf("could not request opt_set_meta_context_queries: %s", err)
  86         }
  87         if r != setq_count || r != 0 || setq_seen {
  88                 t.Fatalf("unexpected set_count after opt_set_meta_context_queries")
  89         }
  90         meta, err := h.CanMetaContext(context_base_allocation)
  91         if err != nil {
  92                 t.Fatalf("could not check can meta context: %s", err)
  93         }
  94         if meta {
  95                 t.Fatalf("unexpected can meta context state")
  96         }
  97
  98         /* Request 2 with expectation of 1; with SetRequestMetaContext off */
  99         setq_count = 0
 100         setq_seen = false
 101         r, err = h.OptSetMetaContextQueries([]string{
 102                 "x-nosuch:context", context_base_allocation},
 103                 func(name string) int {
 104                 return setmetaqf(42, name)
 105                 })
 106         if err != nil {
 107                 t.Fatalf("could not request opt_set_meta_context_queries: %s", err)
 108         }
 109         if r != 1 || r != setq_count || !setq_seen {
 110                 t.Fatalf("unexpected set_count after opt_set_meta_context_queries")
 111         }
 112         meta, err = h.CanMetaContext(context_base_allocation)
 113         if err != nil {
 114                 t.Fatalf("could not check can meta context: %s", err)
 115         }
 116         if !meta {
 117                 t.Fatalf("unexpected can meta context state")
 118         }
 119
 120         /* Transition to transmission phase; our last set should remain active */
 121         err = h.SetRequestMetaContext(false)
 122         if err != nil {
 123                 t.Fatalf("could not set_request_meta_context: %s", err)
 124         }
 125         err = h.OptGo()
 126         if err != nil {
 127                 t.Fatalf("could not request opt_go: %s", err)
 128         }
 129         meta, err = h.CanMetaContext(context_base_allocation)
 130         if err != nil {
 131                 t.Fatalf("could not check can meta context: %s", err)
 132         }
 133         if !meta {
 134                 t.Fatalf("unexpected can meta context state")
 135         }
 136
 137         err = h.Shutdown(nil)
 138         if err != nil {
 139                 t.Fatalf("could not request shutdown: %s", err)
 140         }
 141 }