content_test.go

package
 
differ_test


import
 
(
	
"bytes"

	
"encoding/gob"

	
"encoding/json"

	
"fmt"

	
"io"

	
"net/http"

	
"net/http/httptest"

	
"os"

	
"testing"


	
"github.com/RedHatInsights/ccx-notification-service/conf"

	
"github.com/RedHatInsights/ccx-notification-service/differ"

	
"github.com/RedHatInsights/ccx-notification-service/types"

	
utypes
 
"github.com/RedHatInsights/insights-results-types"

	
"github.com/rs/zerolog/log"

	
"github.com/stretchr/testify/assert"

)


func
 
makeFetchRequest
(
testFileName
 
string
)
 
(
rules
 
types
.
RulesMap
,
 
err
 
error
)
 
{
	
config
,
 
err
 
:=
 
conf
.
LoadConfiguration
(
conf
.
ConfigFileEnvVariableName
,
 
conf
.
DefaultConfigFileName
)

	
if
 
err
 
!=
 
nil
 
{
		
return
 
rules
,
 
err

	
}

	
server
 
:=
 
httptest
.
NewServer
(
http
.
HandlerFunc
(
Handler
)
)

	
depencenciesConfiguration
 
:=
 
conf
.
GetDependenciesConfiguration
(
&
config
)

	
depencenciesConfiguration
.
ContentServiceServer
 
=
 
server
.
URL

	
depencenciesConfiguration
.
ContentServiceEndpoint
 
=
 
"/tests/content/"
 
+
 
testFileName

	
rules
,
 
err
 
=
 
differ
.
FetchAllRulesContent
(
&
depencenciesConfiguration
)

	
return
 
rules
,
 
err

}

func
 
TestFetchAllRulesContentOk
(
t
 
*
testing
.
T
)
 
{
	
rules
,
 
err
 
:=
 
makeFetchRequest
(
"ok.json"
)

	
assert
.
NoError
(
t
,
 
err
)

	
allContentFromMap
 
:=
 
differ
.
GetAllContentFromMap
(
rules
)

	
assert
.
NotEmpty
(
t
,
 
allContentFromMap
)

	
assert
.
Len
(
t
,
 
allContentFromMap
,
 
2
)


	
expected1
 
:=
 
utypes
.
RuleContent
{
		
Plugin
:
     
utypes
.
RulePluginInfo
{
Name
:
 
""
,
 
NodeID
:
 
""
,
 
ProductCode
:
 
""
,
 
PythonModule
:
 
""
}
,
		
ErrorKeys
:
  
map
[
string
]
utypes
.
RuleErrorKeyContent
{
"err_key"
:
 
{
Metadata
:
 
utypes
.
ErrorKeyMetadata
{
Description
:
 
""
,
 
Impact
:
 
utypes
.
Impact
{
Name
:
 
"Two"
,
 
Impact
:
 
2
}
,
 
Likelihood
:
 
0
,
 
PublishDate
:
 
"2020-04-03T16:13:30+02:00"
,
 
ResolutionRisk
:
 
3
,
 
Status
:
 
"inactive"
,
 
Tags
:
 
[
]
string
(
nil
)
}
,
 
TotalRisk
:
 
0
,
 
Generic
:
 
"Generic message"
,
 
Summary
:
 
"# Rule 1 Summary\n"
,
 
Resolution
:
 
""
,
 
MoreInfo
:
 
"# Some more information\n\n## would be put\n\n### into this file\n"
,
 
Reason
:
 
"Reason"
,
 
HasReason
:
 
true
}
}
,
		
Generic
:
    
""
,
		
Summary
:
    
"# Rule 1 Summary\n"
,
		
Resolution
:
 
""
,
		
MoreInfo
:
   
"# Some more information\n\n## would be put\n\n### into this file\n"
,
		
Reason
:
     
"Reason"
,
		
HasReason
:
  
true
,
	
}


	
expected2
 
:=
 
utypes
.
RuleContent
{
		
Plugin
:
     
utypes
.
RulePluginInfo
{
Name
:
 
""
,
 
NodeID
:
 
""
,
 
ProductCode
:
 
""
,
 
PythonModule
:
 
""
}
,
		
ErrorKeys
:
  
map
[
string
]
utypes
.
RuleErrorKeyContent
{
"err_key"
:
 
{
Metadata
:
 
utypes
.
ErrorKeyMetadata
{
Description
:
 
""
,
 
Impact
:
 
utypes
.
Impact
{
Name
:
 
""
,
 
Impact
:
 
0
}
,
 
Likelihood
:
 
0
,
 
PublishDate
:
 
"2020-04-03T16:13:30+02:00"
,
 
ResolutionRisk
:
 
0
,
 
Status
:
 
"inactive"
,
 
Tags
:
 
[
]
string
(
nil
)
}
,
 
TotalRisk
:
 
0
,
 
Generic
:
 
"Generic"
,
 
Summary
:
 
"# Rule 1 Summary\n"
,
 
Resolution
:
 
""
,
 
MoreInfo
:
 
"# Some more information\n\n## would be put\n\n### into this file\n"
,
 
Reason
:
 
"Reason"
,
 
HasReason
:
 
true
}
}
,
		
Generic
:
    
""
,
		
Summary
:
    
"# Rule 1 Summary\n"
,
		
Resolution
:
 
""
,
		
MoreInfo
:
   
"# Some more information\n\n## would be put\n\n### into this file\n"
,
		
Reason
:
     
"Reason"
,
		
HasReason
:
  
true
,
	
}


	
assert
.
Contains
(
t
,
 
allContentFromMap
,
 
expected1
)

	
assert
.
Contains
(
t
,
 
allContentFromMap
,
 
expected2
)

}


