add tst

Author: Eslam-Nawara

node-registrar/tst/main.go

@@ -0,0 +1,85 @@
+package main
+
+import (
+	"encoding/hex"
+	"net/url"
+	"os"
+
+	"github.com/rs/zerolog"
+	"github.com/rs/zerolog/log"
+	"github.com/threefoldtech/tfgrid4-sdk-go/node-registrar/client"
+)
+
+var localRegistrar = "http://192.168.68.85:8080"
+
+func main() {
+	// test flow:
+	// - we need to create an account
+	// - try to get the account by id and by pk
+	// - try to update the account
+	// - try to ensure account
+	//
+	// - then we create a farm with that twin
+	// - try to get the farm
+	// - try to update the farm
+	//
+	// - try to register a node
+	// - try to update the node
+	// - try to get the node
+	// - try to send the node up time report
+	// - try to the node by twin id
+
+	log.Logger = log.Output(zerolog.ConsoleWriter{Out: os.Stderr})
+
+	hexKey := "020b5f908e821446ad3b7efd3774605f55ee41aa017968e45bbb9c84c4ca18d2"
+
+	seed, err := hex.DecodeString(hexKey)
+	if err != nil {
+		log.Fatal().Err(err).Msg("failed to decode private key")
+	}
+
+	baseURL, err := url.JoinPath(localRegistrar, "v1")
+	if err != nil {
+		log.Fatal().Err(err).Msg("failed to construct registrar url")
+	}
+
+	c, err := client.NewRegistrarClient(baseURL, seed)
+	if err != nil {
+		log.Fatal().Err(err).Msg("failed to create new registrar client")
+	}
+
+	// log.Info().Msg("test create account")
+	// relays := []string{"relay1"}
+	// rmbEncKey := "encKey"
+	// account, err := c.CreateAccount(relays, rmbEncKey)
+	// if err != nil {
+	// 	log.Fatal().Err(err).Msg("failed to create new account on registrar")
+	// }
+	//
+	// log.Info().Uint64("twinID", account.TwinID).Send()
+
+	// log.Info().Msg("test get account by public key")
+	// privateKey := ed25519.NewKeyFromSeed(seed)
+	// publicKey := privateKey.Public().(ed25519.PublicKey)
+	// account, err := c.GetAccountByPK(publicKey)
+	// if err != nil {
+	// 	log.Fatal().Err(err).Msg("failed to get account from registrar")
+	// }
+	// log.Info().Any("account", account).Send()
+
+	// log.Info().Msg("test get account by twinID")
+	// account, err = c.GetAccount(uint64(8))
+	// if err != nil {
+	// 	log.Fatal().Err(err).Msg("failed to get account from registrar")
+	// }
+	// log.Info().Any("account", account).Send()
+
+	relays := []string{"relay1"}
+	rmbEncKey := "encKey"
+	log.Info().Msg("test update account")
+	err = c.UpdateAccount(client.UpdateAccountWithRelays(relays), client.UpdateAccountWithRMBEncKey(rmbEncKey))
+	if err != nil {
+		log.Fatal().Err(err).Msg("failed to get account from registrar")
+	}
+	log.Info().Msg("account updated successfully")
+}