diff options
Diffstat (limited to 'relay/relay_test.go')
-rw-r--r-- | relay/relay_test.go | 86 |
1 files changed, 0 insertions, 86 deletions
diff --git a/relay/relay_test.go b/relay/relay_test.go deleted file mode 100644 index 25fcc35..0000000 --- a/relay/relay_test.go +++ /dev/null @@ -1,86 +0,0 @@ -package relay - -import ( - "context" - "testing" - - pb "github.com/boozec/rahanna/relay/proto" - "github.com/stretchr/testify/assert" - "github.com/stretchr/testify/require" -) - -func TestRegisterName(t *testing.T) { - server := &Server{} - ctx := context.Background() - - t.Run("Valid IP Registration", func(t *testing.T) { - response, err := server.RegisterName(ctx, &pb.RelayRequest{Ip: "192.168.1.1"}) - require.NoError(t, err) - assert.NotEmpty(t, response.Name) - assert.Equal(t, "192.168.1.1", response.Ip) - }) - - t.Run("Empty IP Registration", func(t *testing.T) { - response, err := server.RegisterName(ctx, &pb.RelayRequest{Ip: ""}) - assert.Error(t, err) - assert.Nil(t, response) - }) - -} - -func TestLookup(t *testing.T) { - server := &Server{} - ctx := context.Background() - - registerResponse, err := server.RegisterName(ctx, &pb.RelayRequest{Ip: "192.168.1.1"}) - require.NoError(t, err) - - t.Run("Successful Lookup", func(t *testing.T) { - response, err := server.Lookup(ctx, &pb.LookupRequest{Name: registerResponse.Name}) - require.NoError(t, err) - assert.Equal(t, registerResponse.Name, response.Name) - assert.Equal(t, "192.168.1.1", response.Ip) - }) - - t.Run("Lookup with Empty Name", func(t *testing.T) { - response, err := server.Lookup(ctx, &pb.LookupRequest{Name: ""}) - assert.Error(t, err) - assert.Nil(t, response) - }) - - t.Run("Lookup Non-Existent Name", func(t *testing.T) { - response, err := server.Lookup(ctx, &pb.LookupRequest{Name: "nonexistent"}) - assert.Error(t, err) - assert.Nil(t, response) - }) -} - -func TestCloseName(t *testing.T) { - server := &Server{} - ctx := context.Background() - - // Prepare a registered name - registerResponse, err := server.RegisterName(ctx, &pb.RelayRequest{Ip: "192.168.1.1"}) - require.NoError(t, err) - - t.Run("Successful Close", func(t *testing.T) { - response, err := server.CloseName(ctx, &pb.LookupRequest{Name: registerResponse.Name}) - require.NoError(t, err) - assert.True(t, response.Status) - - // Verify the name is no longer in the table - _, err = server.Lookup(ctx, &pb.LookupRequest{Name: registerResponse.Name}) - assert.Error(t, err) - }) - - t.Run("Close with Empty Name", func(t *testing.T) { - _, err := server.CloseName(ctx, &pb.LookupRequest{Name: ""}) - assert.Error(t, err) - }) - - t.Run("Close Non-Existent Name", func(t *testing.T) { - response, err := server.CloseName(ctx, &pb.LookupRequest{Name: "nonexistent"}) - assert.Error(t, err) - assert.False(t, response.Status) - }) -} |