func
 
TestFetchAllRulesContentNoInternal
(
t
 
*
testing
.
T
)
 
{
	
rules
,
 
err
 
:=
 
makeFetchRequest
(
"no_internal.json"
)

	
assert
.
NoError
(
t
,
 
err
)

	
allContentFromMap
 
:=
 
differ
.
GetAllContentFromMap
(
rules
)

	
assert
.
NotEmpty
(
t
,
 
allContentFromMap
)

	
assert
.
Len
(
t
,
 
allContentFromMap
,
 
1
)


	
expected1
 
:=
 
utypes
.
RuleContent
{
		
Plugin
:
     
utypes
.
RulePluginInfo
{
Name
:
 
""
,
 
NodeID
:
 
""
,
 
ProductCode
:
 
""
,
 
PythonModule
:
 
""
}
,
		
ErrorKeys
:
  
map
[
string
]
utypes
.
RuleErrorKeyContent
{
"err_key"
:
 
{
Metadata
:
 
utypes
.
ErrorKeyMetadata
{
Description
:
 
""
,
 
Impact
:
 
utypes
.
Impact
{
Name
:
 
"Two"
,
 
Impact
:
 
2
}
,
 
Likelihood
:
 
0
,
 
PublishDate
:
 
"2020-04-03T16:13:30+02:00"
,
 
ResolutionRisk
:
 
3
,
 
Status
:
 
"inactive"
,
 
Tags
:
 
[
]
string
(
nil
)
}
,
 
TotalRisk
:
 
0
,
 
Generic
:
 
"Generic"
,
 
Summary
:
 
"# Rule 1 Summary\n"
,
 
Resolution
:
 
""
,
 
MoreInfo
:
 
"# Some more information\n\n## would be put\n\n### into this file\n"
,
 
Reason
:
 
"Reason"
,
 
HasReason
:
 
true
}
}
,
		
Generic
:
    
""
,
		
Summary
:
    
"# Rule 1 Summary\n"
,
		
Resolution
:
 
""
,
		
MoreInfo
:
   
"# Some more information\n\n## would be put\n\n### into this file\n"
,
		
Reason
:
     
"Reason"
,
		
HasReason
:
  
true
,
	
}


	
assert
.
Contains
(
t
,
 
allContentFromMap
,
 
expected1
)

}


func
 
TestFetchAllRulesContentNoExternal
(
t
 
*
testing
.
T
)
 
{
	
rules
,
 
err
 
:=
 
makeFetchRequest
(
"no_external.json"
)

	
assert
.
NoError
(
t
,
 
err
)

	
allContentFromMap
 
:=
 
differ
.
GetAllContentFromMap
(
rules
)

	
assert
.
Empty
(
t
,
 
allContentFromMap
)

}


func
 
TestFetchAllRulesContentMissing
(
t
 
*
testing
.
T
)
 
{
	
rules
,
 
err
 
:=
 
makeFetchRequest
(
"missing.json"
)

	
assert
.
NoError
(
t
,
 
err
)

	
allContentFromMap
 
:=
 
differ
.
GetAllContentFromMap
(
rules
)

	
assert
.
Empty
(
t
,
 
allContentFromMap
)

}


func
 
Handler
(
w
 
http
.
ResponseWriter
,
 
r
 
*
http
.
Request
)
 
{
	
var
 
con
 
utypes
.
RuleContentDirectory

	
print
(
r
.
RequestURI
)

	
jsonFile
,
 
err
 
:=
 
os
.
Open
(
".."
 
+
 
r
.
RequestURI
)

	
if
 
err
 
!=
 
nil
 
{
		
log
.
Fatal
(
)
.
Msg
(
err
.
Error
(
)
)

	
}

	
jsonValue
,
 
err
 
:=
 
io
.
ReadAll
(
jsonFile
)

	
if
 
err
 
!=
 
nil
 
{
		
log
.
Fatal
(
)
.
Msg
(
err
.
Error
(
)
)

	
}

	
err
 
=
 
json
.
Unmarshal
(
jsonValue
,
 
&
con
)

	
if
 
err
 
!=
 
nil
 
{
		
log
.
Fatal
(
)
.
Msg
(
err
.
Error
(
)
)

	
}

	
var
 
buffer
 
bytes
.
Buffer

	
enc
 
:=
 
gob
.
NewEncoder
(
&
buffer
)

	
err
 
=
 
enc
.
Encode
(
con
)

	
var
 
tosend
 
string

	
if
 
err
 
!=
 
nil
 
{
		
return

	
}

	
tosend
 
=
 
buffer
.
String
(
)

	
_
,
 
err
 
=
 
fmt
.
Fprint
(
w
,
 
tosend
)

	
if
 
err
 
!=
 
nil
 
{
		
log
.
Fatal
(
)
.
Msg
(
err
.
Error
(
)
)

	
}

}