Skip to content

Commit

Permalink
Add tests for wireguard interface
Browse files Browse the repository at this point in the history
  • Loading branch information
gfenn-newbury committed Dec 21, 2021
1 parent bb6afaf commit 7a0668e
Showing 1 changed file with 61 additions and 0 deletions.
61 changes: 61 additions & 0 deletions client/interface_wireguard_test.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,61 @@
package client

import (
"testing"

"github.com/stretchr/testify/assert"
)

func CreateInterfaceWireguardTestObjects() (*InterfaceWireguard, error) {
c := NewClient(GetCredentialsFromEnvVar())
interface_wireguard := new(InterfaceWireguard)
interface_wireguard.Name = "test_wg_interface"
interface_wireguard.ListenPort = "13231"
res, err := c.CreateInterfaceWireguard(interface_wireguard)
if err != nil {
return nil, err
}
return res, nil
}

func TestCreateInterfaceWireguard(t *testing.T) {
c := NewClient(GetCredentialsFromEnvVar())
interface_wireguard := new(InterfaceWireguard)
interface_wireguard.Name = "test_wg_interface"
interface_wireguard.ListenPort = "13231"
res, err := c.CreateInterfaceWireguard(interface_wireguard)
assert.Nil(t, err, "expecting a nil error")
assert.NotNil(t, res, "expecting a non-nil result")
assert.NotNil(t, res.ID, "expecting the to have an id")
assert.Equal(t, interface_wireguard.Name, res.Name)
assert.Equal(t, interface_wireguard.ListenPort, res.ListenPort)
err = c.DeleteInterfaceWireguard(res)
assert.Nil(t, err, "expecting a nil error on delete")
}

func TestReadInterfaceWireguard(t *testing.T) {
c := NewClient(GetCredentialsFromEnvVar())
interface_wireguard, err := CreateInterfaceWireguardTestObjects()
assert.Nil(t, err, "expecting a nil error")
res, err := c.ReadInterfaceWireguard(interface_wireguard.ID)
assert.Nil(t, err, "expecting a nil error")
assert.Equal(t, res.ID, interface_wireguard.ID)
assert.Equal(t, interface_wireguard.Name, res.Name)
assert.Equal(t, interface_wireguard.ListenPort, res.ListenPort)
err = c.DeleteInterfaceWireguard(res)
assert.Nil(t, err, "expecting a nil error on delete")
}

func TestUpdateInterfaceWireguard(t *testing.T) {
c := NewClient(GetCredentialsFromEnvVar())
interface_wireguard, err := CreateInterfaceWireguardTestObjects()
assert.Nil(t, err, "expecting a nil error")
interface_wireguard_id := interface_wireguard.ID
new_interface_wireguard := InterfaceWireguard{}
new_interface_wireguard.ListenPort = "51821"
res, err := c.UpdateInterfaceWireguard(interface_wireguard_id, &new_interface_wireguard)
assert.Nil(t, err, "expecting a nil error")
assert.Equal(t, res.ListenPort, new_interface_wireguard.ListenPort)
err = c.DeleteInterfaceWireguard(res)
assert.Nil(t, err, "expecting a nil error on delete")
}

0 comments on commit 7a0668e

Please sign in to comment.