145 lines
3.6 KiB
Go
145 lines
3.6 KiB
Go
package structfs
|
|
|
|
import (
|
|
"encoding/json"
|
|
"io"
|
|
"net/http"
|
|
"reflect"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
var doOrig = []byte(`{
|
|
"droplet_id":2756294,
|
|
"hostname":"sample-droplet",
|
|
"vendor_data":"#cloud-config\ndisable_root: false\nmanage_etc_hosts: true\n\ncloud_config_modules:\n - ssh\n - set_hostname\n - [ update_etc_hosts, once-per-instance ]\n\ncloud_final_modules:\n - scripts-vendor\n - scripts-per-once\n - scripts-per-boot\n - scripts-per-instance\n - scripts-user\n",
|
|
"public_keys":["ssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAABAQCcbi6cygCUmuNlB0KqzBpHXf7CFYb3VE4pDOf/RLJ8OFDjOM+fjF83a24QktSVIpQnHYpJJT2pQMBxD+ZmnhTbKv+OjwHSHwAfkBullAojgZKzz+oN35P4Ea4J78AvMrHw0zp5MknS+WKEDCA2c6iDRCq6/hZ13Mn64f6c372JK99X29lj/B4VQpKCQyG8PUSTFkb5DXTETGbzuiVft+vM6SF+0XZH9J6dQ7b4yD3sOder+M0Q7I7CJD4VpdVD/JFa2ycOS4A4dZhjKXzabLQXdkWHvYGgNPGA5lI73TcLUAueUYqdq3RrDRfaQ5Z0PEw0mDllCzhk5dQpkmmqNi0F sammy@digitalocean.com"],
|
|
"region":"nyc3",
|
|
"interfaces":{
|
|
"private":[
|
|
{
|
|
"ipv4":{
|
|
"ip_address":"10.132.255.113",
|
|
"netmask":"255.255.0.0",
|
|
"gateway":"10.132.0.1"
|
|
},
|
|
"mac":"04:01:2a:0f:2a:02",
|
|
"type":"private"
|
|
}
|
|
],
|
|
"public":[
|
|
{
|
|
"ipv4":{
|
|
"ip_address":"104.131.20.105",
|
|
"netmask":"255.255.192.0",
|
|
"gateway":"104.131.0.1"
|
|
},
|
|
"ipv6":{
|
|
"ip_address":"2604:A880:0800:0010:0000:0000:017D:2001",
|
|
"cidr":64,
|
|
"gateway":"2604:A880:0800:0010:0000:0000:0000:0001"
|
|
},
|
|
"mac":"04:01:2a:0f:2a:01",
|
|
"type":"public"}
|
|
]
|
|
},
|
|
"floating_ip": {
|
|
"ipv4": {
|
|
"active": false
|
|
}
|
|
},
|
|
"dns":{
|
|
"nameservers":[
|
|
"2001:4860:4860::8844",
|
|
"2001:4860:4860::8888",
|
|
"8.8.8.8"
|
|
]
|
|
},
|
|
"features":{
|
|
"dhcp_enabled": true
|
|
}
|
|
}
|
|
`)
|
|
|
|
func server(t *testing.T, ch chan error) {
|
|
stfs := DigitalOceanMetadata{}
|
|
err := json.Unmarshal(doOrig, &stfs.Metadata.V1)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
http.Handle("/metadata/v1/", FileServer(&stfs, "json", time.Now()))
|
|
http.Handle("/metadata/v1.json", &stfs)
|
|
go func() {
|
|
ch <- http.ListenAndServe("127.0.0.1:8080", nil)
|
|
}()
|
|
time.Sleep(2 * time.Second)
|
|
}
|
|
|
|
func get(path string) ([]byte, error) {
|
|
res, err := http.Get(path)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer res.Body.Close()
|
|
return io.ReadAll(res.Body)
|
|
}
|
|
|
|
func TestAll(t *testing.T) {
|
|
ch := make(chan error)
|
|
server(t, ch)
|
|
|
|
tests := []struct {
|
|
in string
|
|
out string
|
|
}{
|
|
{"http://127.0.0.1:8080/metadata/v1/", "droplet_id\nhostname\nvendor_data\npublic_keys\nregion\ninterfaces\nfloating_ip\ndns\nfeatures"},
|
|
{"http://127.0.0.1:8080/metadata/v1/droplet_id", "2756294"},
|
|
{"http://127.0.0.1:8080/metadata/v1/dns/", "nameservers"},
|
|
{"http://127.0.0.1:8080/metadata/v1/dns/nameservers", "2001:4860:4860::8844\n2001:4860:4860::8888\n8.8.8.8"},
|
|
{"http://127.0.0.1:8080/metadata/v1/features/dhcp_enabled", "true"},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
select {
|
|
case err := <-ch:
|
|
t.Fatal(err)
|
|
default:
|
|
buf, err := get(tt.in)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if string(buf) != tt.out {
|
|
t.Errorf("req %s output %s not match requested %s", tt.in, string(buf), tt.out)
|
|
}
|
|
}
|
|
}
|
|
|
|
select {
|
|
case err := <-ch:
|
|
t.Fatal(err)
|
|
default:
|
|
doTest, err := get("http://127.0.0.1:8080/metadata/v1.json")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
oSt := DigitalOceanMetadata{}
|
|
err = json.Unmarshal(doOrig, &oSt.Metadata.V1)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
nSt := DigitalOceanMetadata{}
|
|
|
|
err = json.Unmarshal(doTest, &nSt.Metadata.V1)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if !reflect.DeepEqual(oSt, nSt) {
|
|
t.Fatalf("%v not match %v", oSt, nSt)
|
|
}
|
|
}
|
|
}
|