databag/net/server/main_test.go

package main

import (
  "strings"
  "testing"
  "net/http/httptest"
  "encoding/base64"
  "encoding/json"
  app "databag/internal"
  "databag/internal/store"
)

func TestSetup(t *testing.T) {

  store.SetPath("file::memory:?cache=shared");
  //store.SetPath("databag.db");
  Claimable(t);
  Claim(t);
  Config(t);
}

func Claimable(t *testing.T) {
  r := httptest.NewRequest("GET", "/admin/claimable", nil)
  w := httptest.NewRecorder()
  app.GetNodeClaimable(w, r)
  if w.Code != 200 {
    t.Errorf("server not initially claimable")
  }
}

func Claim(t *testing.T) {
  auth := base64.StdEncoding.EncodeToString([]byte("admin:pass"))
  r := httptest.NewRequest("PUT", "/admin/claim", nil)
  r.Header.Add("Authorization","Basic " + auth)
  w := httptest.NewRecorder()
  app.SetNodeClaim(w, r)
  if w.Code != 200 {
    t.Errorf("server not initially claimable")
  }
}

func Config(t *testing.T) {
  config := app.NodeConfig{Domain: "example.com", PublicLimit: 1024, AccountStorage: 4096}
  auth := base64.StdEncoding.EncodeToString([]byte("admin:pass"))
  body,_ := json.Marshal(config)
  r := httptest.NewRequest("PUT", "/admin/config", strings.NewReader(string(body)))
  r.Header.Add("Authorization","Basic " + auth)
  w := httptest.NewRecorder()
  app.SetNodeConfig(w, r);
  if w.Code != 200 {
    t.Errorf("failed to set node config")
  }
}